refactor(runtime): separate node and edge enhancement code for edge-runtime compatibility

packages/plugins/trpc/tests/projects/t3-trpc-v10/package.json

@@ -36,7 +36,7 @@
     "eslint": "^8.54.0",
     "eslint-config-next": "^14.0.4",
     "prisma": "^5.13.0",
-    "typescript": "^5.1.6"
+    "typescript": "^5.5.4"
   },
   "ct3aMetadata": {
     "initVersion": "7.26.0"

packages/plugins/trpc/tests/projects/t3-trpc-v10/src/pages/index.tsx

@@ -4,19 +4,23 @@ import styles from './index.module.css';
 export default function Home() {
     const hello = api.greet.hello.useQuery({ text: 'from tRPC' });
     const posts = api.post.findMany.useQuery({ where: { published: true }, include: { author: true } });
-    const postsTransformed = api.post.findMany.useQuery({}, { select: (data) => data.map((p) => ({ title: p.name })) });
+    const postsTransformed = api.post.findMany.useQuery(
+        {},
+        { select: (data) => data.map((p) => ({ id: p.id, title: p.name })) }
+    );
 
     return (
         <>
             <main className={styles.main}>
                 {hello.data && <h1 className={styles.title}>{hello.data.greeting}</h1>}
-                {posts.data &&
-                    posts.data.map((post) => (
-                        <p>
-                            {post.name} by {post.author.email}
-                        </p>
-                    ))}
-                {postsTransformed.data && postsTransformed.data.map((post) => <p>{post.title}</p>)}
+                {posts.data?.map((post) => (
+                    <p key={post.id}>
+                        {post.name} by {post.author.email}
+                    </p>
+                ))}
+                {postsTransformed.data?.map((post) => (
+                    <p key={post.id}>{post.title}</p>
+                ))}
             </main>
         </>
     );

packages/runtime/package.json

@@ -25,9 +25,17 @@
             "types": "./edge.d.ts",
             "default": "./edge.js"
         },
-        "./enhancements": {
-            "types": "./enhancements/index.d.ts",
-            "default": "./enhancements/index.js"
+        "./enhancements/node": {
+            "types": "./enhancements/node/index.d.ts",
+            "default": "./enhancements/node/index.js"
+        },
+        "./enhancements/edge": {
+            "types": "./enhancements/edge/index.d.ts",
+            "default": "./enhancements/edge/index.js"
+        },
+        "./validation": {
+            "types": "./validation.d.ts",
+            "default": "./validation.js"
         },
         "./constraint-solver": {
             "types": "./constraint-solver.d.ts",

packages/runtime/res/enhance-edge.d.ts

@@ -0,0 +1 @@
+export { auth, enhance, type PrismaClient } from '.zenstack/enhance-edge';

packages/runtime/res/enhance-edge.js

@@ -0,0 +1,10 @@
+'use strict';
+Object.defineProperty(exports, '__esModule', { value: true });
+
+try {
+    exports.enhance = require('.zenstack/enhance-edge').enhance;
+} catch {
+    exports.enhance = function () {
+        throw new Error('Generated "enhance" function not found. Please run `zenstack generate` first.');
+    };
+}

packages/runtime/src/edge.ts

@@ -0,0 +1 @@
+export * from './enhance-edge';

packages/runtime/src/enhance-edge.d.ts

@@ -0,0 +1,2 @@
+// @ts-expect-error stub for re-exporting generated code
+export { auth, enhance } from '.zenstack/enhance-edge';

packages/runtime/src/enhancements/edge/create-enhancement.ts

@@ -0,0 +1 @@
+../node/create-enhancement.ts

packages/runtime/src/enhancements/edge/default-auth.ts

@@ -0,0 +1 @@
+../node/default-auth.ts

packages/runtime/src/enhancements/edge/delegate.ts

@@ -0,0 +1 @@
+../node/delegate.ts

packages/runtime/src/enhancements/edge/index.ts

@@ -0,0 +1 @@
+../node/index.ts

packages/runtime/src/enhancements/edge/logger.ts

@@ -0,0 +1 @@
+../node/logger.ts

packages/runtime/src/enhancements/edge/omit.ts

@@ -0,0 +1 @@
+../node/omit.ts

packages/runtime/src/enhancements/edge/password.ts

@@ -0,0 +1 @@
+../node/password.ts

packages/runtime/src/enhancements/edge/policy/check-utils.ts

@@ -0,0 +1,16 @@
+import { ModelMeta } from '..';
+import type { DbClientContract } from '../../../types';
+import { PermissionCheckArgs } from '../types';
+import { PolicyUtil } from './policy-utils';
+
+export async function checkPermission(
+    _model: string,
+    _args: PermissionCheckArgs,
+    _modelMeta: ModelMeta,
+    _policyUtils: PolicyUtil,
+    _prisma: DbClientContract,
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    _prismaModule: any
+): Promise<boolean> {
+    throw new Error('`check()` API is not supported on edge runtime');
+}

packages/runtime/src/enhancements/edge/policy/handler.ts

@@ -0,0 +1 @@
+../../node/policy/handler.ts

packages/runtime/src/enhancements/edge/policy/index.ts

@@ -0,0 +1 @@
+../../node/policy/index.ts

packages/runtime/src/enhancements/edge/policy/policy-utils.ts

@@ -0,0 +1 @@
+../../node/policy/policy-utils.ts

packages/runtime/src/enhancements/edge/promise.ts

@@ -0,0 +1 @@
+../node/promise.ts

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

@@ -0,0 +1 @@
+../node/proxy.ts

packages/runtime/src/enhancements/edge/query-utils.ts

@@ -0,0 +1 @@
+../node/query-utils.ts

packages/runtime/src/enhancements/edge/types.ts

@@ -0,0 +1 @@
+../node/types.ts

packages/runtime/src/enhancements/edge/utils.ts

@@ -0,0 +1 @@
+../node/utils.ts

packages/runtime/src/enhancements/edge/where-visitor.ts

@@ -0,0 +1 @@
+../node/where-visitor.ts

packages/runtime/src/enhancements/create-enhancement.ts → packages/runtime/src/enhancements/node/create-enhancement.ts

@@ -1,68 +1,20 @@
 import semver from 'semver';
-import { PRISMA_MINIMUM_VERSION } from '../constants';
-import { isDelegateModel, type ModelMeta } from '../cross';
-import type { AuthUser } from '../types';
+import { PRISMA_MINIMUM_VERSION } from '../../constants';
+import { isDelegateModel, type ModelMeta } from '../../cross';
+import type { EnhancementContext, EnhancementKind, EnhancementOptions, ZodSchemas } from '../../types';
 import { withDefaultAuth } from './default-auth';
 import { withDelegate } from './delegate';
 import { Logger } from './logger';
 import { withOmit } from './omit';
 import { withPassword } from './password';
 import { withPolicy } from './policy';
-import type { ErrorTransformer } from './proxy';
-import type { PolicyDef, ZodSchemas } from './types';
-
-/**
- * Kinds of enhancements to `PrismaClient`
- */
-export type EnhancementKind = 'password' | 'omit' | 'policy' | 'validation' | 'delegate';
+import type { PolicyDef } from './types';
 
 /**
  * All enhancement kinds
  */
 const ALL_ENHANCEMENTS: EnhancementKind[] = ['password', 'omit', 'policy', 'validation', 'delegate'];
 
-/**
- * Transaction isolation levels: https://www.prisma.io/docs/orm/prisma-client/queries/transactions#transaction-isolation-level
- */
-export type TransactionIsolationLevel =
-    | 'ReadUncommitted'
-    | 'ReadCommitted'
-    | 'RepeatableRead'
-    | 'Snapshot'
-    | 'Serializable';
-
-export type EnhancementOptions = {
-    /**
-     * The kinds of enhancements to apply. By default all enhancements are applied.
-     */
-    kinds?: EnhancementKind[];
-
-    /**
-     * Whether to log Prisma query
-     */
-    logPrismaQuery?: boolean;
-
-    /**
-     * Hook for transforming errors before they are thrown to the caller.
-     */
-    errorTransformer?: ErrorTransformer;
-
-    /**
-     * The `maxWait` option passed to `prisma.$transaction()` call for transactions initiated by ZenStack.
-     */
-    transactionMaxWait?: number;
-
-    /**
-     * The `timeout` option passed to `prisma.$transaction()` call for transactions initiated by ZenStack.
-     */
-    transactionTimeout?: number;
-
-    /**
-     * The `isolationLevel` option passed to `prisma.$transaction()` call for transactions initiated by ZenStack.
-     */
-    transactionIsolationLevel?: TransactionIsolationLevel;
-};
-
 /**
  * Options for {@link createEnhancement}
  *
@@ -91,13 +43,6 @@ export type InternalEnhancementOptions = EnhancementOptions & {
     prismaModule: any;
 };
 
-/**
- * Context for creating enhanced `PrismaClient`
- */
-export type EnhancementContext<User extends AuthUser = AuthUser> = {
-    user?: User;
-};
-
 /**
  * Gets a Prisma client enhanced with all enhancement behaviors, including access
  * policy, field validation, field omission and password hashing.

packages/runtime/src/enhancements/default-auth.ts → packages/runtime/src/enhancements/node/default-auth.ts

@@ -9,9 +9,9 @@ import {
     enumerate,
     getFields,
     requireField,
-} from '../cross';
-import { DbClientContract } from '../types';
-import { EnhancementContext, InternalEnhancementOptions } from './create-enhancement';
+} from '../../cross';
+import { DbClientContract, EnhancementContext } from '../../types';
+import { InternalEnhancementOptions } from './create-enhancement';
 import { DefaultPrismaProxyHandler, PrismaProxyActions, makeProxy } from './proxy';
 import { isUnsafeMutate, prismaClientValidationError } from './utils';
 

packages/runtime/src/enhancements/delegate.ts → packages/runtime/src/enhancements/node/delegate.ts

@@ -2,7 +2,7 @@
 import deepmerge, { type ArrayMergeOptions } from 'deepmerge';
 import { isPlainObject } from 'is-plain-object';
 import { lowerCaseFirst } from 'lower-case-first';
-import { DELEGATE_AUX_RELATION_PREFIX } from '../constants';
+import { DELEGATE_AUX_RELATION_PREFIX } from '../../constants';
 import {
     FieldInfo,
     ModelInfo,
@@ -13,8 +13,8 @@ import {
     getModelInfo,
     isDelegateModel,
     resolveField,
-} from '../cross';
-import type { CrudContract, DbClientContract } from '../types';
+} from '../../cross';
+import type { CrudContract, DbClientContract } from '../../types';
 import type { InternalEnhancementOptions } from './create-enhancement';
 import { Logger } from './logger';
 import { DefaultPrismaProxyHandler, makeProxy } from './proxy';

packages/runtime/src/enhancements/index.ts → packages/runtime/src/enhancements/node/index.ts

@@ -1,4 +1,4 @@
-export * from '../cross';
+export * from '../../cross';
 export * from './create-enhancement';
 export * from './types';
 export * from './utils';

packages/runtime/src/enhancements/omit.ts → packages/runtime/src/enhancements/node/omit.ts

@@ -1,8 +1,8 @@
 /* eslint-disable @typescript-eslint/no-unused-vars */
 /* eslint-disable @typescript-eslint/no-explicit-any */
 
-import { enumerate, getModelFields, resolveField } from '../cross';
-import { DbClientContract } from '../types';
+import { enumerate, getModelFields, resolveField } from '../../cross';
+import { DbClientContract } from '../../types';
 import { InternalEnhancementOptions } from './create-enhancement';
 import { DefaultPrismaProxyHandler, makeProxy } from './proxy';
 

packages/runtime/src/enhancements/password.ts → packages/runtime/src/enhancements/node/password.ts

@@ -1,9 +1,9 @@
 /* eslint-disable @typescript-eslint/no-explicit-any */
 /* eslint-disable @typescript-eslint/no-unused-vars */
 
-import { DEFAULT_PASSWORD_SALT_LENGTH } from '../constants';
-import { NestedWriteVisitor, type PrismaWriteActionType } from '../cross';
-import { DbClientContract } from '../types';
+import { DEFAULT_PASSWORD_SALT_LENGTH } from '../../constants';
+import { NestedWriteVisitor, type PrismaWriteActionType } from '../../cross';
+import { DbClientContract } from '../../types';
 import { InternalEnhancementOptions } from './create-enhancement';
 import { DefaultPrismaProxyHandler, PrismaProxyActions, makeProxy } from './proxy';