TypeScript环境变量断言函数三元返回类型不生效问题排查

核心原因 #

你的条件类型判断总是返回false，本质是TypeScript无法将isLocal的类型窄化为字面量true/false，只能识别为宽泛的boolean类型。当你写IEnvConfig<LOCAL, LIVE>['isLocal'] extends true时，实际是boolean extends true，这显然不成立，所以条件判断总是走false分支。

造成问题的关键：

• 原IEnvConfig接口中isLocal的类型是boolean，而非字面量类型；
• 函数调用时，TypeScript默认会把传入的true/false推断为boolean，而非true或false字面量，导致泛型无法捕获具体布尔值。

修复方案 #

要让TypeScript根据isLocal的具体值推导返回类型，需强制isLocal为字面量类型，并利用泛型的类型推断能力：

1. 调整类型定义，约束isLocal为字面量类型 #

修改IEnvConfig为泛型接口，让isLocal的类型限定为字面量true或false：

type IEnvConfig<LOCAL extends string[], LIVE extends string[], IsLocal extends boolean> = {
  local: LOCAL;
  live: LIVE;
  isLocal: IsLocal;
};

2. 重构函数返回类型推导逻辑 #

直接基于泛型参数IsLocal做条件判断，而非从接口属性中提取：

function assertEnv<
  COMMON extends string[],
  LOCAL extends string[],
  LIVE extends string[],
  IsLocal extends boolean
>(
  config: COMMON | IEnvConfig<LOCAL, LIVE, IsLocal>
): IsLocal extends true 
  ? Record<LOCAL[number], string> & Record<COMMON[number], string>
  : Record<LIVE[number], string> & Record<COMMON[number], string> {
  // 环境变量断言逻辑实现
  const env = process.env;
  const result: Record<string, string> = {};

  if (Array.isArray(config)) {
    config.forEach(key => {
      if (!env[key]) throw new Error(`缺失环境变量: ${key}`);
      result[key] = env[key]!;
    });
  } else {
    const targetKeys = config.isLocal ? config.local : config.live;
    targetKeys.forEach(key => {
      if (!env[key]) throw new Error(`缺失环境变量: ${key}`);
      result[key] = env[key]!;
    });
  }

  return result as any;
}

3. 确保函数调用时推断字面量类型 #

调用函数时，通过as const断言强制TypeScript识别字面量类型：

// 本地环境调用
const localEnv = assertEnv({
  local: ['LOCAL_DEBUG_KEY'],
  live: ['PROD_API_KEY'],
  isLocal: true as const
});

// 生产环境调用
const liveEnv = assertEnv({
  local: ['LOCAL_DEBUG_KEY'],
  live: ['PROD_API_KEY'],
  isLocal: false as const
});

也可以通过函数重载优化参数类型的自动推断，让两种参数格式（数组/配置对象）的处理更清晰：

// 重载1：仅公共环境变量
function assertEnv<COMMON extends string[]>(keys: COMMON): Record<COMMON[number], string>;
// 重载2：区分环境的变量配置
function assertEnv<LOCAL extends string[], LIVE extends string[], IsLocal extends boolean>(
  config: { local: LOCAL; live: LIVE; isLocal: IsLocal }
): IsLocal extends true ? Record<LOCAL[number], string> : Record<LIVE[number], string>;
// 函数实现
function assertEnv(config: any) {
  const env = process.env;
  const result: Record<string, string> = {};

  if (Array.isArray(config)) {
    config.forEach(key => {
      if (!env[key]) throw new Error(`缺失环境变量: ${key}`);
      result[key] = env[key]!;
    });
  } else {
    const targetKeys = config.isLocal ? config.local : config.live;
    targetKeys.forEach(key => {
      if (!env[key]) throw new Error(`缺失环境变量: ${key}`);
      result[key] = env[key]!;
    });
  }

  return result;
}

修改后，TypeScript就能根据传入的isLocal字面量值，精确推导返回的环境变量对象类型。

内容的提问来源于stack exchange，提问作者Pieter-Jan De Bruyne