如何基于类型重写TypeScript内置类型的方法？

问题场景 #

定义语言代码的字面量类型：

type LangCodeUpper = 'EN' | 'DE' | 'ES';
type LangCodeLower = Lowercase<LangCodeUpper>;

直接调用原生toLowerCase()时，返回类型是string，无法自动推导为期望的LangCodeLower：

const lang: LangCodeUpper = 'EN';
const lowerLang = lang.toLowerCase(); // 类型为`string`，期望是`LangCodeLower`

尝试修改类型结构解决问题，却引发新错误——类型无法用作索引：

type LangCodeUpperDecl = 'EN' | 'DE' | 'ES';
type LangCodeLowerDecl = Lowercase<LangCodeUpper>;

type LanguageCaseImpl<T> = Omit<T, 'toLowerCase' | 'toUpperCase'> & {
    toUpperCase?: () => LangCodeUpperDecl;
    toLowerCase?: () => LangCodeLowerDecl;
};

type LangCodeUpper = LanguageCaseImpl<LangCodeUpperDecl>;
type LangCodeLower = LanguageCaseImpl<LangCodeLowerDecl>;

// 报错：TS2538: 类型'LangCodeUpper'无法用作索引类型
function a(lang: LangCodeUpper) {
    return this.title[lang];
}

需求是保留基础字符串字面量类型，同时实现类型安全的大小写转换，且不影响索引能力。

可行解决方案 #

方案1：自定义类型安全转换函数 #

通过封装小函数，将转换结果断言为目标类型，既保留原始字面量类型的索引能力，又实现精准类型推导：

type LangCodeUpper = 'EN' | 'DE' | 'ES';
type LangCodeLower = Lowercase<LangCodeUpper>;

// 小写转换函数，严格约束输入输出类型
const toLangLower = (lang: LangCodeUpper): LangCodeLower => {
  return lang.toLowerCase() as LangCodeLower;
};

// 大写转换函数（按需使用）
const toLangUpper = (lang: LangCodeLower): LangCodeUpper => {
  return lang.toUpperCase() as LangCodeUpper;
};

// 使用示例
const lang: LangCodeUpper = 'EN';
const lowerLang = toLangLower(lang); // 类型为 LangCodeLower

// 正常作为索引使用
const titleMap = {
  EN: 'Hello World',
  DE: 'Hallo Welt',
  ES: 'Hola Mundo'
};
function getTitle(lang: LangCodeUpper) {
  return titleMap[lang]; // 无类型错误
}

方案2：全局扩充字符串方法类型（谨慎使用） #

通过类型扩充，让特定字面量类型调用原生方法时返回目标类型，无需额外封装函数：

type LangCodeUpper = 'EN' | 'DE' | 'ES';
type LangCodeLower = Lowercase<LangCodeUpper>;

declare global {
  interface String {
    // 仅当调用者是LangCodeUpper类型时，返回LangCodeLower
    toLowerCase(this: LangCodeUpper): LangCodeLower;
    // 仅当调用者是LangCodeLower类型时，返回LangCodeUpper
    toUpperCase(this: LangCodeLower): LangCodeUpper;
  }
}

// 使用示例
const lang: LangCodeUpper = 'EN';
const lowerLang = lang.toLowerCase(); // 类型自动推导为 LangCodeLower

const titleMap = { EN: 'Hello', DE: 'Hallo', ES: 'Hola' };
function getTitle(lang: LangCodeUpper) {
  return titleMap[lang]; // 无索引类型错误
}

注意：此方案会全局修改String接口，仅适合在特定项目中使用，避免影响其他类型的方法行为。

方案说明 #

之前的实现问题在于：LanguageCaseImpl将原始字符串字面量类型修改为复合类型，导致TS无法识别其为合法的索引类型。而上述两种方案均基于原始字符串字面量类型进行扩展或封装，既满足类型安全的转换需求，又保留了索引能力。

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