Conditional Types

At the heart of most useful programs, we have to make decisions based on input. JavaScript programs are no different, but given the fact that values can be easily introspected, those decisions are also based on the types of the inputs. Conditional types help describe the relation between the types of inputs and outputs.

  1. interface Animal {
    live(): void;
    }
    interface Dog extends Animal {
    woof(): void;
    }
  2. type Example1 = Dog extends Animal ? number : string;
    // ^ = type Example1 = number
  3. type Example2 = RegExp extends Animal ? number : string;
    // ^ = type Example2 = string

Conditional types take a form that looks a little like conditional expressions (condition ? trueExpression : falseExpression) in JavaScript:

  1. SomeType extends OtherType ? TrueType : FalseType;

When the type on the left of the extends is assignable to the one on the right, then you’ll get the type in the first branch (the “true” branch); otherwise you’ll get the type in the latter branch (the “false” branch).

From the examples above, conditional types might not immediately seem useful - we can tell ourselves whether or not Dog extends Animal and pick number or string! But the power of conditional types comes from using them with generics.

For example, let’s take the following createLabel function:

  1. interface IdLabel {
    id: number /* some fields */;
    }
    interface NameLabel {
    name: string /* other fields */;
    }
  2. function createLabel(id: number): IdLabel;
    function createLabel(name: string): NameLabel;
    function createLabel(nameOrId: string | number): IdLabel | NameLabel;
    function createLabel(nameOrId: string | number): IdLabel | NameLabel {
    throw "unimplemented";
    }

These overloads for createLabel describe a single JavaScript function that makes a choice based on the types of its inputs. Note a few things:

  1. If a library has to make the same sort of choice over and over throughout its API, this becomes cumbersome.
  2. We have to create three overloads: one for each case when we’re sure of the type (one for string and one for number), and one for the most general case (taking a string | number). For every new type createLabel can handle, the number of overloads grows exponentially.

Instead, we can encode that logic in a conditional type:

  1. type NameOrId<T extends number | string> = T extends number
    ? IdLabel
    : NameLabel;

We can then use that conditional type to simplify out overloads down to a single function with no overloads.

  1. function createLabel<T extends number | string>(idOrName: T): NameOrId<T> {
    throw "unimplemented";
    }
  2. let a = createLabel("typescript");
    // ^ = let a: NameLabel
  3. let b = createLabel(2.8);
    // ^ = let b: IdLabel
  4. let c = createLabel(Math.random() ? "hello" : 42);
    // ^ = let c: NameLabel | IdLabel

Conditional Type Constraints

Often, the checks in a conditional type will provide us with some new information. Just like with narrowing with type guards can give us a more specific type, the true branch of a conditional type will further constraint generics by the type we check against.

For example, let’s take the following:

  1. type MessageOf<T> = T["message"];
    Type '"message"' cannot be used to index type 'T'.2536Type '"message"' cannot be used to index type 'T'.

In this example, TypeScript errors because T isn’t known to have a property called message. We could constrain T, and TypeScript would no longer complain:

  1. type MessageOf<T extends { message: unknown }> = T["message"];
  2. interface Email {
    message: string;
    }
  3. interface Dog {
    bark(): void;
    }
  4. type EmailMessageContents = MessageOf<Email>;
    // ^ = type EmailMessageContents = string

However, what if we wanted MessageOf to take any type, and default to something like never if a message property isn’t available? We can do this by moving the constraint out and introducing a conditional type:

  1. type MessageOf<T> = T extends { message: unknown } ? T["message"] : never;
  2. interface Email {
    message: string;
    }
  3. interface Dog {
    bark(): void;
    }
  4. type EmailMessageContents = MessageOf<Email>;
    // ^ = type EmailMessageContents = string
  5. type DogMessageContents = MessageOf<Dog>;
    // ^ = type DogMessageContents = never

Within the true branch, TypeScript knows that T will have a message property.

As another example, we could also write a type called Flatten that flattens array types to their element types, but leaves them alone otherwise:

  1. type Flatten<T> = T extends any[] ? T[number] : T;
  2. // Extracts out the element type.
    type Str = Flatten<string[]>;
    // ^ = type Str = string
  3. // Leaves the type alone.
    type Num = Flatten<number>;
    // ^ = type Num = number

When Flatten is given an array type, it uses an indexed access with number to fetch out string[]’s element type. Otherwise, it just returns the type it was given.

Inferring Within Conditional Types

We just found ourselves using conditional types to apply constraints and then extract out types. This ends up being such a common operation that conditional types make it easier.

Conditional types provide us with a way to infer from types we compare against in the true branch using the infer keyword. For example, we could have inferred the element type in Flatten instead of fetching it out “manually” with an indexed access type:

  1. type Flatten<Type> = Type extends Array<infer Item> ? Item : Type;

Here, we used the infer keyword declaratively introduced a new generic type variable named U instead of specifying how to retrieve the element type of T within the true branch. This frees us from having to think about how to dig through and probing apart the structure of the types we’re interested.

We can write some useful helper type aliases using the infer keyword. For example, for simple cases, we can extract the return type out from function types:

  1. type GetReturnType<Type> = Type extends (...args: never[]) => infer Return
    ? Return
    : never;
  2. type Num = GetReturnType<() => number>;
    // ^ = type Num = number
  3. type Str = GetReturnType<(x: string) => string>;
    // ^ = type Str = string
  4. type Bools = GetReturnType<(a: boolean, b: boolean) => boolean[]>;
    // ^ = type Bools = boolean[]

Distributive Conditional Types

When conditional types act on a generic type, they become distributive when given a union type. For example, take the following:

  1. type ToArray<Type> = Type extends any ? Type[] : never;

If we plug a union type into ToArray, then the conditional type will be applied to each member of that union.

  1. type ToArray<Type> = Type extends any ? Type[] : never;
  2. type StrArrOrNumArr = ToArray<string | number>;
    // ^ = type StrArrOrNumArr = string[] | number[]

What happens here is that StrOrNumArray distributes on:

  1. string | number;

and maps over each member type of the union, to what is effectively:

  1. ToArray<string> | ToArray<number>;

which leaves us with:

  1. string[] | number[];

Typically, distributivity is the desired behavior. To avoid that behavior, you can surround each side of the extends keyword with square brackets.

  1. type ToArrayNonDist<Type> = [Type] extends [any] ? Type[] : never;
  2. // 'StrOrNumArr' is no longer a union.
    type StrOrNumArr = ToArrayNonDist<string | number>;
    // ^ = type StrOrNumArr = (string | number)[]