如何实现TypeScript元组与Partial元组的差集类型？

纠正原假设错误 #

你之前认为Partial<T>可以表示元组T的前n个元素组成的元组，这是错误的：

• Partial<T>作用于元组时，会将元组的每个元素变为可选且允许为undefined，比如Partial<[1,2,3]>的实际类型是[(1 | undefined)?, (2 | undefined)?, (3 | undefined)?]，和“取前n个元素的元组”完全不是一回事。
• 正确表示T的前n个元素组成的元组，应该约束T2是T的前缀元组，即T可以被拆分为[...T2, ...Rest]的形式，其中Rest是剩余元素组成的元组。

核心需求实现 #

我们需要实现一个类型，输入元组T和它的前缀元组T2，输出T中去掉T2后剩余的后缀元组T3，使得[...T2, ...T3]等于T。

实现剩余类型提取工具 #

定义递归类型RemovePrefix，用来移除元组T的前缀T2，返回剩余部分：

type RemovePrefix<T extends any[], P extends any[]> = 
  // 如果前缀P还有元素
  P extends [infer First, ...infer Rest] 
    ? // 检查T是否以First开头
      T extends [First, ...infer TRest] 
        ? // 递归处理剩余部分
          RemovePrefix<TRest, Rest> 
        : // 前缀不匹配则返回never
          never 
    : // 前缀匹配完，返回剩余的T
      T;

修正函数泛型约束与实现 #

调整原函数的泛型约束，确保T2是T的合法前缀，同时使用RemovePrefix作为返回函数的参数类型：

const f =
  <T extends any[]>() =>
  // 约束T2必须是T的前缀元组
  <T2 extends any[]>(t1: T extends [...T2, ...any[]] ? T2 : never) =>
  (t2: RemovePrefix<T, T2>): T =>
    // 断言为T，TypeScript无法自动推导拼接后的类型完全匹配
    [...t1, ...t2] as T;

测试示例 #

const f1 = f<[1, 2, 3, 4, 5]>()([1,2])
// f1的参数类型被正确推导为[3,4,5]
f1([3,4,5]) // 返回类型为[1,2,3,4,5]，符合预期

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