Question

我正在使用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())。我不明白为什么它在我的源代码上包含一个括号。我的源代码有什么问题吗？

Answer 1

您生成的AST不正确。 ExpressionStatement的格式为{type: "ExpressionStatement", expression... }。

您正在修改ExpressionStatement，并附加arguments和callee，并且您正在更改其type（至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();      
        }
    },
});

我希望你觉得这很有用。

为什么escodegen和esprima在我的源代码上生成一个括号包装？

1 个答案: