如何使用Babel基于AST修改JS代码，实现Map参数键存在性校验插桩？

一、Babel AST插桩转换方案 #

你可以按以下步骤编写Babel插件逻辑：

1. 定位目标函数
   访问FunctionDeclaration、FunctionExpression等函数节点，先匹配参数中存在datamap的目标函数，再进入函数体节点遍历。
2. 语句清理
   遍历函数体的所有顶级语句、以及if块内的子语句：

• 仅保留包含datamap.has调用的IfStatement节点，其余的console.log、变量声明、else分支等非目标语句直接移除。

3. 收集datamap.get调用
   遍历所有CallExpression节点，判断调用对象是否为datamap.get，提取每个调用传入的键字面量，记录所有被访问的键名。
4. 生成存在性检查代码
   对每个收集到的键，在对应datamap.get调用的同级位置插入新的If语句：

if (!datamap.has("你的键名")) {
  console.log('你的键名 is not present in datamap')
}

5. 最后再做一次冗余节点清理，删除所有残留的非检查类语句即可。

二、纯AST静态检测方案 #

不需要插桩运行也可以实现缺失键检测，属于静态代码分析范畴，实现逻辑如下：

1. 遍历目标函数AST，分别收集两类信息：

• 所有datamap.get调用用到的键集合，同时记录每个get调用所在的作用域位置
• 所有datamap.has调用用到的键集合，同时记录每个has检查覆盖的作用域范围

2. 做作用域匹配校验：如果某个get调用的键，没有对应的has检查覆盖它所在的作用域，就判定为缺失存在性检查的键，直接输出即可。
   该方案的局限性是仅支持字面量形式的键，如果键是动态变量（比如datamap.get(varKey)），静态阶段无法获取具体键值，就需要依赖插桩后的运行时检测。

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