打字稿基於數組推斷對象鍵

Question

我正在嘗試根據數組的值創建一個具有特定鍵的對象。

這是代碼：

type AllowedKeys = "foo" | "bar" | "baz"

interface Options {
    keys: AllowedKeys[]
}

interface AllTypesDeclaration {
    foo: string[];
    baz: object[];
    bar: number[];
}

function createObj(arg: Options): Pick<AllTypesDeclaration, typeof arg.keys[number]> {
    return {}
}

const myObj = createObj({ keys: ['baz'] })

// This should work:
myObj.baz

// This should fail:
myObj.foo
myObj.bar

如您所見，我嘗試使用Pick ，但這似乎不起作用。

演示： 這里

Answer 1

您將需要泛型，否則鍵的類型將不受輸入的限制。

function createObj<T extends AllowedKeys[]>(arg: { keys: T })
    : Pick<AllTypesDeclaration, typeof arg.keys[number]> {
    return {}
}