TypeScript泛化类型安全函数返回类型错误推断：预期对象却得到数组的问题排查

你的泛化版本返回类型被推断成数组，核心问题其实出在**keyof Spec的用法上**。咱们来拆解一下：当你把Spec定义为string[]类型时，keyof string[]拿到的并不是数组里的元素（比如"horizontal"、"vertical"），而是数组本身的固有键——像number（索引）、"length"、"push"这些属性名。所以用[key in keyof Spec]做映射类型时，TypeScript自然会生成一个类似数组的类型，而不是你想要的自定义字段对象。

下面是具体的解决步骤和修正后的代码：

1. 让Spec推断为字面量类型数组 #

首先要让TypeScript识别出你传入的数组是具体的字符串字面量集合，而不是宽泛的string[]。我们可以把Spec的泛型约束改成Spec extends readonly string[]，同时在调用函数时给数组加上as const（让TypeScript锁定数组元素的字面量类型）。

2. 用数组元素类型作为映射键 #

正确的映射应该基于数组里的元素类型，也就是Spec[number]——它表示数组中所有字符串字面量的联合类型。替换掉原来的keyof Spec，改用[K in Spec[number]]来定义返回对象的键。

3. 调整返回类型的条件逻辑 #

针对每个自定义字段K，我们需要分别判断：

• 如果传入的prop是原始类型，那么所有字段都返回该原始值；
• 如果prop是对象类型，提取对应字段K的类型；
• 以上都不满足时，使用默认值。

修正后的完整代码 #

export type Primitive = string | number | boolean | null | undefined;
export type PossiblySpecific<T, Spec extends string> = T | { [key in Spec]?: T };

const parsePossiblySpecific = <
  Spec extends readonly string[],
  Prop extends PossiblySpecific<Primitive, Spec[number]>,
  Default extends NonNullable<Primitive> | undefined = undefined
>(
  prop: Prop,
  fields: Spec,
  defaultValue?: Default
): {
  [K in Spec[number]]: 
    Prop extends Primitive 
      ? Prop 
      : Prop extends { [key in K]?: infer V } 
        ? V extends Primitive ? V : Default 
        : Default;
} => {
  // 示例实现逻辑（可根据需求调整）
  const result = {} as Record<Spec[number], Primitive | Default>;
  for (const field of fields) {
    if (typeof prop === 'object' && prop !== null) {
      result[field] = (prop as Record<string, any>)[field] ?? defaultValue;
    } else {
      result[field] = prop ?? defaultValue;
    }
  }
  return result as any;
};

// 测试类型推断是否符合预期
const c1 = parsePossiblySpecific(10, ["horizontal", "vertical"] as const); 
// => { horizontal: 10; vertical: 10; } ✅

const c2 = parsePossiblySpecific({ horizontal: 5 } as const, ["horizontal", "vertical"] as const, 10);
// => { horizontal: 5; vertical: 10; } ✅

const c3 = parsePossiblySpecific(10, ["x", "y"] as const, 30);
// => { x: 10; y: 10; } ✅

关键细节说明 #

• readonly string[]约束：确保TypeScript将传入的数组推断为包含具体字符串字面量的只读数组，比如["horizontal", "vertical"] as const会被推断为readonly ["horizontal", "vertical"]，此时Spec[number]就是"horizontal" | "vertical"。
• Spec[number]映射键：直接使用数组元素的联合类型作为对象的键，这样就能生成你期望的以自定义字段为键的返回类型。
• 条件类型优化：针对每个字段单独判断，确保即使对象只传入部分字段，类型推断也能准确提取对应值的类型，缺失的字段则使用默认值。

这样修改后，TypeScript就能正确推断出返回类型为预期的对象，而不是数组了。

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