37 lines
708 B
PHP
37 lines
708 B
PHP
<?php
|
|
declare(strict_types=1);
|
|
|
|
namespace StubTests\Parsers\Visitors;
|
|
|
|
use PhpParser\Node;
|
|
use PhpParser\NodeVisitorAbstract;
|
|
|
|
/**
|
|
* The visitor is required to provide "parent" attribute to nodes
|
|
*/
|
|
class ParentConnector extends NodeVisitorAbstract
|
|
{
|
|
/**
|
|
* @var Node[]
|
|
*/
|
|
private array $stack;
|
|
|
|
public function beforeTraverse(array $nodes)
|
|
{
|
|
$this->stack = [];
|
|
}
|
|
|
|
public function enterNode(Node $node)
|
|
{
|
|
if (!empty($this->stack)) {
|
|
$node->setAttribute('parent', $this->stack[count($this->stack) - 1]);
|
|
}
|
|
$this->stack[] = $node;
|
|
}
|
|
|
|
public function leaveNode(Node $node)
|
|
{
|
|
array_pop($this->stack);
|
|
}
|
|
}
|