Adding support to custom association parser.

lib/i18n_alchemy/proxy.rb

@@ -53,12 +53,11 @@ def method_missing(*args, &block)
       private
 
       def active_record_compatible?
-        target_class = @target.class
         target_class.respond_to?(:columns) && target_class.respond_to?(:nested_attributes_options)
       end
 
       def build_attributes
-        @target.class.columns.each do |column|
+        columns.each do |column|
           column_name = column.name
           next if column.primary || column_name.ends_with?("_id") || @localized_attributes.key?(column_name)
 
@@ -68,16 +67,16 @@ def build_attributes
       end
 
       def build_methods
-        @target.class.localized_methods.each_pair do |method, parser_type|
+        localized_methods.each_pair do |method, parser_type|
           method = method.to_s
           parser = detect_parser(parser_type)
           build_attribute(method, parser)
         end
       end
 
       def build_associations
-        @target.class.nested_attributes_options.each_key do |association_name|
-          create_localized_association(association_name)
+        nested_attributes_options.each_key do |association_name|
+          create_localized_association(association_name, detect_parser_from_association(association_name))
         end
       end
 
@@ -87,9 +86,8 @@ def build_attribute(name, parser)
         define_localized_methods(name)
       end
 
-      def create_localized_association(association_name)
-        @localized_associations <<
-          AssociationParser.new(@target.class, association_name)
+      def create_localized_association(association_name, parser)
+        @localized_associations << parser.new(target_class, association_name)
       end
 
       def create_localized_attribute(column_name, parser)
@@ -121,6 +119,10 @@ def detect_parser_from_column(column)
         detect_parser(column.number? ? :number : column.type)
       end
 
+      def detect_parser_from_association(association_name)
+        detect_parser(localized_methods[association_name] ? localized_methods[association_name] : :association)
+      end
+
       def detect_parser(type_or_parser)
         case type_or_parser
         when :number
@@ -129,10 +131,28 @@ def detect_parser(type_or_parser)
           DateParser
         when :datetime, :timestamp
           TimeParser
+        when :association
+          AssociationParser
         when ::Module
           type_or_parser
         end
       end
+
+      def localized_methods
+        target_class.localized_methods
+      end
+
+      def columns
+        target_class.columns
+      end
+
+      def nested_attributes_options
+        target_class.nested_attributes_options
+      end
+
+      def target_class
+        @target.class
+      end
     end
   end
 end

test/custom_parsers/my_custom_association_parser.rb

@@ -0,0 +1,13 @@
+class MyCustomAssociationParser < I18n::Alchemy::AssociationParser
+  def parse(attributes)
+    if association.options[:polymorphic]
+      @proxy = target_class.i18n_alchemy_proxy.localized
+    end
+
+    super
+  end
+
+  def self.localize(object)
+    object
+  end
+end

test/db/test_schema.rb

@@ -27,4 +27,14 @@
     t.string     :account_number
     t.decimal    :total_money
   end
+
+  create_table :people do |t|
+    t.string :name
+    t.integer :personable_id
+    t.string :personable_type
+  end
+
+  create_table :companies do |t|
+    t.string :register
+  end
 end

test/i18n_alchemy/proxy/attributes_parsing_test.rb

@@ -29,6 +29,8 @@ def test_assign_attributes
   end
 
   def test_mass_assigning_invalid_attribute
+    @person = Person.new
+    @person_localized = @person.localized
     assert_raises(ActiveRecord::UnknownAttributeError) do
       @localized.assign_attributes('i_dont_even_exist' => 40)
     end
@@ -128,6 +130,12 @@ def test_attributes_assignment_for_nested
     assert_equal '88,12', @supplier_localized.account.localized.total_money
   end
 
+  def test_should_assign_for_nested_attributes_for_polymorphic_association
+    @person_localized.assign_attributes(:personable_type => "Company", :personable_attributes => {:register => 10})
+    company = @person_localized.personable
+    assert_equal 10, company.register
+  end
+
   private
 
   def attributes_hash

test/models/company.rb

@@ -0,0 +1,3 @@
+class Company < ActiveRecord::Base
+  include I18n::Alchemy
+end

test/models/person.rb

@@ -0,0 +1,16 @@
+class Person < ActiveRecord::Base
+  include I18n::Alchemy
+  localize :personable, :using => MyCustomAssociationParser
+
+  belongs_to :personable, :polymorphic => true
+
+  accepts_nested_attributes_for :personable
+
+  def build_personable(attributes, options = {})
+    self.personable = personable_type.constantize.new(attributes, options)
+  end
+
+  def self.i18n_alchemy_proxy
+    Company.new
+  end
+end

test/test_helper.rb

@@ -36,6 +36,8 @@ def setup
       @supplier_localized = @supplier.localized
       @user = User.new
       @user_localized = @user.localized
+      @person = Person.new
+      @person_localized = @person.localized
 
       I18n.locale = :pt
     end