Fix PHPStan errors.

loophp · Dec 30, 2018 · f422c8b · f422c8b
1 parent db5f5a6
commit f422c8b
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/src/Exporter/Ascii.php b/src/Exporter/Ascii.php
@@ -5,6 +5,7 @@
 namespace drupol\phptree\Exporter;
 
 use drupol\phptree\Node\NodeInterface;
+use drupol\phptree\Node\ValueNodeInterface;
 
 /**
  * Class Ascii
@@ -48,11 +49,19 @@ public function export(NodeInterface $node): string
     /**
      * Export the tree in an array.
      *
+     * @param \drupol\phptree\Node\NodeInterface $node
+     *   The node.
+     *
      * @return array
      *   The tree exported into an array.
      */
     private function doExportAsArray(NodeInterface $node): array
     {
+        if (!($node instanceof ValueNodeInterface))
+        {
+            throw new \InvalidArgumentException('Must implements ValueNodeInterface');
+        }
+
         $children = [];
         /** @var ValueNodeInterface $child */
         foreach ($node->children() as $child) {

diff --git a/src/Exporter/SimpleArray.php b/src/Exporter/SimpleArray.php
@@ -5,6 +5,7 @@
 namespace drupol\phptree\Exporter;
 
 use drupol\phptree\Node\NodeInterface;
+use drupol\phptree\Node\ValueNodeInterface;
 
 /**
  * Class SimpleArray
@@ -16,6 +17,11 @@ class SimpleArray implements ExporterInterface
      */
     public function export(NodeInterface $node)
     {
+        if (!($node instanceof ValueNodeInterface))
+        {
+            throw new \InvalidArgumentException('Must implements ValueNodeInterface');
+        }
+
         $children = [];
         /** @var ValueNodeInterface $child */
         foreach ($node->children() as $child) {

diff --git a/src/Exporter/Text.php b/src/Exporter/Text.php
@@ -5,6 +5,7 @@
 namespace drupol\phptree\Exporter;
 
 use drupol\phptree\Node\NodeInterface;
+use drupol\phptree\Node\ValueNodeInterface;
 
 /**
  * Class Text
@@ -16,6 +17,11 @@ class Text implements ExporterInterface
      */
     public function export(NodeInterface $node): string
     {
+        if (!($node instanceof ValueNodeInterface))
+        {
+            throw new \InvalidArgumentException('Must implements ValueNodeInterface');
+        }
+
         $children = [];
         /** @var ValueNodeInterface $child */
         foreach ($node->children() as $child) {