利用ast修改js，并保存硬盘

案例js文件

1. 文件路径：src/router/a.js
2. 文件内容：

export default {
  fixed: [],
  fluid: []
}

js解析器

1. 使用@babel/core 把js 解析为jsonTree

	const babelPath = '@babel/core'
	const babel = require(babelPath);
	/**
	 * @code 字符串code
	 * @description 生成ast
	 */
	function generatingAst (code) {
	  const ast =  babel.parse(code);
	  return ast
	}
	// 读取js 文件
	const jsPath = 'src/router/a.js'
	fs.readFile(jsPath, 'utf8', function(err, data) {
	  if (err) {
	      console.error(err);
	      return;
	  } 
     const ast = generatingAst(data)
  });

2. 循环ast

/**
 * @ast ast json
 * @description 循环ast对象,修改ast
 */
 // 添加ast循环包
const traversePath = '@babel/traverse'
const traverse = require(traversePath).default;
function estraverseAst (ast) {
  const templateAST = generatingTemplateAst('{a:1,()=>{}}');
  traverse(ast, {
    enter(path) {
      // 向 fixed 数组添加节点
      if (path.node.type === 'ObjectProperty' && path.node.key.name === 'fixed') {
        path.node.value.elements.push(templateAST, templateAST);
      }
      // // 向 fluid 数组添加节点
      if (path.node.type === 'ObjectProperty' && path.node.key.name === 'fluid') {
        path.node.value.elements.push(templateAST);
      }
    }
  });
}

3. 使用修改后的ast 生成code

/**
 * 
 * @param {*} ast 
 * @description 根据ast 生成code
 * @returns 
 */
// 添加生成包
const generatePath = '@babel/generator'
const generate = require(generatePath).default;
function generatingCode (ast) {
  return generate(ast)
}

4. 为了快速插入ast，创建ast模版

/**
 * 
 * @param {*} type 模版类型
 * @param {*} title 模版数据
 * @returns 
 * @description 快速生成模版ast
 */
function generatingTemplateAst (code) {
  const ast = generatingAst(`[${code}]`).program.body[0].expression?.elements[0];
  return ast
}

5. 如何格式不对，可进一步处理格式

// 添加美化包
const prettier = require("prettier");
// 把生成后的code字符串传给prettier（属性自查）
const formattedCode = prettier.format(code, { 
  parser: "babel",
  trailingComma: "es5",
  tabWidth: 2,
  semi: false, 
  printWidth: 300,
  singleQuote: true,
  bracketSpacing: false,
  trailingComma: 'none'
});

6. 写入修改后的文件

// 添加fs node.js 的包
const fs = require('fs')
fs.writeFile('src/router/a.js', formattedCode, (err) => {
   if (err) throw err;
 });