如何让TypeScript自动推断函数返回值为参数指定的字面量联合类型？

这问题问得太到位了！你现在的核心问题是，原来的函数把validInputs定义成了string[]，TypeScript会自动把你传入的具体字面量（比如"y""n"）拓宽成宽泛的string类型，导致返回值只能被推断成Promise<string>。要让TypeScript精准追踪到这些具体的字面量联合类型，用泛型就能轻松解决！

直接上修改后的代码，我给你逐行拆解关键变化：

static async prompt<T extends string>(message: string, ...validInputs: T[]): Promise<T> {
  while(true) {
    const input = promptGetInput(message); // 假设这个底层函数返回string
    // 因为input是string类型，find会返回T|undefined，但我们循环会一直等到有效输入，所以可以安全断言
    const validInput = validInputs.find(x => x === input) as T;
    if (validInput) return validInput;
    else console.log("Wrong input.");
  }
}

几个核心调整：

• 给函数添加泛型T extends string：T被约束为string的子类型，会自动捕获你传入的每一个具体字面量值
• 把...validInputs: string[]改成...validInputs: T[]：告诉TypeScript，这个参数数组里的每一项都是T类型的成员，而非普通的string
• 返回值声明为Promise<T>：这样当你调用函数时，T会被自动推断成所有传入的validInputs字面量的联合类型

比如你调用这个函数时：

const result = prompt("Cancel?", "y", "n");
// 此时TypeScript会自动把result的类型推断为Promise<"y" | "n">

VS Code的智能提示也会准确识别返回值的可能类型，再也不会只显示宽泛的string啦。

如果你的底层promptGetInput是异步函数（毕竟原函数用了async），只要给input加个await就行，泛型的逻辑完全不变：

static async prompt<T extends string>(message: string, ...validInputs: T[]): Promise<T> {
  while(true) {
    const input = await promptGetInput(message); // 异步场景加await
    const validInput = validInputs.find(x => x === input) as T;
    if (validInput) return validInput;
    else console.log("Wrong input.");
  }
}

不管底层是同步还是异步，类型推断都能完美匹配你的需求～

内容来源于stack exchange