TypeScript无法推断扩展原始类型联合的字面量类型问题

首先展示定义的类型与函数：

interface ILabel<T> {
  readonly label: string;
  readonly key: T
}

interface IProps<T> {
  readonly labels: Array<ILabel<T>>;
  readonly defaultValue: T;
  readonly onChange: (state: ILabel<T>) => void;
}

const testFunc = <T extends string | number>(labels: IProps<T>) => labels

需求是让key支持混合类型（如字符串、数字、布尔值），希望TypeScript能从labels中推断出字面量联合类型，并应用到defaultValue、onChange等属性，避免重复编写类型守卫。

正常工作的场景 #

同一类型的key #

当所有key为同一类型时，函数能正确推断联合字面量类型：

testFunc({
  labels: [{
    label: 'whatever',
    key: 'a',
  }, {
    label: 'whatever',
    key: 'b',
  }, {
    label: 'whatever',
    key: 'c',
  }],
  defaultValue: 'a',
  onChange: (state) => {}
}) // 正确推断为IProps<'a' | 'b' | 'c'>

单一枚举类型 #

使用单一枚举时，类型推断也能正常工作：

enum TestEnum {
  ONE = 'one',
  TWO = 'two',
  THREE = 3,
}

enum TestEnum2 {
  FOUR = 'four',
  FIVE = 'five',
  SIX = 6,
}

testFunc({
  labels: [{
    label: 'whatever',
    key: TestEnum.ONE,
  }, {
    label: 'whatever',
    key: TestEnum.TWO,
  }, {
    label: 'whatever',
    key: TestEnum.THREE,
  }],
  defaultValue: TestEnum.TWO,
  onChange: (state) => {}
}) // 正确推断为IProps<TestEnum>

报错的场景 #

混合基础类型 #

混合字符串与数字类型的key时，TypeScript会将T推断为第一个元素的key类型，导致后续元素报错：

testFunc({
  labels: [{
    label: 'whatever',
    key: 'a',
  }, {
    label: 'whatever',
    key: 'b',
  }, {
    label: 'whatever',
    key: 2,
  }],
  defaultValue: 'c',
  onChange: (state) => {}
}) // 报错，推断为IProps<'a'>

错误信息：

Type '"b"' is not assignable to type '"a"'.ts(2322)
TabBar.types.ts(79, 12): The expected type comes from property 'key' which is declared here on type 'ILabel<"a">'

混合不同枚举 #

混合不同枚举的key时，同样会出现推断错误：

testFunc({
  labels: [{
    label: 'whatever',
    key: TestEnum.ONE,
  }, {
    label: 'whatever',
    key: TestEnum.TWO,
  }, {
    label: 'whatever',
    key: TestEnum2.FOUR,
  }],
  defaultValue: TestEnum.TWO,
  onChange: (state) => {}
}) // 报错，推断为IProps<TestEnum.ONE>

混合枚举与字面量 #

枚举与字面量混合时，推断逻辑同样失效：

testFunc({
  labels: [{
    label: 'whatever',
    key: TestEnum.ONE,
  }, {
    label: 'whatever',
    key: 'two',
  }],
  defaultValue: 'two',
  onChange: (state) => {}
}) // 报错，推断为IProps<TestEnum.ONE>

原因分析 #

1. 类型推断的优先策略
   TypeScript推断泛型类型时，会优先从第一个匹配元素确定初始类型候选，再尝试让后续元素适配该候选。当第一个元素的key是字面量或枚举成员时，TS会将这个具体类型作为T的初始推断值，后续元素类型不兼容时直接报错，不会自动扩展T为联合类型。

2. 数组的同构约束
   IProps<T>中的labels被定义为Array<ILabel<T>>，意味着数组所有元素的key必须是同一类型T。TS默认假设数组元素是同构的，混合不同类型时，不会主动将T推断为这些类型的联合，除非所有元素类型能被统一到更宽泛的联合类型，而字面量与枚举混合场景下，TS不会主动触发这种类型拓宽。

3. 枚举类型的特殊性
   枚举成员本身是特定类型（如TestEnum.ONE的类型是TestEnum.ONE，而非TestEnum），当第一个元素key是枚举成员时，TS会推断T为该具体成员类型，而非整个枚举类型。只有当所有元素key都属于同一枚举时，TS才会将T拓宽为枚举本身，因为此时所有成员都兼容于枚举类型。

4. 泛型约束的限制
   testFunc的泛型约束是T extends string | number，虽然联合类型满足该约束，但TS推断时不会主动将多个不同子类型合并为联合类型，除非有明确提示或结构支持这种推断。比如同一类型的字面量能推断联合，是因为它们都属于同一父类型的子类型，而跨类型混合时，TS不会自动合并为联合。

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