Prisma Client 扩展类型提取与模块化管理教程

import { PrismaClient, CompanyStatus, AccountLockedReason } from '@prisma/client';

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

const prismaClient = _prismaClient.$extends({
  query: {
    company: {
      update: async ({ args, query }) => {
        if (args.data?.status === CompanyStatus.DECLINED) {
          args.data.user = {
            update: {
              accountLocked: AccountLockedReason.COMPANY_DECLINED,
            },
          };
        }
        return query(args);
      },
    },
  },
});

export type ExtendedPrismaClient = typeof prismaClient;

// companyExtensions.ts
// 我们需要为 companyExtensions 提供一个类型定义
export const companyExtensions = {
  update: async ({ args, query }) => {
    if (args.data?.status === CompanyStatus.DECLINED) {
      args.data.user = {
        update: {
          accountLocked: AccountLockedReason.COMPANY_DECLINED,
        },
      };
    }
    return query(args);
  },
};

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

const _prismaClient = new PrismaClient();

const prismaClient = _prismaClient.$extends({
  query: {
    company: companyExtensions, // 这里需要 companyExtensions 具有正确的类型
  },
});

// types/prisma-extensions.d.ts 或任何 .ts 文件
import { PrismaClient } from '@prisma/client';

// 假设 _prismaClient 是未扩展的 PrismaClient 实例
// 这里我们使用 typeof new PrismaClient() 来获取基础客户端的类型
// 避免循环依赖，或者从一个已知的基础客户端实例获取类型
type BasePrismaClient = PrismaClient; // 也可以直接使用 PrismaClient

/**
 * 提取 Prisma Client $extends 方法的配置参数类型。
 * 通过 Extract<{ name?: string }> 来筛选出符合扩展配置对象特征的类型。
 * 这有助于从复杂的联合类型中精确地获取所需的类型结构。
 */
export type ExtensionArgs = Extract<
  Parameters<typeof BasePrismaClient['$extends']>[0],
  { name?: string }
>;

// companyExtensions.ts
import { CompanyStatus, AccountLockedReason } from '@prisma/client';
import { ExtensionArgs } from './types/prisma-extensions'; // 导入我们定义的类型

// 声明 companyExtensions 为 ExtensionArgs 类型的一部分
// 这里我们只关注 query.company 部分，所以可以进一步缩小类型
export const companyExtensions: ExtensionArgs['query']['company'] = {
  update: async ({ args, query }) => {
    if (args.data?.status === CompanyStatus.DECLINED) {
      args.data.user = {
        update: {
          accountLocked: AccountLockedReason.COMPANY_DECLINED,
        },
      };
    }
    return query(args);
  },
};

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

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

export const prismaClient = _prismaClient.$extends({
  query: {
    company: companyExtensions,
  },
});

export type ExtendedPrismaClient = typeof prismaClient;