Как написать функцию которая возвращает класс в зависимости от параметра?

Question

[Alex]

Есть набор классов.

class TimeoutTextValidator {
  public static type = 'timeout-text' as const
  public schema = {timeout: 1}
}

 class ImageValidator {
  public static type = 'image' as const
  public schema = {src: ''}
}

У всех есть статическое уникальное поле type.

Хочу написать функцию, которая бы по переданному типу возвращала нужный класс:

export function getValidatorByType(type) {
  switch (type) {
    case 'timeout-text':
      return TimeoutTextValidator
    case 'image':
      return ImageValidator
    default:
      return assertUnreachable(type)
  }
}

function assertUnreachable(_: never): never {
  throw new Error("Didn't expect to get here");
}

Проблема в типах. Не могу понять, как заставить ТС корректно выводить возвращаемый тип в зависимости от переданного аргумента.

Мои попытки в playground

Answer 1

[Дмитрий Беляев]

Я бы так сделал:
https://www.typescriptlang.org/play?#code/PTAEiQQR...

Answer 2

[RAX7]

https://www.typescriptlang.org/play?#code/PTAEiQQR...

Comments

[Alex]

Благодарю. То что нужно.
Один вопрос: Возможно ли как-то обойтись без ReturnMap, чтобы TS сам выводил возвращаемый тип?

getValidatorByType('image') // ImageValidator

// 
getValidatorByType('timeout-text') // TimeoutTextValidator

//
const unknownType = '' as 'image' | 'another'
const validator = getValidatorByType(unknownType ) // typeof ImageValidator | typeof AnotherValidator

[RAX7]

Alex, честно говоря я сам не понял почему требуется уточнение типа через as ReturnMap[T]. Наверняка можно сделать как-то по другому. Попробуй позвать Дмитрий Беляев , он гораздо лучше разбирается в черной магии тайпскрипта)

[Дмитрий Беляев]

RAX7, проблема в том, что тип ReturnMap[T] представляет собой конкретный тип снаружи функции и обобщенный тип внутри функции, так как дженерики резолвятся в конкретные типы в момент вызова функции. TS не в состоянии сам сопоставить значение аргумента с конкретным классом, но ему можно об этом рассказать через type-guard

Answer 3

[Игорь Махов]

Это то что нужно: https://www.typescriptlang.org/play?#code/MYGwhgzh... ?