Moving ActiveRecord extension call into the rails initialization blocks

mrkurt · Jan 31, 2010 · a336a9c · a336a9c
1 parent 6d20ff7
commit a336a9c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/lib/loofah.rb b/lib/loofah.rb
@@ -87,8 +87,10 @@ def scrub_xml_document(string_or_io, method)
   Rails.configuration.after_initialize do
     require 'loofah/active_record'
     require 'loofah/xss_foliate'
+    ActiveRecord::Base.extend(Loofah::ActiveRecordExtension)
   end
 elsif defined? ActiveRecord::Base # rails 2.0
   require 'loofah/active_record'
   require 'loofah/xss_foliate'
+  ActiveRecord::Base.extend(Loofah::ActiveRecordExtension)
 end
diff --git a/lib/loofah/active_record.rb b/lib/loofah/active_record.rb
@@ -59,4 +59,3 @@ def html_document(attr, options={})
   end
 end
 
-ActiveRecord::Base.extend(Loofah::ActiveRecordExtension)