TypeScript能否从函数参数值推断对象键并限定返回类型？

举个字符串解析函数的例子：

const { a, b } = parseString({
  string, 
  mapping: [{
    param: 'a',
    // 其他属性
  },{
    param: 'b',
    // 其他属性
  }]
})

希望TypeScript能识别到返回对象仅包含'a'和'b'这两个可用键。

当前的类型定义：

interface ParseStringArgs {
    string: string;
    mapping: {
        start: string | null;
        end: string | null;
        param: string;
    }[];
}

尝试过的写法，但返回类型还是Record<string, string>：

export function parseString({ string, mapping }) {
    // ...
    type ParamKey = typeof mapping[number]['param'];
    const result: Record<ParamKey, string> = {};
    // ... 填充result的逻辑
    return result;
}

现在需要把返回类型改成Record<'a' | 'b', string>这类具体的字面量联合类型，该怎么做？

要实现这个需求，核心是让TypeScript捕获mapping里param的具体字符串字面量，而不是把它当成宽泛的string类型，这里用泛型就能搞定：

步骤1：给参数类型加泛型约束 #

把原来的接口改成泛型版本，让param的类型由泛型参数决定：

interface ParseStringArgs<K extends string> {
    string: string;
    mapping: {
        start: string | null;
        end: string | null;
        param: K;
    }[];
}

步骤2：改造函数，绑定泛型 #

给parseString函数加上泛型参数，让它能从输入的mapping里自动推断出K的具体值：

export function parseString<K extends string>({ string, mapping }: ParseStringArgs<K>) {
    // 用类型断言初始化结果，因为空对象一开始不满足Record<K, string>的结构
    const result = {} as Record<K, string>;
    
    // 这里写你的解析逻辑，确保每个param对应的键都被赋值
    mapping.forEach(item => {
        // 示例解析逻辑，替换成你实际的代码
        result[item.param] = `解析${item.param}得到的值`;
    });

    return result;
}

步骤3：调用时自动推断类型 #

现在调用函数时，TypeScript会自动识别返回对象的键是'a'和'b'：

const { a, b } = parseString({
    string: '需要解析的字符串',
    mapping: [
        { param: 'a', start: '<a>', end: '</a>' },
        { param: 'b', start: '<b>', end: '</b>' }
    ]
});
// 此时返回值类型是Record<'a' | 'b', string>，a、b会被正确提示，不存在的键会报错

简化写法（不用单独定义接口） #

如果不想额外定义泛型接口，也可以直接在函数参数里写约束：

export function parseString<K extends string>({ 
    string, 
    mapping 
}: {
    string: string;
    mapping: { start: string | null; end: string | null; param: K }[]
}) {
    const result = {} as Record<K, string>;
    mapping.forEach(item => {
        result[item.param] = '解析后的值';
    });
    return result;
}

注意点 #

• 必须确保你的解析逻辑会给每个param对应的键赋值，否则运行时可能出现属性缺失的问题
• 泛型K extends string限定了param只能是字符串字面量，避免了宽泛的string类型导致返回类型不精确

内容的提问来源于stack exchange，提问作者Vincent Taylor