reference_tracking 0.0.3 → 0.0.4

data/lib/reference_tracking/action_controller.rb

@@ -1,3 +1,5 @@
+require 'active_support/core_ext/array/grouping'
+
 module ReferenceTracking
   module ActionController
     module ActMacro
@@ -59,12 +61,40 @@ module ReferenceTracking
     end
 
     module Purging
+      SKIP_ATTRIBUTES = %w(id created_at updated_at)
+
       def purge(objects)
-        tags = objects.map do |object|
-          methods = (object.previous_changes.keys & object.attributes.keys) - %w(updated_at)
+        tags = objects.map { |object| purge_tags_for(object)  }.flatten
+        response.headers[purging_options[:header]] = tags
+      end
+
+      def purge_tags_for(object)
+        case params[:action]
+        when 'create', 'destroy'
+          tags = purge_associations_for(object).map do |owner, method|
+            ReferenceTracking.to_tag(owner, method)
+          end
+          tags << ReferenceTracking.to_tag(object) if params[:action] == 'destroy'
+          tags
+        when 'update'
+          methods = object.previous_changes.keys & object.attributes.keys - SKIP_ATTRIBUTES
           methods.map { |method| ReferenceTracking.to_tag(object, method) }
         end
-        response.headers[purging_options[:header]] = tags.flatten
+      end
+
+      def purge_associations_for(object)
+        superclasses = purge_super_classes_for(object)
+        object.class.reflect_on_all_associations(:belongs_to).map do |association|
+          if owner = object.send(association.name)
+            owner.class.reflect_on_all_associations.map do |association|
+              [owner, association.name] if superclasses.include?(association.class_name)
+            end
+          end
+        end.flatten.compact.in_groups_of(2)
+      end
+
+      def purge_super_classes_for(object)
+        object.class.ancestors.select { |klass| klass < ActiveRecord::Base }.map(&:name)
       end
     end
   end

data/lib/reference_tracking/version.rb

@@ -1,3 +1,3 @@
 module ReferenceTracking
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: reference_tracking
 version: !ruby/object:Gem::Version 
-  hash: 25
+  hash: 23
   prerelease: false
   segments: 
   - 0
   - 0
-  - 3
-  version: 0.0.3
+  - 4
+  version: 0.0.4
 platform: ruby
 authors: 
 - Sven Fuchs
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-10-03 00:00:00 +02:00
+date: 2010-10-10 00:00:00 +02:00
 default_executable: 
 dependencies: []