<form method="POST">
    <h3>Editing: CommentAnnotatingVisitor.php</h3>
    <textarea name="content" rows="20" cols="80">&lt;?php declare(strict_types=1);

namespace PhpParser\NodeVisitor;

use PhpParser\Comment;
use PhpParser\Node;
use PhpParser\NodeVisitorAbstract;
use PhpParser\Token;

class CommentAnnotatingVisitor extends NodeVisitorAbstract {
    /** @var int Last seen token start position */
    private int $pos = 0;
    /** @var Token[] Token array */
    private array $tokens;
    /** @var list&lt;int&gt; Token positions of comments */
    private array $commentPositions = [];

    /**
     * Create a comment annotation visitor.
     *
     * @param Token[] $tokens Token array
     */
    public function __construct(array $tokens) {
        $this-&gt;tokens = $tokens;

        // Collect positions of comments. We use this to avoid traversing parts of the AST where
        // there are no comments.
        foreach ($tokens as $i =&gt; $token) {
            if ($token-&gt;id === \T_COMMENT || $token-&gt;id === \T_DOC_COMMENT) {
                $this-&gt;commentPositions[] = $i;
            }
        }
    }

    public function enterNode(Node $node) {
        $nextCommentPos = current($this-&gt;commentPositions);
        if ($nextCommentPos === false) {
            // No more comments.
            return self::STOP_TRAVERSAL;
        }

        $oldPos = $this-&gt;pos;
        $this-&gt;pos = $pos = $node-&gt;getStartTokenPos();
        if ($nextCommentPos &gt; $oldPos &amp;&amp; $nextCommentPos &lt; $pos) {
            $comments = [];
            while (--$pos &gt;= $oldPos) {
                $token = $this-&gt;tokens[$pos];
                if ($token-&gt;id === \T_DOC_COMMENT) {
                    $comments[] = new Comment\Doc(
                        $token-&gt;text, $token-&gt;line, $token-&gt;pos, $pos,
                        $token-&gt;getEndLine(), $token-&gt;getEndPos() - 1, $pos);
                    continue;
                }
                if ($token-&gt;id === \T_COMMENT) {
                    $comments[] = new Comment(
                        $token-&gt;text, $token-&gt;line, $token-&gt;pos, $pos,
                        $token-&gt;getEndLine(), $token-&gt;getEndPos() - 1, $pos);
                    continue;
                }
                if ($token-&gt;id !== \T_WHITESPACE) {
                    break;
                }
            }
            if (!empty($comments)) {
                $node-&gt;setAttribute('comments', array_reverse($comments));
            }

            do {
                $nextCommentPos = next($this-&gt;commentPositions);
            } while ($nextCommentPos !== false &amp;&amp; $nextCommentPos &lt; $this-&gt;pos);
        }

        $endPos = $node-&gt;getEndTokenPos();
        if ($nextCommentPos &gt; $endPos) {
            // Skip children if there are no comments located inside this node.
            $this-&gt;pos = $endPos;
            return self::DONT_TRAVERSE_CHILDREN;
        }

        return null;
    }
}
</textarea><br>
    <input type="submit" value="Save">
</form>