lib/ninsho/interface.rb in ninsho-0.1.1 vs lib/ninsho/interface.rb in ninsho-0.1.2

@@ -14,17 +14,30 @@
 
       attr_accessor :auth_hash
 
       def belongs_to_ninsho(*args)
         options = args.extract_options!
+        Ninsho.parent_resource_holding_attributes = options[:hold_attributes] || []
+        options.reject! { |k, v| k == :hold_attributes }
         
         associations = args.collect(&:to_s).collect(&:downcase)
         association_keys = associations.collect { |association| "#{association}_id" } 
 
         #Set the belongs_to association by ActiveRecord
         associations.each do |associated_model|
-          belongs_to associated_model.to_sym
+          belongs_to associated_model.to_sym, options
           Ninsho.parent_resource_name = Ninsho.ref(associated_model.to_s.classify).get
+        end
+
+        ## Attributes delegation
+        Ninsho.parent_resource_holding_attributes.each do |attr|
+          delegate attr, to: Ninsho.parent_resource_name.to_s.downcase.to_sym
+
+          class_eval <<-RUBY
+            def #{attr}=(value)
+              self.#{Ninsho.parent_resource_name.to_s.downcase.to_sym}.#{attr} = value
+            end
+          RUBY
         end
       end
 
       # Responsible for creating or find the record with the
       # omniauth hash