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

Author: ymc9

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/edge.ts

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

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

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

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

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