rector/src/NodeVisitor/Traverse/NodeConnectorNodeVisitor.php

<?php declare(strict_types=1);

namespace Rector\NodeVisitor\Traverse;

use PhpParser\Node;
use PhpParser\NodeVisitorAbstract;

final class NodeConnectorNodeVisitor extends NodeVisitorAbstract
{
    /**
     * @var Node[]
     */
    private $stack = [];

    /**
     * @var Node
     */
    private $prev;

    public function beforeTraverse(array $nodes): void
    {
        $this->stack = [];
        $this->prev = null;
    }

    public function enterNode(Node $node): void
    {
        if (! empty($this->stack)) {
            $node->setAttribute('parent', $this->stack[count($this->stack)-1]);
        }

        if ($this->prev && $this->prev->getAttribute('parent') === $node->getAttribute('parent')) {
            $node->setAttribute('prev', $this->prev);
            $this->prev->setAttribute('next', $node);
        }

        $this->stack[] = $node;
    }

    public function leaveNode(Node $node): void
    {
        $this->prev = $node;
        array_pop($this->stack);
    }
}