有没有一种方法可以根据输入值定义函数的返回类型

Question

我正在尝试声明以下函数的返回类型

function wrap(input: string) {
    return {
      [input]: '',
      [input + 'Data']: []
    }
}

我可以定义输入attr如下

function wrap<T extends string>(input: T): Record<T, string>;

有没有一种方法可以定义输入+“数据”的类型