如何让 TS 在通用构建器函数中推断回调？

Question

考虑以下用例演示 ( playground )：

// A builder that can self-reference its keys using a ref function
declare function makeObj<K extends string>(
    builder: (ref: (k: K) => number) => Record<K, number>
): Record<K, number>;

// Not using `ref` for now. All good, K is inferred as <"x" | "y">.
const obj1 = makeObj(() => ({ x: 1, y: 2 }));

// Oops, now that we try to use `ref`, K is inferred as <string>.
const obj2 = makeObj(ref => ({ x: 1, y: ref("invalid key, only x or y") }));

// This works, but we'd want K to be automatically inferred.
const obj3 = makeObj<"x" | "y">(ref => ({ x: 1, y: ref("x") }));

所以，我应该怎么写makeObj，以便K自动推断？

Answer 1

尝试声明Record<K, number>为第二个泛型类型。

操场

declare function makeObj<K extends string, T extends Record<K, number> = Record<K, number>>(
  builder: (ref: (k: K) => number) => T
): T

const obj1 = makeObj(() => ({ x: 1, y: 2 }));
const obj2 = makeObj(ref => ({ x: 1, y: ref("invalid key, only x or y") }));
const obj3 = makeObj(ref => ({ x: 1, y: ref("x") }));

局限性

好吧，正如@kaya3 在下面评论的那样。此解析只能推断返回类型。除非显式设置泛型类型，否则它仍然找不到无效键。

// error will shown when given explicit generic type
const obj2 = makeObj<'x' | 'y'>(ref => ({x: 1, y: ref("invalid key, only x or y")}));