mosya<TC> - Replace型を作って型で文字を置換しよう
この記事はmosya<TC>の問題の一つであるReplace型の解説になります。
問題
文字列S
に含まれる文字FromをToに一度だけ置き換える型Replace<S, From, To>
を実装します。
例えば
type replaced = Replace<
"types are fun!",
"fun",
"awesome"
>; // expected to be 'types are awesome!'
受け取った文字列を Union 型に変換する型を実装します。
例えば以下のようなコードを満たすようにStringToUnion型を実装しましょう。
type Test = "123";
type Result = StringToUnion<Test>; // expected to be "1" | "2" | "3"
前提知識
この問題を解くにあたって型についての以下の知識を理解しておく必要があります。
Conditional Types
を理解するinfer
を理解するTemplate Literal Types
を理解する
Conditional Typesを理解する
Conditional Typesは、条件によって型を変更することができる機能です。
例えば、以下のような型が考えられます。
type Foo<T> = T extends string
? string
: number;
この型は、T
がstring
型を継承している場合はstring
型を、そうでない場合はnumber
型を返します。
このように、extends
を使って条件を指定することで、型を変更することができます。
inferを理解する
infer
は、型を推論することができる機能です。
例えば、以下のような型が考えられます。
type ArrayItem<T> =
T extends (infer R)[] ? R : never;
この型は、T
が配列の場合は、配列の中の型を返します。
以下のように使うことができます。
type Foo = ArrayItem<string[]>; // string
この場合、infer R
にはstring[]
型が当てはまるので、R
はstring
型に推論され、R
を返すのでstring
型が返されます。
このように推論される型を取得するのにinfer
は役立ちます。
Template Literal Typesを理解する
Template Literal Typesは、文字列リテラルを使って型を作成することができる機能です。型を組み合わせて新しい文字列の型を作成することができます。
例えば、以下のような型が考えられます。
type Foo<T> = `${T} World`;
type HelloWorld = Foo<"Hello">; // "Hello World"
このように、文字列リテラルを使って型を作成することができます。
解答例
以上の知識を使って、以下のように解答することができます。
type Replace<
S extends string,
From extends string,
To extends string
> = From extends ""
? S
: S extends `${infer L}${From}${infer R}`
? `${L}${To}${R}`
: S;
まず、引数の、S, From, Toにはそれぞれ文字列型が入るので、S extends string, From extends string, To extends string
というように制約をつけます。
type Replace<S extends string, From extends string, To extends string> = ...
次に、From
が空文字の場合は、S
をそのまま返すようにします。
type Replace<S extends string, From extends string, To extends string> = From extends '' ? S : ...
そうでない場合は、S
がFrom
を含んでいるかどうかを判定します。含んでいなければ、S
をそのまま返します。
type Replace<S extends string, From extends string, To extends string> = From extends '' ? S : S extends `${infer L}${From}${infer R}` ? ... : S
S
がFrom
を含んでいる場合は、S
の中からFrom
を取り除いて、To
を挿入します。
type Replace<
S extends string,
From extends string,
To extends string
> = From extends ""
? S
: S extends `${infer L}${From}${infer R}`
? `${L}${To}${R}`
: S;
Authored by
Godai@steelydylan
Webサービスを作るのが好きなWebエンジニア。子供が産まれたことをきっかけに独立し法人化。サービス開発が大好き。
好きな言語はTypeScript。