如何使用language-javascript模块遍历AST？

const parser = require('@babel/parser');
const traverse = require('@babel/traverse').default;
const babel = require('@babel/core');

// 原始代码
const code = `
function add(a, b) {
  return a + b;
}
`;

// 解析成AST
const ast = parser.parse(code, { sourceType: 'module' });

// 遍历AST并修改
traverse(ast, {
  FunctionDeclaration(path) {
    // 修改函数名
    path.node.id.name = 'sum';
    // 修改函数体
    path.node.body.body[0].returnStatement.argument = 'a - b';
  }
});

// 将修改后的AST转换回代码
const { code: modifiedCode } = babel.transformFromAst(ast, code, {
  presets: ['@babel/preset-env']
});

console.log(modifiedCode);