Make allowObjectBinding configurable for Fieldsets

library/Zend/Form/Annotation/ComposedObject.php

@@ -51,6 +51,6 @@ public function isCollection()
      */
     public function getOptions()
     {
-        return isset($this->value['options']) ? $this->value['options'] : array();
+        return is_array($this->value) && isset($this->value['options']) ? $this->value['options'] : array();
     }
 }

library/Zend/Form/Annotation/ElementAnnotationsListener.php

@@ -163,8 +163,14 @@ public function handleComposedObjectAnnotation($e)
                 $specification['type'] = 'Zend\Form\Fieldset';
             }
 
+            // Merge options of composed object with the ones of the target object:
+            $options = (isset($elementSpec['spec']['options']) && is_array($elementSpec['spec']['options'])) ? $elementSpec['spec']['options'] : array();
+            $options = array_merge($options, $annotation->getOptions());
+
+            // Add element spec:
             $elementSpec['spec'] = $specification;
             $elementSpec['spec']['name'] = $name;
+            $elementSpec['spec']['options'] = new ArrayObject($options);
         }
     }
 

library/Zend/Form/Fieldset.php

@@ -10,6 +10,7 @@
 namespace Zend\Form;
 
 use Traversable;
+use Zend\Code\Reflection\ClassReflection;
 use Zend\Stdlib\Hydrator;
 use Zend\Stdlib\Hydrator\HydratorAwareInterface;
 use Zend\Stdlib\Hydrator\HydratorInterface;
@@ -68,6 +69,13 @@ class Fieldset extends Element implements FieldsetInterface
      */
     protected $useAsBaseFieldset = false;
 
+    /**
+     * The class or interface of objects that can be bound to this fieldset.
+     *
+     * @var string
+     */
+    protected $allowedObjectBindingClass;
+
     /**
      * @param  null|int|string  $name    Optional name for the element
      * @param  array            $options Optional options for the element
@@ -94,6 +102,10 @@ public function setOptions($options)
             $this->setUseAsBaseFieldset($options['use_as_base_fieldset']);
         }
 
+        if (isset($options['allowed_object_binding_class'])) {
+            $this->setAllowedObjectBindingClass($options['allowed_object_binding_class']);
+        }
+
         return $this;
     }
 
@@ -465,6 +477,26 @@ public function getObject()
         return $this->object;
     }
 
+    /**
+     * Set the class or interface of objects that can be bound to this fieldset.
+     *
+     * @param string $allowObjectBindingClass
+     */
+    public function setAllowedObjectBindingClass($allowObjectBindingClass)
+    {
+        $this->allowedObjectBindingClass = $allowObjectBindingClass;
+    }
+
+    /**
+     * Get The class or interface of objects that can be bound to this fieldset.
+     *
+     * @return string
+     */
+    public function allowedObjectBindingClass()
+    {
+        return $this->allowedObjectBindingClass;
+    }
+
     /**
      * Checks if the object can be set in this fieldset
      *
@@ -473,7 +505,17 @@ public function getObject()
      */
     public function allowObjectBinding($object)
     {
-        return ($this->object && $object instanceof $this->object);
+        $validBindingClass = false;
+        if (is_object($object) && $this->allowedObjectBindingClass()) {
+            $objectClass = ltrim($this->allowedObjectBindingClass(), '\\');
+            $reflection = new ClassReflection($object);
+            $validBindingClass = (
+                $reflection->getName() == $objectClass
+                || $reflection->isSubclassOf($this->allowedObjectBindingClass())
+            );
+        }
+
+        return ($validBindingClass || $this->object && $object instanceof $this->object);
     }
 
     /**

tests/ZendTest/Form/FieldsetTest.php

@@ -452,6 +452,16 @@ public function testSetOptionsUseAsBaseFieldset()
         $this->assertEquals('bar', $option);
     }
 
+    public function testSetOptionAllowedObjectBindingClass()
+    {
+        $this->fieldset->setOptions(array(
+                                         'allowed_object_binding_class' => 'bar'
+                                    ));
+        $option = $this->fieldset->getOption('allowed_object_binding_class');
+
+        $this->assertEquals('bar', $option);
+    }
+
     /**
      * @expectedException Zend\Form\Exception\InvalidElementException
      */
@@ -471,4 +481,23 @@ public function testSetObjectWithStringRaisesException()
         $this->setExpectedException('Zend\Form\Exception\InvalidArgumentException');
         $this->fieldset->setObject('foo');
     }
+
+    public function testShouldValidateAllowObjectBindingByClassname()
+    {
+        $object = new \stdClass();
+        $this->fieldset->setAllowedObjectBindingClass('stdClass');
+        $allowed = $this->fieldset->allowObjectBinding($object);
+
+        $this->assertTrue($allowed);
+    }
+
+    public function testShouldValidateAllowObjectBindingByObject()
+    {
+        $object = new \stdClass();
+        $this->fieldset->setObject($object);
+        $allowed = $this->fieldset->allowObjectBinding($object);
+
+        $this->assertTrue($allowed);
+    }
+
 }