TypeScriptのユニオン型の判別キーからもとの型を推論する方法

Question

実現したいこと

TypeScriptの型推論の話です。
フィールドを判別キーとしたユニオン型を定義したとき、そのキーからもとの型を推論する型を作りたいです。

例えば以下のような Action 型と、その判別キーの型 ActionKey 型があるとします。

type ClickAction = {
  key: 'click';
  x: number;
  y: number;
}
type CopyAction = {
  key: 'copy';
  data: string;
}
type Action = ClickAction | CopyAction;

type ActionKey = Action['key'];

このとき、取りうるActionKeyによって本来の型を表す型 ActionType<ActionKey> を作りたいです。

ActionType<'click'> // => ClickAction を表す
ActionType<'copy'> // =>CopyAction を表す

試したこと

以下のような定義の仕方で一応実現できているかと思います。

type ClickAction = {
  key: 'click';
  x: number;
  y: number;
}
type CopyAction = {
  key: 'copy';
  data: string;
}
type ActionMap = {
  click: ClickAction;
  copy: CopyAction;
}
type ActionKey = keyof ActionMap;
type Action = ActionMap[ActionKey];
type ActionType<T extends ActionKey> = ActionMap[T];

ただ、このやり方ではActionの追加時にActionのkeyフィールドとActionMapのキーの文字列をあわせて追加しなければならず、少し保守性が悪いように思えます。
本当ならば ActionMap の定義など使わず、一撃でActionTypeを作る方法があればよいのですが、何かアイデアありませんでしょうか。

Answer 1

試してみたところ一応できたと思うので回答します。

type Actions = ClickAction | CopyAction;
type ActionKey = Actions['key'];
type ActionType<T extends ActionKey> = Extract<Actions, { key: T } & Actions>;

軽く解説ですが
交差型(&)は、同じプロパティがある場合、一致していないものは除外されます。
そのため、仮に T が 'click' の場合、 { key: T } & Actions は
{ key: "click" } & ClickAction となります。
また、Extract は２つの型の共通項を取るので、結果的に ClickAction のみ残るという形です。

Comments

[kakusuke]

Extract の使い方、初めて知りました。
ありがとうございます！