feat: enable setting custom prisma client type for adapter

Papooch · Feb 22, 2024 · a2ab89f · a2ab89f
1 parent bebefb6
commit a2ab89f
Showing 1 changed file with 17 additions and 12 deletions.
diff --git a/...ansactional-adapters/transactional-adapter-prisma/src/lib/transactional-adapter-prisma.ts b/...ansactional-adapters/transactional-adapter-prisma/src/lib/transactional-adapter-prisma.ts
@@ -1,13 +1,17 @@
 import { TransactionalAdapter } from '@nestjs-cls/transactional';
 import { PrismaClient } from '@prisma/client';
 
-export type PrismaTransactionalClient = Parameters<
-    Parameters<PrismaClient['$transaction']>[0]
->[0];
+interface AnyTransactionClient {
+    $transaction: (fn: (client: any) => Promise<any>, options?: any) => any;
+}
+
+export type PrismaTransactionalClient<
+    TClient extends AnyTransactionClient = PrismaClient,
+> = Parameters<Parameters<TClient['$transaction']>[0]>[0];
 
-export type PrismaTransactionOptions = Parameters<
-    PrismaClient['$transaction']
->[1];
+export type PrismaTransactionOptions<
+    TClient extends AnyTransactionClient = PrismaClient,
+> = Parameters<TClient['$transaction']>[1];
 
 export interface PrismaTransactionalAdapterOptions {
     /**
@@ -16,12 +20,13 @@ export interface PrismaTransactionalAdapterOptions {
     prismaInjectionToken: any;
 }
 
-export class TransactionalAdapterPrisma
-    implements
+export class TransactionalAdapterPrisma<
+    TClient extends AnyTransactionClient = PrismaClient,
+> implements
         TransactionalAdapter<
-            PrismaClient,
-            PrismaTransactionalClient,
-            PrismaTransactionOptions
+            TClient,
+            PrismaTransactionalClient<TClient>,
+            PrismaTransactionOptions<TClient>
         >
 {
     connectionToken: any;
@@ -30,7 +35,7 @@ export class TransactionalAdapterPrisma
         this.connectionToken = options.prismaInjectionToken;
     }
 
-    optionsFactory = (prisma: PrismaClient) => ({
+    optionsFactory = (prisma: TClient) => ({
         wrapWithTransaction: async (
             options: PrismaTransactionOptions,
             fn: (...args: any[]) => Promise<any>,