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

Question

[Александр Антонов]

Всем привет, на ts часто пишу но в основном использую базовые штуки, до сложного пока как то не добирался по причине ненадобности но сейчас не могу понять как правильно описать возвращаемый тип для функции, на ум приходят всякие условия (как в примере ниже) но учитывая что такое я еще не писал то хз как это правильно делается, заранее благодарю за помощь, пример тестовый так что сильно не ругайте

// Enum нужен потому что я хочу потом где-то использовать эти ключи
const enum TestKeys {
  A = 'a',
  B = 'b',
  C = 'c',
}

type BaseTest = { label: string, value: any }

interface Type extends Record<TestKeys, BaseTest> {
  [TestKeys.A]: { label: string, value: number },
  [TestKeys.B]: { label: string, value: string },
  [TestKeys.C]: { label: string, value: boolean }
}

const test: Type = {
  [TestKeys.A]: { label: '', value: 12 },
  [TestKeys.B]: { label: '', value: 'test' },
  [TestKeys.C]: { label: '', value: true }
}

// Partial так как это пример объекта
// Но по ходу рантайма он может менять кол-во объектов внутри
function testFn<T extends Partial<Type>, K extends TestKeys>(
  obj: T, key: K
): T[K] extends unknown ? undefined : T[K]['value'] {
  return key in obj ? obj[key]!.value : undefined
}

// Return type должен быть string | undefined
const result = testFn(test, TestKeys.B)

Comments

[WbICHA]

key in obj ? obj[key]!.value

А теперь представим, что у нас там свойство со значением андефайнд.

[Александр Антонов]

WbICHA, это вариант для примера, в проекте 100% не будет undefined так как там где его можно менять при сохранении идет проверка структуры через Joi

[WbICHA]

Александр Антонов, тогда зачем партиал?

[Александр Антонов]

WbICHA, так самих ключей может и не быть в объекте, так как позволено убирать или добавлять элементы в основной обьект,

const defObj = {
  a: { label: string, value: number },
  b: { label: string, value: string },
  c: { label: string, value: boolean }
}

Потом юзер захотел убрать свойство "a" так как это некоторая experemental настройка в админке, а сам набор доступных ключей в типах нужен для удобства в коде, что бы не писать обобщенный тип

type Generic = 
  | { label: string, value: number }
  | { label: string, value: string}
  | { label: string, value: boolean}

type Test = Record<string, Generic>

Да и так тогда не получиться в коде правильно типизировать значение когда функцией (которая в вопросе) буду забирать нужную exp настройку

[WbICHA]

Александр Антонов, а зачем вообще эта функция нужна? Не проще напрямую к св-ву обращаться?
obj[key]?.value

[Александр Антонов]

WbICHA, там всякие доп проверки

Answer 1

[Stockholm Syndrome]

function testFn<T extends Type, K extends TestKeys>(
  obj: Partial<T>, key: K
): T[K]['value'] | undefined {
  return key in obj ? obj[key]!.value : undefined
}

Comments

[Александр Антонов]

Тьфу блин, вот я протупил конечно с тем что от Partial расширялся, спасибо

[Alexandroppolus]

Александр Антонов, можно в генерике обойтись одним параметром

function testFn<K extends TestKeys>(
  obj: Partial<Type>, key: K
): Type[K]['value'] | undefined {
  return key in obj ? obj[key]!.value : undefined
}

[Александр Антонов]

Alexandroppolus, Согласен