如何阻止Babel在修改AST后格式化生成的代码？

方法一：配置Babel生成器的preserveWhitespace选项 #

Babel的代码生成器默认会格式化输出，你可以通过开启preserveWhitespace参数来保留原代码的空白结构。

• 手动调用generate函数时传入配置：

const { parse, generate } = require('@babel/core');
const fs = require('fs');

// 读取原文件代码
const sourceCode = fs.readFileSync('./your-file.js', 'utf8');
// 解析成AST
const ast = parse(sourceCode);

// 执行你的AST修改逻辑（示例：添加一个console.log）
const consoleLog = require('@babel/types').expressionStatement(
  require('@babel/types').callExpression(
    require('@babel/types').identifier('console.log'),
    [require('@babel/types').stringLiteral('AST modified')]
  )
);
ast.program.body.push(consoleLog);

// 生成代码时开启preserveWhitespace
const { code: outputCode } = generate(ast, { preserveWhitespace: true }, sourceCode);
// 写回原文件
fs.writeFileSync('./your-file.js', outputCode);

• 如果你用babel.transformFromAstSync，同样可以传入该配置：

const result = babel.transformFromAstSync(ast, sourceCode, {
  generatorOpts: {
    preserveWhitespace: true
  }
});
fs.writeFileSync('./your-file.js', result.code);

方法二：使用recast库（更可靠的格式保留方案） #

如果preserveWhitespace无法满足需求（比如复杂注释、特殊缩进的保留），可以用recast——这个工具专门针对“修改AST但保留原代码格式”的场景设计，基于Babel生态但格式保留能力更强。

1. 先安装依赖：

npm install recast --save-dev

2. 示例代码：

const recast = require('recast');
const fs = require('fs');
const { types: t } = recast;

// 读取原文件并解析成带格式信息的AST
const sourceCode = fs.readFileSync('./your-file.js', 'utf8');
const ast = recast.parse(sourceCode);

// 执行AST修改操作（示例：修改某个变量的值）
const targetVar = ast.program.body.find(node => 
  node.type === 'VariableDeclaration' && 
  node.declarations[0].id.name === 'oldVar'
);
if (targetVar) {
  targetVar.declarations[0].init = t.numericLiteral(999);
}

// 生成代码并保留原格式
const outputCode = recast.print(ast).code;
fs.writeFileSync('./your-file.js', outputCode);

命令行工具场景的配置 #

如果你用babel命令行工具处理文件，需要在你的Babel配置文件（如babel.config.json或.babelrc）中添加生成器配置：

{
  "generatorOpts": {
    "preserveWhitespace": true
  }
}

内容的提问来源于stack exchange，提问作者wanannaturalcurl