获取函数返回类型

约 297 字小于 1 分钟

2022-12-01

题目

不使用 ReturnType 实现 TypeScript 的 ReturnType<T> 泛型。

function fn(v: boolean) {
  if (v)
    return 1
  else return 2
}

type a = MyReturnType<typeof fn> // 应推导出 "1 | 2"

解题思路

使用条件类型中的类型推断，泛型参数 T 是否可以分配给 () => void。

通过类型参数从条件类型推断为真时，获取函数返回类型 () => infer R 。

我们并不关心函数是否带参数，需要带上参数 ...args: any[] 表示可以接受任何参数。

答案

type MyReturnType<T> = T extends (...args: any[]) => infer R ? R : never

验证

type cases
 = [
  Expect
<Equal
<string, MyReturnType
<() => string>>>,
  Expect
<Equal
<123, MyReturnType
<() => 123>>>,
  Expect
<Equal
<ComplexObject, MyReturnType
<() => ComplexObject>>>,
  Expect
<Equal
<Promise
<boolean>, MyReturnType
<() => Promise
<boolean>>>>,
  Expect
<Equal
<() => 'foo', MyReturnType
<() => () => 'foo'>>>,
  Expect
<Equal
<1 | 2, MyReturnType
<typeof fn
>>>,
  Expect
<Equal
<1 | 2, MyReturnType
<typeof fn1
>>>,
]

interface ComplexObject {
  a
: [12, 'foo']
  bar
: 'hello'
  prev
: () => number
}

const fn
 = (v
: boolean) => v
 ? 1 : 2
const fn1
 = (v
: boolean, w
: any) => v
 ? 1 : 2

参考

条件类型 Conditional Types
条件类型中的类型推断 Type Inference in Conditional Types