javascript - codemod/transform 包括 BlockStatement

Question

我正在执行 codemod/transform 以更改代码中的 if/return 语句。

我有很多if(err) do something，我需要重构那个行为。

我如何为此进行转换？

我有什么:

if (err) {
  return connection.rollback(function() {
    throw err;
  });
}

我想要的:

if (err){
    return rollback(connection, err);
}

到目前为止，我设法替换了 node.consequent 并直接使用了 callExpression:

root.find(j.IfStatement).replaceWith(function (_if) {
  var fnCall = j.callExpression(j.identifier('rollback'), [
    j.identifier('connection'), 
    j.identifier('err')
  ]);

  _if.node.consequent = fnCall; 
  return _if.node;
});

...导致:

if (err)
  rollback(connection, err);

如何在其中包含 BlockStatement 和 return？这是执行此操作的正确方法吗？codemod？

实例here

Answer 1

好的，成功了!多么好的工具!

please do comment or post a new answer if there is a better way to do this!

因此，我缺少的是 if 语句中的 block 语句 {}，以及其中的 return。

所以我补充说:

var ret = j.returnStatement(fnCall);
var block = j.blockStatement([ret]);
_if.node.consequent = block;

结果在这里:http://astexplorer.net/#/84e8ZqEAwQ/1