3

我escodegen用来在我的语句中添加一个结束代码，如下所示。在 leave 方法中，我.toArray()在语句的末尾附加了一个调用。

const esprima = require('esprima');
const estraverse = require('estraverse');
const escodegen = require('escodegen');

const ast = esprima.parse('db.find()');
let finished = false;
estraverse.traverse(ast, {
  leave: (node, parent) => {
    if (node.type === esprima.Syntax.ExpressionStatement && !finished) {
      finished = true;
      let statement = escodegen.generate(node);
      statement = `${statement.substring(0, statement.lastIndexOf(';'))}.toArray()`;
      const findAst = esprima.parse(statement);
      node.arguments = findAst.body[0].expression.arguments;
      node.callee = findAst.body[0].expression.callee;
      node.type = findAst.body[0].expression.type;
    }
  },
});

const generated = escodegen.generate(ast);
console.log('generated  code:', generated);

上面代码的输出是：generated code: (db.find().toArray()). 我不明白为什么它在我的源代码上加上括号。我的源代码有什么问题吗？

4

1 回答 1

2

您正在生成不正确的 AST。一个ExpressionStatement有形式{type: "ExpressionStatement", expression... }。

您正在修改您的ExpressionStatement, 附加到它arguments，callee并且您正在更改它的type(to CallExpression)。这里：

  node.arguments = findAst.body[0].expression.arguments;
  node.callee = findAst.body[0].expression.callee;
  node.type = findAst.body[0].expression.type;

导致一个奇怪的 AST。

你可以简单地看到它：console.log('generated ast: %j', ast);

一个快速的解决方案是将提到的部件附加到它们所属的位置（属于expression）。结果：

let finished = false;
estraverse.traverse(ast, {
    leave: (node, parent) => {
        if (node.type === esprima.Syntax.ExpressionStatement && !finished) {
        finished = true;
        let statement = escodegen.generate(node);
        statement = `${statement.substring(0, statement.lastIndexOf(';'))}.toArray()`;
        console.log(statement);
        const findAst = esprima.parse(statement);
        node.expression.arguments = findAst.body[0].expression.arguments;
        node.expression.callee = findAst.body[0].expression.callee;
        node.expression.type = findAst.body[0].expression.type;
        }
    },
});

它将生成一个正确的 AST，输出预期的db.find().toArray();. 但我认为代码有点复杂，工作量太大，它会解析db.find()然后生成代码并再次解析。

此外，您可以返回this.break()以leave停止遍历。

以我的拙见，这将非常清楚：

var new_expr = {
        type: "CallExpression",
        callee: {
            type: "MemberExpression",
            computed: false,
            object: null,
            property: {
                type: "Identifier",
                name: "toArray"
            }
        },
        arguments: []
    };

const ast3 = esprima.parse('db.find()');
estraverse.traverse(ast3, {
    leave: function(node, parent) {
        if (node.type === esprima.Syntax.ExpressionStatement) {
            new_expr.callee.object = node.expression;
            node.expression = new_expr;
            return this.break();      
        }
    },
});

希望这个对你有帮助。

于 2017-09-06T16:51:44.260 回答