Now NodeInterface extends from IteratorAggregate and Traversable.

loophp · Jun 5, 2019 · 170ed5f · 170ed5f
1 parent 0a4587a
commit 170ed5f
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/src/Node/Node.php b/src/Node/Node.php
@@ -94,6 +94,14 @@ public function getAncestors(): \Traversable
         }
     }
 
+    /**
+     * {@inheritdoc}
+     */
+    public function getIterator()
+    {
+        yield from $this->children();
+    }
+
     /**
      * {@inheritdoc}
      */
@@ -194,10 +202,10 @@ public function offsetUnset($offset)
      */
     public function remove(NodeInterface ...$nodes): NodeInterface
     {
-        $this->storage['children'] = new \ArrayObject(
+        $this->storage['children']->exchangeArray(
             \array_filter(
-                (array) $this->storage['children'],
-                function ($child) use ($nodes) {
+                $this->storage['children']->getArrayCopy(),
+                static function ($child) use ($nodes) {
                     return !\in_array($child, $nodes, true);
                 }
             )

diff --git a/src/Node/NodeInterface.php b/src/Node/NodeInterface.php
@@ -7,7 +7,7 @@
 /**
  * Interface NodeInterface.
  */
-interface NodeInterface extends \Countable, \ArrayAccess
+interface NodeInterface extends \Countable, \ArrayAccess, \Traversable, \IteratorAggregate
 {
     /**
      * The node to add.