delegate_all_for 0.0.6 → 0.0.7

data/lib/delegate_all_for/version.rb

@@ -1,3 +1,3 @@
 module DelegateAllFor
-  VERSION = "0.0.6"
+  VERSION = "0.0.7"
 end

data/lib/delegate_all_for.rb

@@ -28,16 +28,33 @@ module DelegateAllFor
     exclude_columns = self.column_names.dup.concat(options[:except].map(&:to_s))
     attr_names.each do |association_name|
       if reflection = reflect_on_association(association_name)
-        delegate_opts = options.slice(:prefix, :allow_nil).merge(to: association_name)
+        to = association_name
+        delegate_opts = options.slice(:prefix, :allow_nil).merge(to: to)
+        method_prefix = delegate_opts[:prefix] ? "#{prefix == true ? to : prefix}_" : ''
+
         options[:also_include].each do |m|
           class_eval(%{delegate :#{m}, #{delegate_opts}})
         end
-        (reflection.klass.column_names - exclude_columns).each do |column_name|
-          next if column_name.in?(reflection.foreign_key, 'updated_at', 'updated_on', 'created_at', 'created_on')
+
+        (reflection.klass.column_names - exclude_columns).each do |method|
+          next if method.in?(reflection.foreign_key, 'updated_at', 'updated_on', 'created_at', 'created_on')
+          class_eval <<-eoruby, __FILE__, __LINE__ + 1
+            delegate :#{method},  #{delegate_opts}
+            delegate :#{method}?, #{delegate_opts}
+          eoruby
+
+          # Create the setter with support for using nested attributes to set it if the delegated object is not present
+          exception = %(raise "#{self}##{method_prefix}#{method} delegated to #{to}.#{method}, but #{to} is nil: \#{self.inspect}")
           class_eval <<-eoruby, __FILE__, __LINE__ + 1
-            delegate :#{column_name},  #{delegate_opts}
-            delegate :#{column_name}=, #{delegate_opts}
-            delegate :#{column_name}?, #{delegate_opts}
+            def #{method_prefix}#{method}=(*args, &block)                             # def customer_name(*args, &block)
+              if #{to} || #{to}.respond_to?(:#{method})                               #   if client || client.respond_to?(:name)
+                #{to}.__send__(:#{method}=, *args, &block)                            #     client.__send__(:name, *args, &block)
+              elsif self.respond_to?(:#{to}_attributes=)                              #   elsif self.respond_to?(:client_attributes=)
+                self.__send__(:#{to}_attributes=, :#{method} => args.first, &block)   #     self.__send__(:client_attributes=, :name => args.first, &block)
+              else                                                                    #   else
+                #{exception}                                                          #     # add helpful exception
+              end                                                                     #   end
+            end                                                                       # end
           eoruby
         end
       else

data/spec/delegate_all_for/delegate_all_for_spec.rb

@@ -55,6 +55,29 @@ describe DelegateAllFor do
       it 'does not delegate to timestamp attributes' do
         lambda { subject.created_at }.should raise_error NoMethodError
       end
+
+      describe 'writer' do
+        it 'sets the value using delegation if child is set' do
+          subject.child.should be_an_instance_of(Child)
+          subject.four = 'FOUR'
+          subject.four.should == 'FOUR'
+          subject.child.four.should == 'FOUR'
+        end
+
+        it 'sets the value using nested attributes if child is not set and nested attributes are supported' do
+          class NestedParent < Parent; accepts_nested_attributes_for :child end
+          subject = NestedParent.new
+          subject.four = 'FOUR'
+          subject.child.should be_an_instance_of(Child)
+          subject.four.should == 'FOUR'
+          subject.child.four.should == 'FOUR'
+        end
+
+        it 'raises an exception if child is not set' do
+          subject = Parent.new
+          lambda { subject.four = 'FOUR' }.should raise_error RuntimeError
+        end
+      end
     end
 
     describe 'guards against user error' do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: delegate_all_for
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-05-08 00:00:00.000000000 Z
+date: 2012-05-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
-  requirement: &70104482200140 !ruby/object:Gem::Requirement
+  requirement: &70148514385220 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -21,10 +21,10 @@ dependencies:
         version: 3.2.3
   type: :runtime
   prerelease: false
-  version_requirements: *70104482200140
+  version_requirements: *70148514385220
 - !ruby/object:Gem::Dependency
   name: rake
-  requirement: &70104482199360 !ruby/object:Gem::Requirement
+  requirement: &70148514384040 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -32,10 +32,10 @@ dependencies:
         version: 0.9.2.2
   type: :development
   prerelease: false
-  version_requirements: *70104482199360
+  version_requirements: *70148514384040
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &70104482198200 !ruby/object:Gem::Requirement
+  requirement: &70148514383580 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -43,10 +43,10 @@ dependencies:
         version: 2.9.0
   type: :development
   prerelease: false
-  version_requirements: *70104482198200
+  version_requirements: *70148514383580
 - !ruby/object:Gem::Dependency
   name: sqlite3
-  requirement: &70104482197160 !ruby/object:Gem::Requirement
+  requirement: &70148514383080 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -54,7 +54,7 @@ dependencies:
         version: 1.3.6
   type: :development
   prerelease: false
-  version_requirements: *70104482197160
+  version_requirements: *70148514383080
 description: Easy delegation of all columns of an ActiveRecord association
 email:
 - jason@lessonplanet.com