如何在TypeScript函数中利用泛型和Zod覆盖接口并保持正确的返回类型

import { z } from 'zod';

// 默认验证器
export const EmailValidator = z.object({
  email: z
    .string({ required_error: 'auth.validation.email' })
    .email({ message: 'auth.validation.email_format' })
});

// 基础插件配置接口
interface PluginConfig {
  validator?: z.ZodType; // 问题点1: z.ZodType 是一个类型，而非一个可赋值的构造函数
}

// 默认插件配置接口
interface DefaultPluginConfig {
  validator?: typeof EmailValidator;
}

const definePlugin = <T extends PluginConfig = DefaultPluginConfig>({
  validator = EmailValidator
}: T) => {
  return validator.parse({});
};

const test = definePlugin({});
// 此时 test.email 会是 any，因为 definePlugin 的返回类型无法被正确推断

// 自定义验证器和接口
const CustomValidator = z.object({
  email: z.string(),
  username: z.string()
});

interface CustomConfig {
  validator?: typeof CustomValidator;
}

const test2 = definePlugin<CustomConfig>({
  validator: CustomValidator
});
// 此时 test2.username 也会是 any

import { z, ZodType } from 'zod'; // 导入 ZodType

export const EmailValidator = z.object({
  email: z
    .string({ required_error: 'auth.validation.email' })
    .email({ message: 'auth.validation.email_format' })
});

// 修正后的基础插件配置接口
interface PluginConfig {
  validator?: ZodType<any>; // 使用 ZodType<any> 提供更宽泛的类型兼容性
}

// 默认插件配置接口，并正确继承 PluginConfig
interface DefaultPluginConfig extends PluginConfig {
  validator?: typeof EmailValidator;
}

// ... definePlugin 函数保持不变，但此时仍有返回类型问题

import { z, ZodType } from "zod";

// 创建默认验证器，添加 default 以确保 parse({}) 总是返回一个对象
export const EmailValidator = z.object({
  email: z.string().default("")
});

// 泛型 PluginConfig 接口，捕获 validator 的具体 ZodType
interface PluginConfig<T extends ZodType = typeof EmailValidator> {
  validator?: T;
}

/**
 * 定义一个插件函数，能够处理默认或自定义的Zod验证器，
 * 并精确推断返回对象的类型。
 *
 * @template T - 插件配置类型，默认为 PluginConfig<typeof EmailValidator>。
 * @template R - 从 T 中推断出的具体 ZodType。
 * @param {T} config - 插件配置对象，包含可选的 validator。
 * @returns {P} - 经过 validator.parse({}) 处理后得到的对象类型。
 */
const definePlugin = <
  // T 是传入的配置对象类型，默认为包含 EmailValidator 的 PluginConfig
  T extends PluginConfig = PluginConfig<typeof EmailValidator>,
  // R 是从 T 中推断出的具体 ZodType (例如 EmailValidator 或 CustomValidator)
  R = T extends PluginConfig<infer V> ? V : ZodType
>(
  { validator = EmailValidator }: T
): R extends ZodType<infer P> ? P : never => { // 返回类型：从 R (ZodType) 中推断出其输出类型 P
  // 这里使用 as any 是因为 TypeScript 编译器在运行时无法完全验证 parse 的结果类型
  // 但我们通过泛型保证了编译时的类型安全
  return validator.parse({}) as any;
};

// 示例1：使用默认验证器
const test = definePlugin({});
// 此时 test 的类型为 { email: string; }
console.log(test.email); // 正确推断，无类型错误

// 创建一个自定义验证器
const CustomValidator = z.object({
  email: z.string().default(""),
  username: z.string().default("")
});

// 定义一个使用 CustomValidator 的配置类型
type CustomConfig = PluginConfig<typeof CustomValidator>;

// 示例2：使用自定义验证器
const test2 = definePlugin<CustomConfig>({
  validator: CustomValidator
});

// 此时 test2 的类型为 { email: string; username: string; }
console.log(test2.username); // 正确推断，无类型错误
console.log(test2.email);   // 同样正确推断