lib/fuzzily/model.rb in fuzzily-0.0.3 vs lib/fuzzily/model.rb in fuzzily-0.1.0

@@ -3,26 +3,29 @@
     # Needs fields: trigram, owner_type, owner_id, score
     # Needs index on [owner_type, trigram] and [owner_type, owner_id]
 
     def self.included(by)
       by.ancestors.include?(ActiveRecord::Base) or raise 'Not included in an ActiveRecord subclass'
+
+      scope_method = ActiveRecord::VERSION::MAJOR == 2 ? :named_scope : :scope
+
       by.class_eval do
         return if class_variable_defined?(:@@fuzzily_trigram_model)
 
         belongs_to :owner, :polymorphic => true
         validates_presence_of     :owner
         validates_uniqueness_of   :trigram, :scope => [:owner_type, :owner_id]
         validates_length_of       :trigram, :is => 3
         validates_presence_of     :score
         validates_presence_of     :fuzzy_field
 
-        named_scope :for_model,  lambda { |model| { 
+        send scope_method, :for_model,  lambda { |model| { 
           :conditions => { :owner_type => model.kind_of?(Class) ? model.name : model  } 
         }}
-        named_scope :for_field,  lambda { |field_name| {
+        send scope_method, :for_field,  lambda { |field_name| {
           :conditions => { :fuzzy_field => field_name }
         }}
-        named_scope :with_trigram, lambda { |trigrams| {
+        send scope_method, :with_trigram, lambda { |trigrams| {
           :conditions => { :trigram => trigrams }
         }}
 
         class_variable_set(:@@fuzzily_trigram_model, true)
       end