如何在TypeScript函数中强制要求使用对象解构参数？

要实现类似Playwright test.beforeEach 的强制对象解构校验，核心是通过**解析函数的抽象语法树（AST）**来判断参数是否符合解构要求。以下是具体实现步骤：

1. 安装依赖 #

需要用到AST解析相关的库，这里选择Babel工具链：

npm install @babel/parser @babel/traverse @babel/types

2. 编写参数解构检测函数 #

这个函数接收回调函数，解析它的源码AST，判断第一个参数是否为对象解构模式：

const parser = require('@babel/parser');
const traverse = require('@babel/traverse').default;
const t = require('@babel/types');

function isFirstArgObjectDestructuring(callback) {
  // 获取函数的源码字符串
  const funcCode = callback.toString();
  
  try {
    // 解析为AST
    const ast = parser.parse(funcCode, { sourceType: 'module' });
    
    let isValid = false;
    // 遍历AST，定位函数定义节点
    traverse(ast, {
      FunctionDeclaration(path) {
        checkParams(path.node.params);
      },
      FunctionExpression(path) {
        checkParams(path.node.params);
      },
      ArrowFunctionExpression(path) {
        checkParams(path.node.params);
      }
    });
    
    function checkParams(params) {
      if (params.length === 0) return;
      // 判定第一个参数是否为对象解构节点
      const firstParam = params[0];
      isValid = t.isObjectPattern(firstParam);
    }
    
    return isValid;
  } catch (err) {
    // 解析失败默认判定不合法
    return false;
  }
}

3. 封装高阶函数实现强制校验 #

基于检测函数，封装高阶函数包裹目标函数，添加校验逻辑：

function createEnforcedDestructuringFn(originalFn) {
  return function(callback, ...rest) {
    // 校验回调函数的参数格式
    if (!isFirstArgObjectDestructuring(callback)) {
      throw new Error('First argument must use the object destructuring pattern: args');
    }
    // 校验通过后调用原函数
    return originalFn(callback, ...rest);
  };
}

4. 示例使用 #

模拟Playwright的test.beforeEach场景：

// 模拟原有beforeEach逻辑
function originalBeforeEach(callback) {
  console.log('执行beforeEach核心逻辑');
  callback({ page: {} }, { testName: 'demo-test' });
}

// 为原函数添加强制解构校验
const enforcedBeforeEach = createEnforcedDestructuringFn(originalBeforeEach);

// 合法调用（使用对象解构）
enforcedBeforeEach(({ page }, testInfo) => {
  console.log('合法调用，page实例:', page);
});

// 非法调用（未使用对象解构）
try {
  enforcedBeforeEach((args, testInfo) => {
    console.log('非法调用');
  });
} catch (err) {
  console.error(err.message); // 输出: First argument must use the object destructuring pattern: args
}

原理说明 #

Playwright内部也是通过AST解析实现该校验：

• 先获取回调函数的源码字符串
• 解析为AST后，检查第一个参数的节点类型是否为ObjectPattern
• 不符合则抛出指定错误

这种方式比正则匹配更可靠，能准确识别箭头函数、普通函数、带默认值的解构等多种函数定义形式。

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