如何改进TypeScript多类型检查器的精准类型推断能力

我在TypeScript 5.0.2中实现了一个针对原始类型的检查器，代码如下：

type PrimitiveType = 'undefined' | 'bigint' | 'boolean' | 'function' | 'number' | 'object' | 'string' | 'symbol';

export class TypeChecker {
  static isOneOf(value: unknown, ...types: (PrimitiveType|null)[]): value is PrimitiveType | null {
    const allowedTypes = new Set(types);
    return allowedTypes.has(typeof value) || allowedTypes.has(null) && value === null;
  }
}

常规使用场景可以正常工作：

const x: string|number|null = TypeChecker.isOneOf(value, 'string', 'number') ? value : null;

但在以下场景中编译器会错误判断类型匹配：

const x: string|null = TypeChecker.isOneOf(value, 'string', 'boolean') ? value : null;

需要更新函数，让它能根据传入的类型参数做准确的类型推断。

核心问题是原函数的返回类型断言过于宽泛，没有和传入的types参数建立关联。我们需要通过泛型将传入的类型参数与返回的类型断言绑定，同时单独处理null的特殊情况（因为typeof null返回'object'，无法直接通过typeof匹配）。

修改后的代码如下：

type PrimitiveType = 'undefined' | 'bigint' | 'boolean' | 'function' | 'number' | 'object' | 'string' | 'symbol';

// 定义类型映射：将字符串形式的类型名转换为对应的TS原生类型
type TypeNameToType<T extends PrimitiveType | null> = 
  T extends 'undefined' ? undefined :
  T extends 'bigint' ? bigint :
  T extends 'boolean' ? boolean :
  T extends 'function' ? Function :
  T extends 'number' ? number :
  T extends 'object' ? object :
  T extends 'string' ? string :
  T extends 'symbol' ? symbol :
  T extends null ? null :
  never;

export class TypeChecker {
  // 用泛型T捕获传入的types参数的具体类型，返回的类型断言对应为映射后的TS类型联合
  static isOneOf<T extends PrimitiveType | null>(
    value: unknown, 
    ...types: T[]
  ): value is TypeNameToType<T> {
    const allowedTypes = new Set(types);
    // 优先判断null情况，再处理typeof匹配逻辑
    return value === null 
      ? allowedTypes.has(null) 
      : allowedTypes.has(typeof value as PrimitiveType);
  }
}

验证效果 #

现在用之前的错误场景测试，编译器会正确报错：

const value: unknown = 'test';
// 编译器报错：类型“string | boolean”不能赋值给类型“string | null”
const x: string|null = TypeChecker.isOneOf(value, 'string', 'boolean') ? value : null;

正常场景依然可以正确工作：

const y: string|number|null = TypeChecker.isOneOf(value, 'string', 'number') ? value : null;
// 无报错，类型推断准确

关键改进点 #

• 新增TypeNameToType映射类型，实现字符串类型名到TS原生类型的转换
• 通过泛型T捕获传入的types参数的具体类型，让返回的类型断言与传入类型强绑定
• 优化null判断逻辑，避免typeof null返回'object'带来的干扰

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