如何实现TypeScript泛型函数第二个参数为首个数组元素的枚举值并支持自动补全？

要实现让第二个参数只能是第一个数组参数中的具体字符串值，同时支持编辑器自动补全，核心是让TypeScript将数组元素推断为字符串字面量联合类型，而非宽泛的string类型。具体实现如下：

修改后的函数代码 #

function x<T extends string>(a: readonly T[], b: T) {
  return ''
}

调用方式 #

调用时需要对数组参数使用as const断言，强制TypeScript识别数组的具体字面量元素：

// 正常调用，编辑器会自动补全'red'/'yellow'/'green'
x(['red', 'yellow', 'green'] as const, 'red')

// 错误调用，编译器会报错（'blue'不在数组元素中）
x(['red', 'yellow', 'green'] as const, 'blue')

原理说明 #

• 原函数中T会被推断为string，导致第二个参数只要是字符串就合法，无法限制为数组内的具体值。
• 通过T extends string约束泛型为字符串类型，结合readonly T[]的参数类型，再配合as const断言，TypeScript会将数组元素推断为具体的字面量（如'red' | 'yellow' | 'green'），此时第二个参数b必须属于这个联合类型，从而实现输入校验和自动补全。

内容的提问来源于stack exchange，提问作者Islam Y-