如何通过AST模块识别并删除赋值操作与函数调用？

嘿，我来帮你搞定这个AST处理的问题！之前你已经搞定了大于号转小于等于的替换，现在要处理赋值和函数调用的删除，核心是找对AST里的对应节点类型，然后用遍历器的路径方法来修改或移除节点，而不是直接返回空值哦。

先搞清楚AST里的对应节点类型 #

首先得明确你要处理的操作在AST里对应的节点类型，这是精准处理的前提：

• 赋值操作：如果是单独的赋值语句（比如x = 10;），对应的是AssignmentExpression节点（被ExpressionStatement包裹）；如果是变量声明时的赋值（比如let x = 10;），则是VariableDeclarator节点里的init属性。
• 函数调用：所有带()的函数调用（不管是foo()还是bar(1, 'test')），对应的都是CallExpression节点。

处理赋值操作的删除逻辑 #

分两种常见场景来处理：

场景1：删掉整行独立的赋值语句 #

比如你想把x = 10;这种单独的赋值行直接删掉，那我们可以遍历AST，找到包裹着AssignmentExpression的ExpressionStatement，然后直接移除这个语句节点。

拿Babel工具链举例子（这是前端最常用的AST处理工具）：

const { parse } = require('@babel/parser');
const traverse = require('@babel/traverse').default;
const generate = require('@babel/generator').default;

// 原始代码
const code = `
let a = 5;
x = 10;
console.log(x);
`;

// 解析成AST
const ast = parse(code);

// 遍历并处理节点
traverse(ast, {
  ExpressionStatement(path) {
    // 判断当前语句是不是赋值表达式
    if (path.node.expression.type === 'AssignmentExpression') {
      path.remove(); // 直接移除整个语句
    }
  }
});

// 生成处理后的代码
const result = generate(ast).code;
console.log(result);

处理前：

let a = 5;
x = 10;
console.log(x);

处理后：

let a = 5;
console.log(x);

场景2：保留变量声明但删掉赋值部分 #

如果是想把let a = 5;变成let a;，也就是保留变量声明但去掉赋值的内容，那我们要处理VariableDeclarator节点，把它的init属性设为null就行：

traverse(ast, {
  VariableDeclarator(path) {
    // 清空赋值的初始值
    path.node.init = null;
  }
});

处理前：let a = 5;
处理后：let a;

处理函数调用的删除逻辑 #

函数调用的情况也分两种，看它是独立语句还是作为子表达式存在：

场景1：删掉整行独立的函数调用 #

比如foo();这种单独的函数调用行，和处理独立赋值的逻辑类似，找到包裹CallExpression的ExpressionStatement然后移除：

traverse(ast, {
  ExpressionStatement(path) {
    if (path.node.expression.type === 'CallExpression') {
      path.remove();
    }
  }
});

处理前：

foo();
console.log('hello');
bar(123);

处理后：

console.log('hello');

场景2：删掉作为子表达式的函数调用 #

比如const b = foo();这种，函数调用是赋值的右值，这时候我们可以找到CallExpression节点，把它替换成null（或者根据上下文换成你需要的占位符）：

traverse(ast, {
  CallExpression(path) {
    // 把函数调用替换成null，也可以换成undefined之类的
    path.replaceWithSourceString('null');
    // 如果想直接去掉赋值部分，结合前面VariableDeclarator的处理逻辑就行
  }
});

处理前：const b = foo();
处理后：const b = null;

关键提示 #

• 不是“返回空值”来删除节点，而是通过AST遍历器提供的路径（path）方法来操作——比如path.remove()移除节点，path.replaceWith()替换节点，或者直接修改节点的属性。
• 如果你用的是其他AST解析工具（比如Acorn），节点类型命名可能略有差异，但核心类型是一致的。你可以用AST可视化工具把你的代码贴进去，直观看到节点结构，找对要处理的节点会更轻松。

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