| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111 |
- <?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;
- use Symfony\Component\Console\Output\OutputInterface;
- /**
- * The TreeHelper class provides methods to display tree-like structures.
- *
- * @author Simon André <smn.andre@gmail.com>
- *
- * @implements \RecursiveIterator<int, TreeNode>
- */
- final class TreeHelper implements \RecursiveIterator
- {
- /**
- * @var \Iterator<int, TreeNode>
- */
- private \Iterator $children;
- private function __construct(
- private readonly OutputInterface $output,
- private readonly TreeNode $node,
- private readonly TreeStyle $style,
- ) {
- $this->children = new \IteratorIterator($this->node->getChildren());
- $this->children->rewind();
- }
- public static function createTree(OutputInterface $output, string|TreeNode|null $root = null, iterable $values = [], ?TreeStyle $style = null): self
- {
- $node = $root instanceof TreeNode ? $root : new TreeNode($root ?? '');
- return new self($output, TreeNode::fromValues($values, $node), $style ?? TreeStyle::default());
- }
- public function current(): TreeNode
- {
- return $this->children->current();
- }
- public function key(): int
- {
- return $this->children->key();
- }
- public function next(): void
- {
- $this->children->next();
- }
- public function rewind(): void
- {
- $this->children->rewind();
- }
- public function valid(): bool
- {
- return $this->children->valid();
- }
- public function hasChildren(): bool
- {
- if (null === $current = $this->current()) {
- return false;
- }
- foreach ($current->getChildren() as $child) {
- return true;
- }
- return false;
- }
- public function getChildren(): \RecursiveIterator
- {
- return new self($this->output, $this->current(), $this->style);
- }
- /**
- * Recursively renders the tree to the output, applying the tree style.
- */
- public function render(): void
- {
- $treeIterator = new \RecursiveTreeIterator($this);
- $this->style->applyPrefixes($treeIterator);
- $this->output->writeln($this->node->getValue());
- $visited = new \SplObjectStorage();
- foreach ($treeIterator as $node) {
- $currentNode = $node instanceof TreeNode ? $node : $treeIterator->getInnerIterator()->current();
- if (isset($visited[$currentNode])) {
- throw new \LogicException(\sprintf('Cycle detected at node: "%s".', $currentNode->getValue()));
- }
- $visited[$currentNode] = true;
- $this->output->writeln($node);
- }
- }
- }
|