lib/reputation/user.rb in reputation-0.0.2 vs lib/reputation/user.rb in reputation-0.0.3

@@ -8,21 +8,35 @@
     
     module Relations
       def self.included(model)
         model.class_eval do
           
-          has_many :behaviours, :class_name => "ReputationBehaviour" do
-            def add(rule, metric)
-              rule = ReputationRule.find_by_name rule unless rule.is_a? ReputationRule
-              b = proxy_reflection.klass.find_by_user_id_and_rule_id proxy_owner.id, rule.id
-              b ||= proxy_reflection.klass.new :rule => rule, :user => proxy_owner
-              b.update_attribute :metric, metric
-              proxy_owner.behaviours(true)
-              b
-            end  
-          end
+          has_many :behaviours, :class_name => "ReputationBehaviour", :extend => AddRails2 if Reputation::RailsVersion.rails_2?
+          has_many :behaviours, :class_name => "ReputationBehaviour", :extend => AddRails3 if Reputation::RailsVersion.rails_3?
           
         end
+      end
+    end
+    
+    module AddRails2
+      def add(rule, metric)
+        rule = ReputationRule.find_by_name rule unless rule.is_a? ReputationRule
+        b = proxy_reflection.klass.find_by_user_id_and_rule_id proxy_owner.id, rule.id
+        b ||= proxy_reflection.klass.new :rule => rule, :user => proxy_owner
+        b.update_attribute :metric, metric
+        proxy_owner.behaviours(true)
+        b
+      end
+    end
+    
+    module AddRails3
+      def add(rule, metric)
+        rule = ReputationRule.find_by_name rule unless rule.is_a? ReputationRule
+        b = reflection.klass.find_by_user_id_and_rule_id owner.id, rule.id
+        b ||= reflection.klass.new :rule => rule, :user => owner
+        b.update_attribute :metric, metric
+        owner.behaviours(true)
+        b
       end
     end
     
     module InstanceMethods
       
\ No newline at end of file