ast demo

getPositionX2x.ts

import generate from "@babel/generator";
import {parse} from "@babel/parser";
import traverse from "@babel/traverse";
import {identifier, isIdentifier, isMemberExpression, memberExpression} from "@babel/types";
import * as fs from 'fs'
const file = process.argv[2]
const ast = parse(fs.readFileSync(file, 'utf-8’))  // 解析得到 ast
traverse(ast, {
  CallExpression(path) {                           // 当遍历到“调用表达式”时进入此回调
    if (isMemberExpression(path.node.callee) &&    // 被调用的函数来自一个成员表达
      isIdentifier(path.node.callee.property, {name: 'getPositionX'})) {  // 成员属性是名为 getPositionX 的标识符
      path.replaceInline(memberExpression(path.node.callee.object, identifier('x')))  // 把调用表达式替换成函数表达式，对象为原本的对象，属性直接用 x
    }
  }
})
console.log(generate(ast, {retainLines: true}).code);