php - 使用 nikic/PHP-Parser 将节点插入 PHP AST

Question

我正在使用https://github.com/nikic/PHP-Parser。想要在 AST 中插入节点时，什么是好的策略？使用遍历器，我可以使用NodeTraverser类轻松更新和删除节点。但是我怎样才能“插入之前”或“插入之后”一个节点？

示例：当遍历 AST 命名空间时，我想Use在第一个非使用语句之前插入一个语句。

我开始使用beforeTraverse并afterTraverse查找数组的索引，但它似乎过于复杂。有任何想法吗？

Answer 1

可以用多个节点替换一个节点。这仅适用于 leaveNode 并且仅当父结构是数组时。

public function leaveNode(Node $node) {
    if ($node instanceof Node\Stmt\Return_ && $node->expr !== null) {
        // Convert "return foo();" into "$retval = foo(); return $retval;"
        $var = new Node\Expr\Variable('retval');
        return [
            new Node\Stmt\Expression(new Node\Expr\Assign($var, $node->expr)),
            new Node\Stmt\Return_($var),
        ];
    }
}

请参阅修改 AST 的最后一节