mosya<TC> - 関数の引数を追加する型を作ろうの解説

この記事はmosya<TC>の問題の一つであるAppendArgument型の解説になります。

問題

与えられた関数型 Fn と任意の型 A に対して、第一引数に Fn を取り、第二引数に A を取り、Fn の引数に A を追加した関数型 G を生成します。

例えば、

type Fn = (a: number, b: string) => number

type Result = AppendArgument<Fn, boolean> 
// expected be (a: number, b: string, x: boolean) => number

解答例

type AppendArgument<Fn, A> = Fn extends (...args: infer R) => infer P ? (...args: [...R, A]) => P : never

1. Fnが関数型であることを条件として、Fn extends (...args: infer R) => infer Pという条件を指定します。
2. ...args: infer Rを指定し、...argsという関数の可変長引数をRという型に推論して代入します
3. 帰り値をinfer Pで、Pという型に推論して代入します
4. Fnがこの条件を満たす場合は、(...args: [...R, A]) => Pという型を返します
5. ...args: [...R, A]は、先ほど可変長引数を配列としてRという型に推論したので、その配列にAを追加した型になります。
6. Pは、Fnの帰り値の型なので、そのまま返します。