Как вывести тип используя union и switch?

Question

[Leel Usama]

пример

type TUnion = "foo"|"bar"|"qux"

function fn(value:TUnion) {
    switch(value){
        case "bar": return '';
        case "foo": return true;
        case "qux": return 1;
        default: throw Error('invalid value')
    }
}

let a = fn("bar")

let a: string | number | boolean

Разве switch не должен сузить тип возвращаемого значения?

Comments

[WbICHA]

const values = {
  bar: '',
  foo: true,
  qux: 1,
}

type Values = typeof values;

function fn<T extends keyof Values>(key: T): Values[T] {
  return values[key]
}

let a = fn('bar');

Answer 1

[Станислав Макаров]

Самый сердитый вариант:

type TUnion = "foo"|"bar"|"qux"

function fn(value: "foo"): boolean;
function fn(value: "bar"): string;
function fn(value: "qux"): number;
function fn(value: TUnion): string | boolean | number {
    switch(value){
        case "bar": return '';
        case "foo": return true;
        case "qux": return 1;
        default: throw Error('invalid value')
    }
}

let f = fn("foo"); // f: boolean
let b = fn("bar"); // b: string
let q = fn("qux"); // q: number

Comments

[WbICHA]

Минус данного решения в том, что вернуть ты можешь любой из трёх типов вне зависимости от ключа.
Впрочем, решение Александр от этого тоже не защищает, но с as хотя бы внимательнее будешь.)

[Александр]

WbICHA, Если быть честным - не знаю как сделать лучше.

[WbICHA]

Александр, ну, я оставил в коментах к посту один вариант, но он далеко не дял всех ситуаций подходит. :/

Answer 2

[Александр]

Например так.

type TResult = {
    foo: boolean;
    bar: string;
    qux: number;
}

type TUnion = keyof TResult;

function fn<T extends TUnion>(value: T): TResult[T] {
    switch (value) {
        case 'bar': return '' as TResult[T];
        case 'foo': return true as TResult[T];
        case 'qux': return 1 as TResult[T];
        default: throw new Error('Invalid value');
    }
}

let a = fn('bar');