refactor: generate enhance API and avoid dynamic require in runtime (#937)

Author: ymc9

.github/workflows/build-test.yml

@@ -9,7 +9,7 @@ env:
 
 on:
     pull_request:
-        branches: ['dev', 'main']
+        branches: ['dev', 'main', 'v2']
 
 jobs:
     build-test:
@@ -32,18 +32,11 @@ jobs:
         strategy:
             matrix:
                 node-version: [18.x]
-                prisma-version: [v4, v5]
 
         steps:
             - name: Checkout
               uses: actions/checkout@v3
 
-            - name: Set Prisma Version
-              if: ${{ matrix.prisma-version == 'v5' }}
-              shell: bash
-              run: |
-                  bash ./script/test-prisma-v5.sh
-
             - name: Install pnpm
               uses: pnpm/action-setup@v2
               with:

packages/plugins/tanstack-query/tsconfig.json

@@ -6,5 +6,5 @@
         "jsx": "react"
     },
     "include": ["src/**/*.ts"],
-    "exclude": ["src/runtime"]
+    "exclude": ["src/runtime", "src/runtime-v5"]
 }

packages/runtime/package.json

@@ -53,11 +53,9 @@
         "linkDirectory": true
     },
     "dependencies": {
-        "@types/bcryptjs": "^2.4.2",
         "bcryptjs": "^2.4.3",
         "buffer": "^6.0.3",
         "change-case": "^4.1.2",
-        "colors": "1.4.0",
         "decimal.js": "^10.4.2",
         "deepcopy": "^2.1.0",
         "lower-case-first": "^2.0.2",

packages/runtime/res/enhance.d.ts

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

packages/runtime/res/enhance.js

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

packages/runtime/src/constants.ts

@@ -66,7 +66,7 @@ export const PRISMA_PROXY_ENHANCER = '$__zenstack_enhancer';
 /**
  * Minimum Prisma version supported
  */
-export const PRISMA_MINIMUM_VERSION = '4.8.0';
+export const PRISMA_MINIMUM_VERSION = '5.0.0';
 
 /**
  * Selector function name for fetching pre-update entity values.

packages/runtime/src/enhance.d.ts

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

packages/runtime/src/enhancements/enhance.ts

@@ -1,4 +1,3 @@
-import { getDefaultModelMeta } from '../loader';
 import { withOmit, WithOmitOptions } from './omit';
 import { withPassword, WithPasswordOptions } from './password';
 import { withPolicy, WithPolicyContext, WithPolicyOptions } from './policy';
@@ -21,16 +20,15 @@ let hasOmit: boolean | undefined = undefined;
  * @param context The context to for evaluating access policies.
  * @param options Options.
  */
-export function enhance<DbClient extends object>(
+export function createEnhancement<DbClient extends object>(
     prisma: DbClient,
-    context?: WithPolicyContext,
-    options?: EnhancementOptions
+    options: EnhancementOptions,
+    context?: WithPolicyContext
 ) {
     let result = prisma;
 
     if (hasPassword === undefined || hasOmit === undefined) {
-        const modelMeta = options?.modelMeta ?? getDefaultModelMeta(options?.loadPath);
-        const allFields = Object.values(modelMeta.fields).flatMap((modelInfo) => Object.values(modelInfo));
+        const allFields = Object.values(options.modelMeta.fields).flatMap((modelInfo) => Object.values(modelInfo));
         hasPassword = allFields.some((field) => field.attributes?.some((attr) => attr.name === '@password'));
         hasOmit = allFields.some((field) => field.attributes?.some((attr) => attr.name === '@omit'));
     }
@@ -46,7 +44,7 @@ export function enhance<DbClient extends object>(
     }
 
     // policy proxy
-    result = withPolicy(result, context, options);
+    result = withPolicy(result, options, context);
 
     return result;
 }

packages/runtime/src/enhancements/index.ts

@@ -3,7 +3,6 @@ export * from './enhance';
 export * from './omit';
 export * from './password';
 export * from './policy';
-export * from './preset';
 export * from './types';
 export * from './utils';
 export * from './where-visitor';

packages/runtime/src/enhancements/omit.ts

@@ -2,7 +2,6 @@
 /* eslint-disable @typescript-eslint/no-explicit-any */
 
 import { enumerate, getModelFields, resolveField, type ModelMeta } from '../cross';
-import { getDefaultModelMeta } from '../loader';
 import { DbClientContract } from '../types';
 import { DefaultPrismaProxyHandler, makeProxy } from './proxy';
 import { CommonEnhancementOptions } from './types';
@@ -14,18 +13,17 @@ export interface WithOmitOptions extends CommonEnhancementOptions {
     /**
      * Model metadata
      */
-    modelMeta?: ModelMeta;
+    modelMeta: ModelMeta;
 }
 
 /**
  * Gets an enhanced Prisma client that supports @omit attribute.
  */
-export function withOmit<DbClient extends object>(prisma: DbClient, options?: WithOmitOptions): DbClient {
-    const _modelMeta = options?.modelMeta ?? getDefaultModelMeta(options?.loadPath);
+export function withOmit<DbClient extends object>(prisma: DbClient, options: WithOmitOptions): DbClient {
     return makeProxy(
         prisma,
-        _modelMeta,
-        (_prisma, model) => new OmitHandler(_prisma as DbClientContract, model, _modelMeta),
+        options.modelMeta,
+        (_prisma, model) => new OmitHandler(_prisma as DbClientContract, model, options.modelMeta),
         'omit'
     );
 }