fix(runtime): proxy $extends for each and every enhancement

Author: Gabrola

packages/runtime/src/enhancements/node/proxy.ts

@@ -267,12 +267,7 @@ export function makeProxy<T extends PrismaProxyHandler>(
                 if ($extends && typeof $extends === 'function') {
                     return (...args: any[]) => {
                         const result = $extends.bind(target)(...args);
-                        if (!result[PRISMA_PROXY_ENHANCER]) {
-                            return makeProxy(result, modelMeta, makeHandler, name + '$ext', errorTransformer);
-                        } else {
-                            // avoid double wrapping
-                            return result;
-                        }
+                        return makeProxy(result, modelMeta, makeHandler, name + '$ext', errorTransformer);
                     };
                 } else {
                     return $extends;

tests/regression/tests/issue-2014.test.ts

@@ -0,0 +1,39 @@
+import { loadSchema } from '@zenstackhq/testtools';
+
+describe('issue 2014', () => {
+    it('regression', async () => {
+        const { prisma, enhance } = await loadSchema(
+            `
+            model Tenant {
+              id    Int    @id @default(autoincrement())
+            
+              users User[]
+            }
+            
+            model User {
+              id       Int    @id @default(autoincrement())
+              tenantId Int    @default(auth().tenantId)
+              tenant   Tenant @relation(fields: [tenantId], references: [id])
+            
+              @@allow('all', true)
+            }
+            `,
+            { logPrismaQuery: true }
+        );
+
+        const tenant = await prisma.tenant.create({ data: {} });
+        const user = await prisma.user.create({ data: { tenantId: tenant.id } });
+
+        const db = enhance(user);
+        const extendedDb = db.$extends({});
+
+        await expect(
+            extendedDb.user.create({
+                data: {},
+            })
+        ).resolves.toEqual({
+            id: 2,
+            tenantId: tenant.id,
+        });
+    });
+});