Challenge
在本次挑战中,你需要实现一个 DeepPick 工具类型,它扩展了 TypeScript 内置工具类型 Pick 的能力。
type Obj = {
a: number
b: string
c: boolean
obj: {
d: number
e: string
f: boolean
obj2: {
g: number
h: string
i: boolean
}
}
obj3: {
j: number
k: string
l: boolean
}
}
type DeepPick = {}
type cases = [
Expect, unknown>>,
Expect, { a: number }>>,
Expect, { a: number } & { b: string }>>,
Expect, { a: number } & { obj: { e: string } }>>,
Expect,
{ a: number } & { obj: { e: string } } & { obj: { obj2: { i: boolean } } }>>,
]
在以上代码中,我们用到了 Expect 和 Equal 这两个工具类型,它们的实现代码如下:
type Expect = T
type Equal =
(() => T extends X ? 1 : 2) extends
(() => T extends Y ? 1 : 2) ? true : false
Solution
在开始实现 DeepPick 工具类型前,我们先来简单回顾一下 TypeScript 内置的 Pick 工具类型。
type Pick = {
[P in K]: T[P];
}
由上图可知,"" 和 "email" 字符串字面量类型不满足 keyof User 约束,所以 TypeScript 编译器会提示相应的错误信息。此外,Pick 工具类型还支持传入字符串字面量类型组合成的联合类型,比如 "name" | "age"。
下面我们先来分析 DeepPick 工具类型的前三个测试用例:
type Obj = {
a: number
b: string
c: boolean
// ...
}
type DeepPick = {}
type cases = [
Expect, unknown>>,
Expect, { a: number }>>,
Expect, { a: number } & { b: string }>>
]
由以上测试用例可知,DeepPick 工具类型,不需要约束泛型变量 K 的类型。当传入 '' 字符串字面量类型时,只需返回 unknown 类型。而当传入的字符串字面量类型是 keyof T 类型的子类型时,则跟 Pick 工具类型的效果一样。所以,我们可以先实现一个简单的 DeepPick 来满足前面两个测试用例:
type DeepPick = K extends keyof T ? Pick : unknown
然而,第三个测试用例,需要生成 { a: number } & { b: string } 交叉类型。但我们目前返回的是 { a: number } | { b: string } 联合类型。
Pick | Pick
// { a: number } | { b: string }
那么如何把联合类型转换成交叉类型呢?这时我们需要用到 UnionToIntersection 工具类型。
type UnionToIntersection =
(U extends any ? (k: U) => void : never) extends (
k: infer R,
) => void
? R
: never
type DeepPick = UnionToIntersection : unknown>
使用了 UnionToIntersection 工具类型之后,DeepPick
Pick & Pick
目前,我们实现的 DeepPick 工具类型已经可以满足前面三个测试用例。接下来,我们来分析剩下的两个测试用例。
type Obj = {
a: number
obj: {
d: number
e: string
f: boolean
obj2: {
g: number
h: string
i: boolean
}
}
// ...
}
type cases = [
Expect, { a: number } & { obj: { e: string } }>>,
Expect,
{ a: number } & { obj: { e: string } } & { obj: { obj2: { i: boolean } } }>>,
]
由以上的测试用例可知,我们的 DeepPick 工具类型需要支持 "obj.e" 或 "obj.obj2.i" 属性路径的形式。对于属性路径来说,我们需要提取每一层级的属性,然后判断该属性是否是当前层级对象的属性。要提取每一层的属性,我们就需要利用到 TypeScript 条件类型、模板字面量类型和 infer 类型推断。
type GetLastProperty = S extends `${infer U}.${infer R}` ? R : S
type P0 = GetLastProperty<"obj.e"> // "e"
type P1= GetLastProperty<"obj.obj2.i"> // "obj2.i"
在以上代码中,GetLastProperty 工具类型用于获取属性路径的最后一级的属性名。但它还不能正确处理 "obj.obj2.i" 路径。要解决这个问题,我们需要用到 TypeScript 的递归类型。
type GetLastProperty = S extends `${infer U}.${infer R}` ? GetLastProperty : S
type P0 = GetLastProperty<"obj.e"> // "e"
type P1= GetLastProperty<"obj.obj2.i"> // "i"
现在我们已经知道如何处理 "obj.e" 或 "obj.obj2.i" 属性路径,接下来我们来更新 DeepPick 工具类型。
type DeepPick = UnionToIntersection>
type PickByPath =
Path extends keyof Obj
? Pick :
Path extends `${infer Head}.${infer Tail}`
? Head extends keyof Obj ? { [P in Head]: PickByPath } : unknown
: never
type UnionToIntersection =
(U extends any ? (k: U) => void : never) extends (
k: infer R,
) => void
? R
: never
为了让代码更清晰一些,我们定义了一个新的 PickByPath 工具类型。在该工具类型中,我们用到了前面介绍过的 TypeScript 条件类型、模板字面量类型、infer 类型推断和递归类型。