Support render :partial => collection of heterogeneous elements. Closes #11491.

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@9177 5ecf4fe2-1ee6-0310-87b1-e25e094e27de

2 files changed, 35 insertions, 10 deletions

diff --git a/actionpack/lib/action_view/partial_template.rb b/actionpack/lib/action_view/partial_template.rb
index a470c915cd..7d9c59a41c 100644
--- a/actionpack/lib/action_view/partial_template.rb
+++ b/actionpack/lib/action_view/partial_template.rb
@@ -7,7 +7,7 @@ module ActionView #:nodoc:
       @path, @variable_name = extract_partial_name_and_path(view, partial_path)
       super(view, @path, true, locals)
       add_object_to_local_assigns!(object)
-      
+
       # This is needed here in order to compile template with knowledge of 'counter'
       initialize_counter
       
@@ -24,11 +24,18 @@ module ActionView #:nodoc:
     def render_member(object)
       @locals[@counter_name] += 1
       @locals[:object] = @locals[@variable_name] = object
-      render
+      returning render do
+        @locals.delete(@variable_name)
+        @locals.delete(:object)
+      end
     end
     
+    def counter=(num)
+      @locals[@counter_name] = num
+    end
+
     private
-    
+
     def add_object_to_local_assigns!(object)
       @locals[:object] ||=
         @locals[@variable_name] ||=
diff --git a/actionpack/lib/action_view/partials.rb b/actionpack/lib/action_view/partials.rb
index 7f1963fb29..a92b05508a 100644
--- a/actionpack/lib/action_view/partials.rb
+++ b/actionpack/lib/action_view/partials.rb
@@ -113,9 +113,8 @@ module ActionView
           render_partial(builder_partial_path, object_assigns, (local_assigns || {}).merge(builder_partial_path.to_sym => partial_path))
         when Array, ActiveRecord::Associations::AssociationCollection, ActiveRecord::Associations::HasManyThroughAssociation
           if partial_path.any?
-            path       = ActionController::RecordIdentifier.partial_path(partial_path.first)
             collection = partial_path
-            render_partial_collection(path, collection, nil, local_assigns)
+            render_partial_collection(nil, collection, nil, local_assigns)
           else
             ""
           end
@@ -126,15 +125,34 @@ module ActionView
 
       def render_partial_collection(partial_path, collection, partial_spacer_template = nil, local_assigns = {}) #:nodoc:
         return " " if collection.empty?
-        
+
         local_assigns = local_assigns ? local_assigns.clone : {}
-        template = ActionView::PartialTemplate.new(self, partial_path, nil, local_assigns)
-        
         spacer = partial_spacer_template ? render(:partial => partial_spacer_template) : ''
-        
+
+        if partial_path.nil?
+          render_partial_collection_with_unknown_partial_path(collection, local_assigns, spacer)
+        else
+          render_partial_collection_with_known_partial_path(collection, partial_path, local_assigns, spacer)
+        end.join(spacer)
+      end
+
+      def render_partial_collection_with_known_partial_path(collection, partial_path, local_assigns, spacer)
+        template = ActionView::PartialTemplate.new(self, partial_path, nil, local_assigns)
         collection.map do |element|
           template.render_member(element)
-        end.join(spacer)
+        end
+      end
+
+      def render_partial_collection_with_unknown_partial_path(collection, local_assigns, spacer)
+        templates = Hash.new
+        i = 0
+        collection.map do |element|
+          partial_path = ActionController::RecordIdentifier.partial_path(element)
+          template = templates[partial_path] ||= ActionView::PartialTemplate.new(self, partial_path, nil, local_assigns)
+          template.counter = i
+          i += 1
+          template.render_member(element)
+        end
       end
   end
 end