TypeScript：返回类型将所有键指定为参数

Question

我想以某种方式键入一个函数，返回类型将是一个对象，它包含数组中指定的键作为参数。这些键也应该都是特定的枚举类型。我已经尝试了几件事，但无法使其正常工作。也许有人知道这是否可能以及如何实现？

enum Status {
  TIME = 'time',
  LOCATION = 'location'
}

function getStatus(statusToGet: Status[]) {
  // This function will always return an object containing all passed in
  // status as a key and boolean as a value, e.g.
  return statusToGet.reduce((prev, stat) => {
    prev[stat] = true; // or false
    return prev;
  }, {});
}

getStatus([]) // -> {}
getStatus([Status.TIME]) // -> { time: true }
getStatus([Status.TIME, Status.LOCATION]) // -> { time: true, location: false }

是否可以正确键入getStatus函数，以便TS知道返回类型中有哪些键？

Answer 1

您可以使用映射类型将字符串字面量类型的数组映射到对象：

enum Status {
    TIME = 'time',
    LOCATION = 'location'
}

function getStatus<T extends Status>(statusToGet: T[]): { [P in T]: boolean } {
    return null as any;
}

getStatus([]) // -> {}
getStatus([Status.TIME]) // -> { time: true }
getStatus([Status.TIME, Status.LOCATION]) // -> { time: true, location: false }

注意 如果未启用严格的空值检查，上述代码会为空数组返回{ [s:string] :any}。发生这种情况是因为T 将被推断为any。为了解决这个问题，我们可以使用条件类型来测试T 是否为any 并显式返回{}（对于来自此answer 的任何测试，10x 到@jcalz）。这确实意味着any 的任何数组都会产生{}，但是由于我们期望枚举值数组而不是任何数组，这可能不是太大的问题，实际上可能是一个特性。

function getStatus<T extends Status>(statusToGet: T[]): true extends false & T?  {} : { [P in T]: boolean } {
     return null as any;
}