TypeScript не верит самому себе?

Question

[Alexandroppolus]

Обезжиренный пример: функция getStrValue возвращает значение строкового поля объекта по ключу.

type StrKeys<T, K extends keyof T = keyof T> = K extends unknown
    ? T[K] extends string ? K : never
    : never;

function getStrValue<T>(obj: T, key: StrKeys<T>): string {
    return obj[key];
}

Ключ key имеет тип StrKeys< T > - только те ключи, по которым в объекте лежат строки. Но TS всё равно сомневается, что в obj[key] строка, и подчеркивает return.

Пробовал разные варианты, например

type OmitNonStrFields<T> = {[K in keyof T as T[K] extends string ? K : never]: T[K]};

function getStrValue<T>(obj: T, key: keyof OmitNonStrFields<T>): string {
    return obj[key];
}

а так же варианты с двумя параметрами генерика для getStrValue, но всё то же самое..

Как сделать, чтобы внутри getStrValue значение obj[key] считалось строкой (там ещё будут строковые операции)?
Или TS в принципе такое не умеет?

Comments

[Александр]

А если так?

function getStrValue<T extends Record<string, any>>(obj: T, key: StrKeys<T>): string {
  return obj[key];
}

OmitNonStrFields можно переделать сразу:

type StringValueKeys<T> = {
    [K in keyof T]: T[K] extends string ? K : never;
}[keyof T];

[WbICHA]

Интересно, что тс не смог.
Можно так:

function getStrValue<T, K extends StrKeys<T>>(obj: T, key: K): T[K] {
  return obj[key];
}

[WbICHA]

Александр, ну, оно будет работать засчёт эни, что не то, чтобы критично в данном случае, но всё же.

[WbICHA]

Можно ещё, кстати, так:

function getStrValue<T>(obj: T, key: StrKeys<T>): T[StrKeys<T>] {
    return obj[key];
}

Либо вообще возвращаемый тип убрать.

[Alexandroppolus]

WbICHA, надо, чтобы obj[key] считался строкой, чтобы, например, я там же (внутри getStrValue) мог с ним что-то сделать как со строкой. Просто пример совсем упрощенный.

[WbICHA]

Alexandroppolus, тогда как насчёт такого варианта?

type ObjStr<T> = {
  [Key in keyof T as [T[Key], string] extends [string, T[Key]] ? Key : never]: string
}

function getStrValue<T>(obj: T & ObjStr<T>, key: keyof ObjStr<T>): string {
  return obj[key];
}

[Alexandroppolus]

WbICHA, как всегда, мастерски! Спасибо

только чуть подправил ObjStr, он стал почти как этот мой OmitNonStrFields, чтобы не терялись поля, у которых тип - подмножество строк:

type ObjStr<T> = {[K in keyof T as T[K] extends string ? K : never]: T[K] & string};

obj: T & ObjStr< T > и строка в значениях ObjStr реально порешали проблему.

[WbICHA]

Alexandroppolus,

только чуть подправил ObjStr, он стал почти как этот мой OmitNonStrFields, чтобы не терялись поля, у которых тип - подмножество строк:

Я специально отсеял юнионы, иначе у тебя на выходе будет не строка и ты не сможешь просто так использовать методы строки, что очевидно. Если тебе нужны опцональные/нуллебл свойства тоже, то лучше добавь в сравнение анндефайнд и нулл, как и в результате.

[Alexandroppolus]

WbICHA,

Я специально отсеял юнионы, иначе у тебя на выходе будет не строка и ты не сможешь просто так использовать методы строки, что очевидно.

Почему не смогу? Если у меня переменная с типом 'foo' | 'bar', то все строковые методы с неё вызываются без проблем.

[WbICHA]

Alexandroppolus, забыл про юнион строк, да.)
Поправил:

type StrType = string | null | undefined

type ObjStr<T> = {
  [Key in keyof T as [T[Key], T[Key] & StrType] extends [StrType, T[Key]] ? Key : never]: StrType;
};

function getStrValue<T>(obj: T & ObjStr<T>, key: keyof ObjStr<T>): StrType {
  return obj[key];
}

[WbICHA]

Alexandroppolus, но выше я говорил про юнион типа string | number.

Забей, твоё решение правильное, я что-то давно сложные типы не писал, забыл как сравнение юнионов работает. :D