initial commit

vendor/symfony/console/Helper/TreeNode.php

@@ -0,0 +1,105 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Component\Console\Helper;
+
+/**
+ * @implements \IteratorAggregate<TreeNode>
+ *
+ * @author Simon André <smn.andre@gmail.com>
+ */
+final class TreeNode implements \Countable, \IteratorAggregate
+{
+    /**
+     * @var array<TreeNode|callable(): \Generator>
+     */
+    private array $children = [];
+
+    public function __construct(
+        private readonly string $value = '',
+        iterable $children = [],
+    ) {
+        foreach ($children as $child) {
+            $this->addChild($child);
+        }
+    }
+
+    public static function fromValues(iterable $nodes, ?self $node = null): self
+    {
+        $node ??= new self();
+        foreach ($nodes as $key => $value) {
+            if (is_iterable($value)) {
+                $child = new self($key);
+                self::fromValues($value, $child);
+                $node->addChild($child);
+            } elseif ($value instanceof self) {
+                $node->addChild($value);
+            } else {
+                $node->addChild(new self($value));
+            }
+        }
+
+        return $node;
+    }
+
+    public function getValue(): string
+    {
+        return $this->value;
+    }
+
+    public function addChild(self|string|callable $node): self
+    {
+        if (\is_string($node)) {
+            $node = new self($node);
+        }
+
+        $this->children[] = $node;
+
+        return $this;
+    }
+
+    /**
+     * @return \Traversable<int, TreeNode>
+     */
+    public function getChildren(): \Traversable
+    {
+        foreach ($this->children as $child) {
+            if (\is_callable($child)) {
+                yield from $child();
+            } elseif ($child instanceof self) {
+                yield $child;
+            }
+        }
+    }
+
+    /**
+     * @return \Traversable<int, TreeNode>
+     */
+    public function getIterator(): \Traversable
+    {
+        return $this->getChildren();
+    }
+
+    public function count(): int
+    {
+        $count = 0;
+        foreach ($this->getChildren() as $child) {
+            ++$count;
+        }
+
+        return $count;
+    }
+
+    public function __toString(): string
+    {
+        return $this->value;
+    }
+}