feat(client): add experimental hooks

src/packages/client/fixtures/blog/main.ts

@@ -5,6 +5,16 @@ const prisma = new PrismaClient({
 })
 
 async function main() {
+  ;(prisma as any).use('all', async ({ params, fetch }) => {
+    console.log(params.clientMethod, `before`, params)
+    const data = await fetch(params)
+    console.log(params.clientMethod, data)
+    data.push({
+      name: 'fake user',
+    })
+    console.log(params.clientMethod, `after`)
+    return data
+  })
   const users = await prisma.transaction([
     prisma.user.findMany({
       include: {

src/packages/client/src/__tests__/batching.test.ts

@@ -40,6 +40,11 @@ describe('batching', () => {
         isList: false,
         rootField: 'query',
         typeName: 'User',
+        args: {
+          where: {
+            id: '1',
+          },
+        },
       }),
       fetcher.request({
         clientMethod: 'findOne',
@@ -57,6 +62,11 @@ describe('batching', () => {
         isList: false,
         rootField: 'query',
         typeName: 'User',
+        args: {
+          where: {
+            id: '2',
+          },
+        },
       }),
     ])
 
@@ -125,6 +135,9 @@ describe('batching', () => {
         isList: false,
         rootField: 'query',
         typeName: 'User',
+        args: {
+          where: { id: '1' },
+        },
       }),
       fetcher.request({
         clientMethod: 'findOne',
@@ -142,6 +155,9 @@ describe('batching', () => {
         isList: false,
         rootField: 'query',
         typeName: 'User',
+        args: {
+          where: { id: '2' },
+        },
       }),
     ])
 
@@ -211,6 +227,7 @@ describe('batching', () => {
         isList: false,
         rootField: 'query',
         typeName: 'User',
+        args: { where: { email: 'a@a.de' } },
       }),
       fetcher.request({
         clientMethod: 'findOne',
@@ -228,6 +245,7 @@ describe('batching', () => {
         isList: false,
         rootField: 'query',
         typeName: 'User',
+        args: { where: { id: '2' } },
       }),
     ])
 

src/packages/client/src/__tests__/runtime-tests/__snapshots__/runtime.test.ts.snap

@@ -17,6 +17,8 @@ exports[`runtime works blog-env-postgresql example should succeed: blog-env-post
 
 exports[`runtime works enums example should succeed: enums example should succeed 1`] = `undefined`;
 
+exports[`runtime works hooks example should succeed: hooks example should succeed 1`] = `undefined`;
+
 exports[`runtime works missing-binary example should not succeed: missing-binary example should not succeed 1`] = `[TypeError: e.message.split(...).slice(...).trim is not a function]`;
 
 exports[`runtime works missing-binary-native example should not succeed: missing-binary-native example should not succeed 1`] = `[TypeError: e.message.split(...).slice(...).trim is not a function]`;

src/packages/client/src/__tests__/runtime-tests/hooks/.gitignore

@@ -0,0 +1 @@
+!dev.db

src/packages/client/src/__tests__/runtime-tests/hooks/index.js

@@ -0,0 +1,49 @@
+const { PrismaClient } = require('@prisma/client')
+const assert = require('assert')
+
+module.exports = async () => {
+  const db = new PrismaClient()
+
+  const allResults = []
+  const engineResults = []
+
+  db.use('all', async ({ params, fetch }) => {
+    const result = await fetch(params)
+    allResults.push(result)
+    return result
+  })
+
+  db.use('engine', async ({ params, fetch }) => {
+    const result = await fetch(params)
+    engineResults.push(result)
+    return result
+  })
+
+  await db.user.findMany()
+  await db.post.findMany()
+
+  assert.deepEqual(allResults, [[], []])
+  assert.deepEqual(
+    engineResults.map((r) => r.data),
+    [
+      {
+        data: {
+          findManyUser: [],
+        },
+      },
+      {
+        data: {
+          findManyPost: [],
+        },
+      },
+    ],
+  )
+  assert(typeof engineResults[0].elapsed === 'number')
+  assert(typeof engineResults[1].elapsed === 'number')
+
+  db.disconnect()
+}
+
+if (require.main === module) {
+  module.exports()
+}

src/packages/client/src/__tests__/runtime-tests/hooks/manifest.json

@@ -0,0 +1,3 @@
+{
+  "shouldSucceed": true
+}

src/packages/client/src/__tests__/runtime-tests/hooks/package.json

@@ -0,0 +1,13 @@
+{
+  "name": "my-prisma-project",
+  "version": "1.0.0",
+  "description": "",
+  "main": "index.js",
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "keywords": [],
+  "author": "",
+  "license": "ISC"
+}
+

src/packages/client/src/__tests__/runtime-tests/hooks/schema.prisma

@@ -0,0 +1,30 @@
+datasource db {
+  provider = "sqlite"
+  url      = "file:dev.db"
+  default  = true
+}
+
+generator client {
+  provider             = "prisma-client-js"
+  experimentalFeatures = ["transactionApi"]
+}
+
+// / User model comment
+model User {
+  id    String  @default(uuid()) @id
+  email String  @unique
+  // / name comment
+  name  String?
+  posts Post[]
+}
+
+model Post {
+  id        String   @default(cuid()) @id
+  createdAt DateTime @default(now())
+  updatedAt DateTime @updatedAt
+  published Boolean
+  title     String
+  content   String?
+  authorId  String?
+  author    User?    @relation(fields: [authorId], references: [id])
+}

src/packages/client/src/__tests__/runtime-tests/runtime.test.ts

@@ -9,6 +9,8 @@ jest.setTimeout(35000)
 
 // TODO: Figure out the problem with debug
 
+process.setMaxListeners(100)
+
 describe('runtime works', () => {
   const subDirs = getSubDirs(__dirname)
   for (const dir of subDirs) {