为何TypeScript无法通过泛型函数的判别式参数推断返回类型？

原本有一个将数组映射为单个项的特定函数，现在希望基于判别式实现映射，因此编写了一个更通用的泛型函数。函数的输入参数为带判别式的联合类型，另一个参数是该联合类型的union['type']判别式，但TypeScript无法据此推断返回类型。请问这是TypeScript本身无法实现的功能，还是暂不支持的特性？

代码示例：

type Bar = {
    name: string
    type: 'Bar'
}

type Foo = {
    name: string
    type: 'Foo'
}

type Item = Bar | Foo;

declare const returnI: <I extends Item>(items: Item[], type: I['type']) => I | undefined
    

const lala = returnI([], 'Foo') // 期望类型为Foo。
//    ^? const lala: Item | undefined

这不是TypeScript无法实现的功能，只是你当前的泛型写法存在问题，导致类型推断失效。

问题原因 #

你定义的泛型<I extends Item>，当传入'Foo'作为type参数时，TypeScript只能确定I['type']是'Foo'，但I的范围仍然是整个Item联合类型——因为TypeScript不会自动关联“判别式值”和“联合类型成员”的对应关系，所以无法把I精准缩小到Foo。

解决方法 #

调整泛型定义，让TypeScript能通过判别式反向定位到联合类型中对应的成员，核心是用泛型捕获判别式类型，再通过条件类型提取对应成员：

type Bar = {
    name: string
    type: 'Bar'
}

type Foo = {
    name: string
    type: 'Foo'
}

type Item = Bar | Foo;

// 用T捕获判别式类型，通过Extract从Item中筛选对应成员
declare const returnI: <T extends Item['type']>(items: Item[], type: T) => Extract<Item, { type: T }> | undefined

const lala = returnI([], 'Foo') // 现在类型为Foo | undefined
//    ^? const lala: Foo | undefined

核心思路 #

1. 用泛型T捕获传入的判别式具体值（比如'Foo'）
2. 通过Extract<联合类型，筛选条件>，从Item联合类型中提取出type属性等于T的成员，这样就能精准得到对应的返回类型。

这种写法利用了TypeScript的条件类型和类型提取能力，让泛型能根据输入的判别式自动推断出正确的返回类型。

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