module MongoMapper
  module Plugins
    module Associations
      class ManyEmbeddedPolymorphicProxy < EmbeddedCollection
        def replace(values)
          @_values = values.map do |v|
            v.respond_to?(:attributes) ? v.attributes.merge(reflection.type_key_name => v.class.name) : v
          end
          reset
        end

        protected
          def find_target
            (@_values || []).map do |hash|
              child = polymorphic_class(hash).load(hash)
              assign_references(child)
              child
            end
          end

          def polymorphic_class(doc)
            if class_name = doc[reflection.type_key_name]
              class_name.constantize
            else
              klass
            end
          end
      end
    end
  end
end