Merge pull request #62 from Maners/collection-template

FormCollection: Set twb-layout on template element
neilime · Mar 19, 2014 · e7b90a1 · e7b90a1
2 parents 19460e2 + ffffdcb
commit e7b90a1
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/TwbBundle/Form/View/Helper/TwbBundleFormCollection.php b/src/TwbBundle/Form/View/Helper/TwbBundleFormCollection.php
@@ -2,6 +2,8 @@
 
 namespace TwbBundle\Form\View\Helper;
 
+use Zend\Form\Element\Collection as CollectionElement;
+
 class TwbBundleFormCollection extends \Zend\Form\View\Helper\FormCollection {
 
     /**
@@ -89,4 +91,19 @@ public function render(\Zend\Form\ElementInterface $oElement) {
         return $sMarkup;
     }
 
+    /**
+     * Only render a template
+     *
+     * @param CollectionElement $collection
+     * @return string
+     */
+    public function renderTemplate(CollectionElement $collection)
+    {
+        if($sElementLayout = $collection->getOption('twb-layout')) {
+            $elementOrFieldset = $collection->getTemplateElement();
+            $elementOrFieldset->setOption('twb-layout', $sElementLayout);
+        }
+
+        return parent::renderTemplate($collection);
+    }
 }