rails4: apply associations extensions to child classes

Braulio Bhavamitra
1 parent 1b6f9e37
Showing 2 changed files with 22 additions and 0 deletions Show diff stats
lib/noosfero/core_ext.rb
lib/noosfero/core_ext/active_record/reflection.rb
 require 'noosfero/core_ext/string'
 require 'noosfero/core_ext/integer'
 require 'noosfero/core_ext/active_record'
+require 'noosfero/core_ext/active_record/reflection'
+
@@ -0,0 +1,20 @@
+
+# on STI classes tike Article and Profile, plugins' extensions
+# on associations should be reflected on descendants
+module ActiveRecord
+  module Reflection
+
+    class << self
+
+      def add_reflection_with_descendants(ar, name, reflection)
+        self.add_reflection_without_descendants ar, name, reflection
+        ar.descendants.each do |k|
+          k._reflections.merge!(name.to_s => reflection)
+        end if ar.base_class == ar
+      end
+
+      alias_method_chain :add_reflection, :descendants
+
+    end
+  end
+end
1	require 'noosfero/core_ext/string'	1	require 'noosfero/core_ext/string'
2	require 'noosfero/core_ext/integer'	2	require 'noosfero/core_ext/integer'
3	require 'noosfero/core_ext/active_record'	3	require 'noosfero/core_ext/active_record'
		4	+require 'noosfero/core_ext/active_record/reflection'
		5	+
@@ -0,0 +1,20 @@		@@ -0,0 +1,20 @@
	1	+
	2	+# on STI classes tike Article and Profile, plugins' extensions
	3	+# on associations should be reflected on descendants
	4	+module ActiveRecord
	5	+ module Reflection
	6	+
	7	+ class << self
	8	+
	9	+ def add_reflection_with_descendants(ar, name, reflection)
	10	+ self.add_reflection_without_descendants ar, name, reflection
	11	+ ar.descendants.each do \|k\|
	12	+ k._reflections.merge!(name.to_s => reflection)
	13	+ end if ar.base_class == ar
	14	+ end
	15	+
	16	+ alias_method_chain :add_reflection, :descendants
	17	+
	18	+ end
	19	+ end
	20	+end