poly_belongs_to 0.1.6 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 83e5505c4c5c26183c2dfa5f8d85fdae05961441
-  data.tar.gz: e5e52a5f7f72f4c6b44ec06d499fd108e676faf3
+  metadata.gz: 6db0bb5452d0a17b0a99a45f875cd30532ade12a
+  data.tar.gz: 976a70a95737ad12583cde8401dc30411b85c1a8
 SHA512:
-  metadata.gz: 54de1623a7a0534038290b76cb9a85731c62466870c3634dc29690c5c77bf6e87756cc406fc908b9e1ce0ad418f1ba14a203a5cbf233566f4939416d2dcda20c
-  data.tar.gz: 36a62d5c63c4474f9840feb49b2c2ce743a8958b12a875a1b03df49018f97682acfb9d5a9c14ac03ba6c496d9a3e4587b933762d20881c3c4ee0b0d4490e89f1
+  metadata.gz: 6d49e5970ec7375eddfc93752569b611c11b741c1caf9181da5e58ac91dea9a702cf28c167d2b87be4347863636d565553983584add6bd9ac21e2f263f782040
+  data.tar.gz: cde4c040599f9cd89c41ed96141d1ed811a8420996cf01e812b31b4ad7ced1d886a9aa6e4867799278d514a85e90d66131b7ea6408a97c748cbb7659ccfab100

data/README.md

@@ -119,6 +119,11 @@ PolyBelongsTo::Pbt::IsPlural[ obj, child ]
 #       For has_one it's the object ref itself.
 PolyBelongsTo::Pbt::CollectionProxy[ obj, child ]
 
+# Always returns a collection proxy; fakes a collection proxy for has_one.
+PolyBelongsTo::Pbt::AsCollectionProxy[ obj, child ]
+
+# Wrapper for has_one objects to be a collection proxy
+PolyBelongsTo::FakedCollection.new(obj, child)
 ```
 ##Record Duplication
 

data/lib/poly_belongs_to/dup.rb

@@ -5,10 +5,8 @@ module PolyBelongsTo
     included do
       def self.pbt_dup_build(item_to_build_on, item_to_duplicate)
         if PolyBelongsTo::Pbt::IsReflected[item_to_build_on, item_to_duplicate]
-          build_cmd = PolyBelongsTo::Pbt::BuildCmd[item_to_build_on, item_to_duplicate]
-          dup_attrs = PolyBelongsTo::Pbt::AttrSanitizer[item_to_duplicate]
-
-          build_cmd ? eval("item_to_build_on.#{build_cmd}(#{dup_attrs})") : nil
+          PolyBelongsTo::Pbt::AsCollectionProxy[item_to_build_on, item_to_duplicate].
+            build PolyBelongsTo::Pbt::AttrSanitizer[item_to_duplicate]
         end
       end
 
@@ -16,26 +14,12 @@ module PolyBelongsTo
         pbt_dup_build(item_to_build_on, item_to_duplicate)
         PolyBelongsTo::Pbt::Reflects[item_to_duplicate].each do |ref|
           child = eval("item_to_duplicate.#{ref}")
-          core  = eval("item_to_build_on.#{PolyBelongsTo::Pbt::CollectionProxy[item_to_build_on, item_to_duplicate]}")
-          if child.respond_to?(:build)
-            child.each do |spawn|
-              if core.respond_to?(:build)
-                core.each do |subscore|
-                  subscore.pbt_deep_dup_build(spawn)
-                end
-              else
-                core.pbt_deep_dup_build(spawn)
-              end
-            end
-          else
-            if core.respond_to?(:build)
-              core.each do |subcore|
-                subcore.pbt_deep_dup_build(child)
-              end
-            else
-              core.pbt_deep_dup_build(child)
+          PolyBelongsTo::Pbt::AsCollectionProxy[item_to_build_on, item_to_duplicate].
+            each do |builder|
+              child.respond_to?(:build) ? child.each {|spawn|
+                builder.pbt_deep_dup_build(spawn)
+              } : builder.pbt_deep_dup_build(child)
             end
-          end
         end
         item_to_build_on       
       end

data/lib/poly_belongs_to/faked_collection.rb

@@ -0,0 +1,64 @@
+module PolyBelongsTo
+  class FakedCollection
+    def initialize(obj, child)
+      raise "Not a has_one rleationship for FakedCollection" unless PolyBelongsTo::Pbt::IsSingular[obj,child]
+      @obj = obj
+      @child = child
+      @instance = eval("@obj.#{PolyBelongsTo::Pbt::CollectionProxy[@obj,@child]}")
+      self
+    end
+
+    def all
+      Array[@instance].compact
+    end
+
+    def first
+      @instance
+    end
+
+    def last
+      @instance
+    end
+
+    def size
+      all.size
+    end
+
+    def ancestors
+      klass.ancestors.unshift(PolyBelongsTo::FakedCollection)
+    end
+
+    def kind_of?(thing)
+      ancestors.include? thing
+    end
+
+    def count
+      size
+    end
+
+    def klass
+      @instance.class
+    end
+
+    def build(*args)
+      @instance = eval("@obj.#{PolyBelongsTo::Pbt::BuildCmd[@obj, @child]}(#{args.join(',')})")
+      self
+    end
+
+    def each(*args, &block)
+      all.each(*args, &block)
+    end
+
+    def respond_to?(method_name)
+      @instance.respond_to?(method_name) || super
+    end
+
+    def method_missing(method_name, *args, &block)
+      if @instance.respond_to?(method_name)
+        @instance.send method_name, *args, &block
+      else
+        super
+      end
+    end
+  end
+end

data/lib/poly_belongs_to/poly_belongs_to.rb

@@ -58,5 +58,15 @@ module PolyBelongsTo
       proxy = ActiveModel::Naming.plural(child).to_sym
       reflects.include?(proxy) ? proxy : nil
     }
+  
+    AsCollectionProxy = lambda {|obj, child|
+      return [] unless obj && child
+      reflects = Reflects[obj]
+      proxy = ActiveModel::Naming.singular(child).to_sym
+      return PolyBelongsTo::FakedCollection.new(obj, child) if reflects.include? proxy
+      proxy = ActiveModel::Naming.plural(child).to_sym
+      reflects.include?(proxy) ? eval("obj.#{PolyBelongsTo::Pbt::CollectionProxy[obj, child]}") : []
+    }
   end
+  
 end

data/lib/poly_belongs_to/version.rb

@@ -1,3 +1,3 @@
 module PolyBelongsTo
-  VERSION = "0.1.6"
+  VERSION = "0.1.7"
 end

data/lib/poly_belongs_to.rb

@@ -5,6 +5,7 @@ $: << File.join(File.dirname(__FILE__), "/poly_belongs_to")
 require 'poly_belongs_to/version'
 require 'poly_belongs_to/dup'
 require 'poly_belongs_to/poly_belongs_to'
+require 'poly_belongs_to/faked_collection'
 require 'active_support/concern'
 
 module PolyBelongsTo