如何为第三方Zod类添加带运行时开关的Fluent扩展方法？

我正在使用Zod库，它支持通过链式函数定义验证规则。其中的optional方法可以将字段声明为可选，但该方法不支持传入开关参数来动态启用/禁用可选行为。我的表单是动态构建的，必须在运行时确定字段是否可选。

我已经实现了一个工具方法来实现这个逻辑：

import { z, ZodType} from 'zod';

const fieldIsRequired = true;

// 方法1：包装函数 / 可用
const makeOptional = (input : ZodType, required : boolean): ZodType => required ? input : input.optional();

const entry1 = makeOptional(
    z
    .string()
    .min(10)
    .max(100)
    , !fieldIsRequired
);

这个方法能正常工作，但失去了Zod的链式调用写法，当规则变多的时候，嵌套的函数调用会变得混乱。

我想为第三方库的Zod类型添加一个新的链式函数，尝试了扩展方法但没能成功，代码如下：

import { z, ZodType} from 'zod';

const fieldIsRequired = true;

// 方法2：扩展方法 / 不可用

declare namespace zod {
    export abstract class ZodType {
        makeOptional: (required: boolean)=> ZodType;
    }
}
// 为Zod模式添加语法糖
ZodType.prototype.makeOptional = function (required: boolean): ZodType {
    return required ? this : this.optional();
};

const entry2 =   
  z
    .string()
    .min(10)
    .max(100)
    .makeOptional(!fieldIsRequired);

该如何修正这段代码？

要正确扩展Zod的类型并添加链式方法，需要遵循TypeScript的模块扩展规则，同时适配Zod的泛型类型结构，具体修正步骤如下：

1. 正确扩展Zod的类型声明 #

Zod的类型都在zod模块下，需要使用模块扩展而非自定义命名空间，同时要针对ZodType的泛型版本进行扩展，确保类型推导不丢失：

import { z, ZodType, ZodTypeDef } from 'zod';

// 扩展ZodType的类型声明
declare module 'zod' {
  interface ZodType<Output = any, Def extends ZodTypeDef = ZodTypeDef, Input = Output> {
    makeOptional(this: ZodType<Output, Def, Input>, required: boolean): ZodType<Output | undefined, Def, Input | undefined>;
  }
}

2. 实现原型方法 #

在ZodType原型上添加makeOptional方法，利用Zod原生的optional方法实现逻辑：

// 实现原型方法
ZodType.prototype.makeOptional = function (required: boolean) {
  return required ? this : this.optional();
};

3. 使用扩展后的链式方法 #

现在可以正常使用链式调用，同时类型系统会自动根据required参数调整类型（可选时包含undefined）：

const fieldIsRequired = true;

const entry2 =   
  z
    .string()
    .min(10)
    .max(100)
    .makeOptional(!fieldIsRequired);

关键注意点 #

• 必须使用declare module 'zod'扩展模块类型，TypeScript才能识别对第三方库类型的扩展。
• 保留ZodType的泛型参数，确保返回类型正确反映字段的可选状态，保证类型安全。
• 原型方法的this会自动绑定当前Zod实例，无需额外处理。

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