TypeScript自定义转换器实现：支持下划线简写箭头函数语法

架构老林

2026-5-26

这确实是个能大幅简化数组操作代码的好想法！用_作为隐式箭头函数参数，替代重复写(x) =>，能让链式调用清爽太多。我帮你梳理下两种可行方案的具体思路：

方案一：SweetJS 宏实现

SweetJS本身就是用来扩展JavaScript/TypeScript语法的工具，很适合做这种语法糖转换。你需要定义宏来识别_的不同使用场景，然后替换成标准箭头函数：

首先安装依赖：npm install sweet.js
编写宏文件（比如underscore-syntax.sjs）：
这个宏需要匹配三种常见场景：
1. 单表达式（比如_ % 2）
2. 属性访问（比如_.name）
3. 方法调用（比如_.toUpperCase()）
示例宏代码大概是这样：
```
macro _ {
  rule { $expr:expr } => { (x) => $expr }
  rule { . $prop:ident } => { (x) => x.$prop }
  rule { . $prop:ident ( $args... ) } => { (x) => x.$prop($args...) }
}
```
使用时，把你的TypeScript文件用SweetJS编译：sjs input.ts -o output.ts，编译后的代码就会把_.name转换成(x) => x.name，_ % 2转换成(x) => x % 2。

方案二：TypeScript 自定义转换器

如果想更深度集成TypeScript生态，自定义转换器（基于TypeScript Compiler API）是更合适的选择，它能直接在TS编译阶段完成转换，不需要额外的编译步骤：

首先搭建转换器环境，需要依赖typescript和相关类型包：
```
npm install typescript @types/estree --save-dev
```

编写转换器逻辑（比如underscore-transformer.ts）：
核心思路是遍历AST，找到数组方法（map/filter等）的回调参数，当参数是包含_标识符的表达式时，将其替换为箭头函数：

识别CallExpression（比如array.map(_.name)）
检查第二个参数（回调）是否是引用_的表达式
创建一个新的箭头函数节点，把原表达式里的_替换成新的参数（比如x）
替换原回调节点为新的箭头函数

示例核心代码片段：

import * as ts from 'typescript';

function transformNode(node: ts.Node): ts.Node {
  if (ts.isCallExpression(node) && ts.isPropertyAccessExpression(node.expression)) {
    // 匹配array.map(...)这类调用
    const methodName = node.expression.name.text;
    if (['map', 'filter', 'reduce', 'forEach'].includes(methodName)) {
      const callbackArg = node.arguments[0];
      if (callbackArg && isUsingUnderscore(callbackArg)) {
        // 创建箭头函数参数
        const param = ts.factory.createParameterDeclaration(
          undefined, undefined, ts.factory.createIdentifier('x')
        );
        // 替换表达式里的_为x
        const replacedExpr = replaceUnderscoreWithParam(callbackArg, 'x');
        // 创建箭头函数
        const arrowFunc = ts.factory.createArrowFunction(
          undefined, undefined, [param], undefined, ts.factory.createToken(ts.SyntaxKind.EqualsGreaterThanToken),
          replacedExpr
        );
        return ts.factory.updateCallExpression(node, node.expression, node.typeArguments, [arrowFunc, ...node.arguments.slice(1)]);
      }
    }
  }
  return ts.visitEachChild(node, transformNode, undefined);
}

// 辅助函数：检查节点是否使用了_
function isUsingUnderscore(node: ts.Node): boolean {
  let hasUnderscore = false;
  ts.visitNode(node, (n) => {
    if (ts.isIdentifier(n) && n.text === '_') hasUnderscore = true;
    return n;
  });
  return hasUnderscore;
}

// 辅助函数：把节点里的_替换成指定参数名
function replaceUnderscoreWithParam(node: ts.Node, paramName: string): ts.Node {
  return ts.visitNode(node, (n) => {
    if (ts.isIdentifier(n) && n.text === '_') {
      return ts.factory.createIdentifier(paramName);
    }
    return n;
  });
}

配置TypeScript使用转换器：
你需要用ttypescript（支持自定义转换器的TS分支）或者ts-patch来启用插件。比如在tsconfig.json里配置：
```
{
  "compilerOptions": {
    "plugins": [
      { "transform": "./underscore-transformer.ts" }
    ]
  }
}
```
然后用ttsc编译，就能得到转换后的JS代码，同时TypeScript也能正确识别类型。

一些注意事项

要避免和用户自己定义的_变量冲突，最好在转换器/宏里限定场景：只处理数组方法的回调参数中的_，而不是全局所有_。
对于复杂嵌套场景（比如_.nested.prop），转换器的替换逻辑需要递归处理，确保所有_都被正确替换。
如果用SweetJS，要注意和TypeScript类型检查的兼容性，可能需要先转TS再做类型检查，或者用SweetJS的TS支持。

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