如何为模块化Prisma客户端扩展提取并精确类型化

// myCompanyExtension.ts
export const companyExtensions: NeedsType = { // <--- 这里的 NeedsType 是挑战
  update: async ({ args, query }) => {
    if (args.data?.status === CompanyStatus.DECLINED) {
      args.data.user = {
        update: {
          accountLocked: AccountLockedReason.COMPANY_DECLINED,
        },
      };
    }
    return query(args);
  },
};

// prismaClient.ts
const prismaClient = _prismaClient.$extends({
  query: {
    company: companyExtensions, // 在这里使用
  },
});

// 假设 _prismaClient 是一个未经扩展的基础 PrismaClient 实例
import { PrismaClient } from '@prisma/client';
const _prismaClient = new PrismaClient(); // 在实际应用中，这通常是你的基础客户端实例

// 获取 _prismaClient 实例的类型
type BasePrismaClientInstance = typeof _prismaClient;

// Parameters<T>[0] 用于获取函数类型 T 的第一个参数的类型
type RawExtensionConfigType = Parameters<BasePrismaClientInstance['$extends']>[0];

type ExtensionConfigType = Extract<
  Parameters<BasePrismaClientInstance['$extends']>[0],
  { name?: string }
>;

// myCompanyExtension.ts
import { PrismaClient } from '@prisma/client';

// 假设这些枚举已定义或可访问
enum CompanyStatus {
  ACTIVE = 'ACTIVE',
  DECLINED = 'DECLINED',
}

enum AccountLockedReason {
  COMPANY_DECLINED = 'COMPANY_DECLINED',
}

// 1. 获取未经扩展的基础 PrismaClient 实例的类型
// 注意：这里需要一个 'typeof _prismaClient' 来推断类型，
// 如果你的 _prismaClient 是一个单例模式，可以直接引用其类型。
// 为了示例的独立性，我们假设它是一个新的实例，但在实际应用中，
// 应该指向你的应用中实际的基础 PrismaClient 实例。
type BasePrismaClientInstance = InstanceType<typeof PrismaClient>;

// 2. 派生顶层扩展配置的精确类型
type ModularExtensionType = Extract<
  Parameters<BasePrismaClientInstance['$extends']>[0],
  { name?: string }
>;

// 3. 使用派生出的类型定义你的模块化扩展
export const companyStatusUpdateExtension: ModularExtensionType = {
  // 推荐为模块化扩展指定一个唯一的名称，有助于调试和潜在的合并逻辑
  name: 'CompanyStatusUpdateExtension',
  query: {
    company: {
      update: async ({ args, query }) => {
        // 原始的业务逻辑：如果公司状态被拒绝，则锁定关联用户账户
        if (args.data?.status === CompanyStatus.DECLINED) {
          args.data.user = {
            update: {
              accountLocked: AccountLockedReason.COMPANY_DECLINED,
            },
          };
        }
        // 调用原始的 update 查询
        return query(args);
      },
    },
  },
};

// prismaClient.ts
import { PrismaClient } from '@prisma/client';
import { companyStatusUpdateExtension } from './myCompanyExtension'; // 导入你的模块化扩展

// 创建基础的 PrismaClient 实例
const _prismaClient = new PrismaClient();

// 应用模块化扩展
const prismaClient = _prismaClient.$extends(companyStatusUpdateExtension);

// 导出扩展后的客户端及其类型，供应用程序其他部分使用
export type ExtendedPrismaClient = typeof prismaClient;
export const extendedPrismaClient = prismaClient;