Let the Graph exporter work with Node instead of ValueNode.

Author: drupol

src/Exporter/Ascii.php

@@ -4,7 +4,7 @@
 
 namespace drupol\phptree\Exporter;
 
-use drupol\phptree\Node\ValueNodeInterface;
+use drupol\phptree\Node\NodeInterface;
 
 /**
  * Class Ascii
@@ -14,7 +14,7 @@ class Ascii implements ExporterInterface
     /**
      * {@inheritdoc}
      */
-    public function export(ValueNodeInterface $node): string
+    public function export(NodeInterface $node): string
     {
         $tree = new \RecursiveTreeIterator(
             new \RecursiveArrayIterator(
@@ -51,7 +51,7 @@ public function export(ValueNodeInterface $node): string
      * @return array
      *   The tree exported into an array.
      */
-    private function doExportAsArray(ValueNodeInterface $node): array
+    private function doExportAsArray(NodeInterface $node): array
     {
         $children = [];
         /** @var ValueNodeInterface $child */

src/Exporter/ExporterInterface.php

@@ -4,7 +4,7 @@
 
 namespace drupol\phptree\Exporter;
 
-use drupol\phptree\Node\ValueNodeInterface;
+use drupol\phptree\Node\NodeInterface;
 
 /**
  * Interface ExporterInterface
@@ -14,11 +14,11 @@ interface ExporterInterface
     /**
      * Export a node into something.
      *
-     * @param \drupol\phptree\Node\ValueNodeInterface $node
+     * @param NodeInterface $node
      *   The node.
      *
      * @return mixed
      *   The node exported.
      */
-    public function export(ValueNodeInterface $node);
+    public function export(NodeInterface $node);
 }

src/Exporter/Graph.php

@@ -61,7 +61,7 @@ public function getTraverser(): TraverserInterface
     /**
      * {@inheritdoc}
      */
-    public function export(ValueNodeInterface $node): OriginalGraph
+    public function export(NodeInterface $node): OriginalGraph
     {
         foreach ($this->getTraverser()->traverse($node) as $node_visited) {
             /** @var int $vertexId */
@@ -95,9 +95,18 @@ protected function createVertex(NodeInterface $node): Vertex
         /** @var int $vertexId */
         $vertexId = $this->createVertexId($node);
 
-        return false === $this->getGraph()->hasVertex($vertexId) ?
-            $this->getGraph()->createVertex($vertexId):
-            $vertex = $this->getGraph()->getVertex($vertexId);
+        if (false === $this->getGraph()->hasVertex($vertexId)) {
+            $vertex = $this->getGraph()->createVertex($vertexId);
+
+            $label = NULL;
+            if (method_exists($node, 'getValue')) {
+                $label = $node->getValue();
+            }
+
+            $vertex->setAttribute('graphviz.label', $label);
+        }
+
+        return $this->getGraph()->getVertex($vertexId);
     }
 
     /**

src/Exporter/SimpleArray.php

@@ -4,7 +4,7 @@
 
 namespace drupol\phptree\Exporter;
 
-use drupol\phptree\Node\ValueNodeInterface;
+use drupol\phptree\Node\NodeInterface;
 
 /**
  * Class SimpleArray
@@ -14,7 +14,7 @@ class SimpleArray implements ExporterInterface
     /**
      * {@inheritdoc}
      */
-    public function export(ValueNodeInterface $node)
+    public function export(NodeInterface $node)
     {
         $children = [];
         /** @var ValueNodeInterface $child */

src/Exporter/Text.php

@@ -4,7 +4,7 @@
 
 namespace drupol\phptree\Exporter;
 
-use drupol\phptree\Node\ValueNodeInterface;
+use drupol\phptree\Node\NodeInterface;
 
 /**
  * Class Text
@@ -14,7 +14,7 @@ class Text implements ExporterInterface
     /**
      * {@inheritdoc}
      */
-    public function export(ValueNodeInterface $node): string
+    public function export(NodeInterface $node): string
     {
         $children = [];
         /** @var ValueNodeInterface $child */