primer 0.1.0 → 0.2.0

data/lib/primer/worker/active_record_agent.rb

@@ -1,36 +1,30 @@
 module Primer
   class Worker
     
-    class ActiveRecordAgent
-      def self.bind_to_bus
-        Primer.bus.subscribe :active_record do |event, class_name, attributes, changes|
-          model = class_name.constantize.new(attributes)
-          model.instance_eval do
-            @attributes = attributes
-            @changed_attributes = changes if changes
-          end
-          __send__("on_#{event}", model)
-        end
-      end
-      
-      def self.macros
-        Watcher::ActiveRecordMacros
-      end
-      
-      def self.on_create(model)
-        notify_belongs_to_associations(model)
-      end
+    class ActiveRecordAgent < Agent
+      topic :active_record
       
-      def self.on_update(model)
-        notify_attributes(model, model.changes)
-      end
-      
-      def self.on_destroy(model)
-        notify_attributes(model, model.attributes)
-        notify_has_many_associations(model)
+      def on_message(message)
+        event, class_name, attributes, changes = *message
+        model = class_name.constantize.new(attributes)
+        
+        model.instance_eval do
+          @attributes = attributes
+          @changed_attributes = changes if changes
+        end
+        
+        case event
+          when 'create'
+            notify_belongs_to_associations(model)
+          when 'update'
+            notify_attributes(model, model.changes)
+          when 'destroy'
+            notify_attributes(model, model.attributes)
+            notify_has_many_associations(model)
+        end
       end
       
-      def self.notify_attributes(model, fields)
+      def notify_attributes(model, fields)
         foreign_keys = model.class.primer_foreign_key_mappings
         
         fields.each do |attribute, value|
@@ -42,14 +36,14 @@ module Primer
         end
       end
       
-      def self.notify_belongs_to_associations(model)
+      def notify_belongs_to_associations(model)
         model.class.reflect_on_all_associations.each do |assoc|
           next unless assoc.macro == :belongs_to
           notify_belongs_to_association(model, assoc.name)
         end
       end
       
-      def self.notify_belongs_to_association(model, assoc_name, change = nil)
+      def notify_belongs_to_association(model, assoc_name, change = nil)
         assoc = model.class.reflect_on_association(assoc_name)
         owner_class = assoc.class_name.constantize
         
@@ -69,7 +63,7 @@ module Primer
         notify_has_many_through_association(previous, mirror.name)
       end
       
-      def self.notify_has_many_associations(model)
+      def notify_has_many_associations(model)
         model.class.reflect_on_all_associations.each do |assoc|
           next unless assoc.macro == :has_many
           next if assoc.options[:dependent] == :destroy
@@ -87,7 +81,7 @@ module Primer
         end
       end
       
-      def self.notify_has_many_through_association(model, through_name)
+      def notify_has_many_through_association(model, through_name)
         model.class.reflect_on_all_associations.each do |assoc|
           next unless assoc.macro == :has_many
           
@@ -107,10 +101,13 @@ module Primer
         end
       end
       
-      def self.mirror_association(object_class, related_class, macro)
+      def mirror_association(object_class, related_class, macro)
+        long_name  = object_class.name
+        short_name = long_name.split('::').last
+        
         related_class.reflect_on_all_associations.find do |mirror_assoc|
           mirror_assoc.macro == macro and
-          mirror_assoc.class_name == object_class.name
+          [long_name, short_name].include?(mirror_assoc.class_name)
         end
       end
     end