TypeScript：如何结合参数可选约束并保留第一个参数返回类型

需求如下：

• 函数参数需符合指定的类型数组约束，每个参数允许传入对应类型或undefined
• 函数返回类型必须是调用时第一个参数的实际类型（例如传入string则返回string，传入undefined则返回undefined）
• 此前尝试的Attempt类型导致返回类型丢失为any，需要修复可行方案

解决方案 #

核心思路是通过泛型捕获第一个参数的实际类型，同时约束整个参数列表符合指定的可选类型规则，避免类型推断丢失。

实现代码 #

// 保留原有的可选参数转换类型
export type FnWithOptions<A extends any[]> = { [I in keyof A]: A[I] | undefined };

// 目标函数类型定义
type TargetFn<Params extends any[]> = 
  <FirstParam extends Params[0] | undefined>(
    ...args: [FirstParam, ...FnWithOptions<Params extends [any, ...infer Rest] ? Rest : []>]
  ) => FirstParam;

代码解释 #

• FirstParam泛型用于捕获调用时第一个参数的实际类型，同时限制它必须是Params[0] | undefined，保证参数类型符合要求
• 剩余参数通过infer Rest提取Params中除第一个参数外的部分，再用FnWithOptions转换为可选类型，确保后续参数的约束
• 返回类型直接绑定到FirstParam，完整保留调用时第一个参数的具体类型

测试验证 #

const targetFn: TargetFn<[first: string, second: number]> = (first, second) => first;

const expectedString = targetFn("str", 1); // 类型为 string
const expectedUndefined = targetFn(undefined, 1); // 类型为 undefined
// @ts-expect-error 第一个参数应为 string 或 undefined，传入 number 报错
targetFn(1, 1);
// @ts-expect-error 第二个参数应为 number 或 undefined，传入 string 报错
targetFn("str", "1");

原尝试失败原因 #

之前的Attempt类型中，Args extends any[]没有和Params做强约束，TypeScript无法从宽泛的any[]中推断出Args[0]的具体类型，最终退化为any。而新的类型通过拆分第一个参数为独立泛型，并约束整个参数列表的结构，既保证了参数类型合规，又保留了返回类型的准确性。

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