added patch for client type in transaction

HRM · Sep 9, 2024 · f868980 · f868980
1 parent 0dcf7f4
commit f868980
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 10 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "prisma-nitro-patcher",
-  "version": "0.0.8",
+  "version": "0.0.9",
   "description": "Applies a patch on the prisma client to improve the performance of extensions",
   "files": [ "dist" ],
   "bin": {

diff --git a/src/patchPrisma.ts b/src/patchPrisma.ts
@@ -73,7 +73,8 @@ export function patchPrisma(contents: string): string {
   }
   if (
     contents.indexOf('runtime.Types.Result') == -1 ||
-    contents.indexOf('$extends: $Extensions.ExtendsHook') == -1
+    contents.indexOf('$extends: $Extensions.ExtendsHook') == -1 ||
+    contents.indexOf('$transaction<R>(fn: (prisma: Omit<PrismaClient') == -1
   ) {
     throw new PresentableError(
       'The prisma version or file provided is not compatible with patching',
@@ -85,8 +86,13 @@ export function patchPrisma(contents: string): string {
     1,
     extendsHookModified,
   );
-  return contentWithPatchType.replace(
-    '$extends: $Extensions.ExtendsHook',
-    '$extends: ExtendsHookModified',
-  );
+  return contentWithPatchType
+    .replace(
+      '$extends: $Extensions.ExtendsHook',
+      '$extends: ExtendsHookModified',
+    )
+    .replace(
+      '$transaction<R>(fn: (prisma: Omit<PrismaClient',
+      '$transaction<R>(fn: (prisma: Omit<this',
+    );
 }
diff --git a/tests/files/ver-5.19.1/index-patched.txt b/tests/files/ver-5.19.1/index-patched.txt
@@ -181,7 +181,7 @@ export class PrismaClient<
    */
   $transaction<P extends Prisma.PrismaPromise<any>[]>(arg: [...P], options?: { isolationLevel?: Prisma.TransactionIsolationLevel }): $Utils.JsPromise<runtime.Types.Utils.UnwrapTuple<P>>
 
-  $transaction<R>(fn: (prisma: Omit<PrismaClient, runtime.ITXClientDenyList>) => $Utils.JsPromise<R>, options?: { maxWait?: number, timeout?: number, isolationLevel?: Prisma.TransactionIsolationLevel }): $Utils.JsPromise<R>
+  $transaction<R>(fn: (prisma: Omit<this, runtime.ITXClientDenyList>) => $Utils.JsPromise<R>, options?: { maxWait?: number, timeout?: number, isolationLevel?: Prisma.TransactionIsolationLevel }): $Utils.JsPromise<R>
 
 
   $extends: ExtendsHookModified<"extends", Prisma.TypeMapCb, ExtArgs>