feat: add "loadPath" options to runtime API and server adapter options (zenstackhq#696)

Author: ymc9

packages/plugins/swr/tests/swr.test.ts

@@ -58,7 +58,7 @@ ${sharedModel}
             {
                 provider: 'postgresql',
                 pushDb: false,
-                extraDependencies: [`${origDir}/dist`, 'react', '@types/react', 'swr'],
+                extraDependencies: [`${origDir}/dist`, 'react@18.2.0', '@types/react@18.2.0', 'swr@^2'],
                 compile: true,
             }
         );

packages/runtime/src/constants.ts

@@ -1,3 +1,8 @@
+/**
+ * Default path for loading CLI-generated code
+ */
+export const DEFAULT_RUNTIME_LOAD_PATH = '.zenstack';
+
 /**
  * Default length of password hash salt (used by bcryptjs to hash password)
  */

packages/runtime/src/enhancements/enhance.ts

@@ -1,4 +1,4 @@
-import { getDefaultModelMeta } from './model-meta';
+import { getDefaultModelMeta } from '../loader';
 import { withOmit, WithOmitOptions } from './omit';
 import { withPassword, WithPasswordOptions } from './password';
 import { withPolicy, WithPolicyContext, WithPolicyOptions } from './policy';
@@ -29,7 +29,7 @@ export function enhance<DbClient extends object>(
     let result = prisma;
 
     if (hasPassword === undefined || hasOmit === undefined) {
-        const modelMeta = options?.modelMeta ?? getDefaultModelMeta();
+        const modelMeta = options?.modelMeta ?? getDefaultModelMeta(options?.loadPath);
         const allFields = Object.values(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'));

packages/runtime/src/enhancements/index.ts

@@ -1,5 +1,5 @@
 export * from './model-meta';
-export * from './nested-write-vistor';
+export * from './nested-write-visitor';
 export * from './omit';
 export * from './password';
 export * from './policy';

packages/runtime/src/enhancements/model-meta.ts

@@ -1,29 +1,8 @@
 /* eslint-disable @typescript-eslint/no-var-requires */
 import { lowerCaseFirst } from 'lower-case-first';
-import path from 'path';
 import { FieldInfo } from '../types';
 import { ModelMeta } from './types';
 
-/**
- * Load model meta from standard location.
- */
-export function getDefaultModelMeta(): ModelMeta {
-    try {
-        // normal load
-        return require('.zenstack/model-meta').default;
-    } catch {
-        if (process.env.ZENSTACK_TEST === '1') {
-            try {
-                // special handling for running as tests, try resolving relative to CWD
-                return require(path.join(process.cwd(), 'node_modules', '.zenstack', 'model-meta')).default;
-            } catch {
-                throw new Error('Model meta cannot be loaded. Please make sure "zenstack generate" has been run.');
-            }
-        }
-        throw new Error('Model meta cannot be loaded. Please make sure "zenstack generate" has been run.');
-    }
-}
-
 /**
  * Resolves a model field to its metadata. Returns undefined if not found.
  */

packages/runtime/src/enhancements/nested-write-vistor.ts renamed to packages/runtime/src/enhancements/nested-write-visitor.ts

@@ -1,27 +1,28 @@
 /* eslint-disable @typescript-eslint/no-unused-vars */
 /* eslint-disable @typescript-eslint/no-explicit-any */
 
+import { getDefaultModelMeta } from '../loader';
 import { DbClientContract } from '../types';
-import { getDefaultModelMeta, resolveField } from './model-meta';
+import { resolveField } from './model-meta';
 import { DefaultPrismaProxyHandler, makeProxy } from './proxy';
-import { ModelMeta } from './types';
+import { CommonEnhancementOptions, ModelMeta } from './types';
 import { enumerate, getModelFields } from './utils';
 
 /**
  * Options for @see withOmit
  */
-export type WithOmitOptions = {
+export interface WithOmitOptions extends CommonEnhancementOptions {
     /**
      * Model metadata
      */
     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();
+    const _modelMeta = options?.modelMeta ?? getDefaultModelMeta(options?.loadPath);
     return makeProxy(
         prisma,
         _modelMeta,

packages/runtime/src/enhancements/omit.ts

@@ -3,27 +3,27 @@
 
 import { hash } from 'bcryptjs';
 import { DEFAULT_PASSWORD_SALT_LENGTH } from '../constants';
+import { getDefaultModelMeta } from '../loader';
 import { DbClientContract, PrismaWriteActionType } from '../types';
-import { getDefaultModelMeta } from './model-meta';
-import { NestedWriteVisitor } from './nested-write-vistor';
+import { NestedWriteVisitor } from './nested-write-visitor';
 import { DefaultPrismaProxyHandler, PrismaProxyActions, makeProxy } from './proxy';
-import { ModelMeta } from './types';
+import { CommonEnhancementOptions, ModelMeta } from './types';
 
 /**
  * Options for @see withPassword
  */
-export type WithPasswordOptions = {
+export interface WithPasswordOptions extends CommonEnhancementOptions {
     /**
-     * Model metatadata
+     * Model metadata
      */
     modelMeta?: ModelMeta;
-};
+}
 
 /**
  * Gets an enhanced Prisma client that supports @password attribute.
  */
 export function withPassword<DbClient extends object = any>(prisma: DbClient, options?: WithPasswordOptions): DbClient {
-    const _modelMeta = options?.modelMeta ?? getDefaultModelMeta();
+    const _modelMeta = options?.modelMeta ?? getDefaultModelMeta(options?.loadPath);
     return makeProxy(
         prisma,
         _modelMeta,

packages/runtime/src/enhancements/password.ts

@@ -7,7 +7,7 @@ import { CrudFailureReason, PRISMA_TX_FLAG } from '../../constants';
 import { AuthUser, DbClientContract, DbOperations, FieldInfo, PolicyOperationKind } from '../../types';
 import { ModelDataVisitor } from '../model-data-visitor';
 import { resolveField } from '../model-meta';
-import { NestedWriteVisitor, NestedWriteVisitorContext } from '../nested-write-vistor';
+import { NestedWriteVisitor, NestedWriteVisitorContext } from '../nested-write-visitor';
 import { PrismaProxyHandler } from '../proxy';
 import type { ModelMeta, PolicyDef, ZodSchemas } from '../types';
 import { enumerate, formatObject, getIdFields, prismaClientValidationError } from '../utils';

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

@@ -1,14 +1,13 @@
 /* eslint-disable @typescript-eslint/no-var-requires */
 /* eslint-disable @typescript-eslint/no-explicit-any */
 
-import path from 'path';
 import semver from 'semver';
 import { PRISMA_MINIMUM_VERSION } from '../../constants';
+import { getDefaultModelMeta, getDefaultPolicy, getDefaultZodSchemas } from '../../loader';
 import { AuthUser, DbClientContract } from '../../types';
 import { hasAllFields } from '../../validation';
-import { getDefaultModelMeta } from '../model-meta';
 import { makeProxy } from '../proxy';
-import type { ModelMeta, PolicyDef, ZodSchemas } from '../types';
+import type { CommonEnhancementOptions, ModelMeta, PolicyDef, ZodSchemas } from '../types';
 import { getIdFields } from '../utils';
 import { PolicyProxyHandler } from './handler';
 
@@ -22,7 +21,7 @@ export type WithPolicyContext = {
 /**
  * Options for @see withPolicy
  */
-export type WithPolicyOptions = {
+export interface WithPolicyOptions extends CommonEnhancementOptions {
     /**
      * Policy definition
      */
@@ -42,7 +41,7 @@ export type WithPolicyOptions = {
      * Whether to log Prisma query
      */
     logPrismaQuery?: boolean;
-};
+}
 
 /**
  * Gets an enhanced Prisma client with access policy check.
@@ -68,9 +67,9 @@ export function withPolicy<DbClient extends object>(
         );
     }
 
-    const _policy = options?.policy ?? getDefaultPolicy();
-    const _modelMeta = options?.modelMeta ?? getDefaultModelMeta();
-    const _zodSchemas = options?.zodSchemas ?? getDefaultZodSchemas();
+    const _policy = options?.policy ?? getDefaultPolicy(options?.loadPath);
+    const _modelMeta = options?.modelMeta ?? getDefaultModelMeta(options?.loadPath);
+    const _zodSchemas = options?.zodSchemas ?? getDefaultZodSchemas(options?.loadPath);
 
     // validate user context
     if (context?.user) {
@@ -103,40 +102,3 @@ export function withPolicy<DbClient extends object>(
         'policy'
     );
 }
-
-function getDefaultPolicy(): PolicyDef {
-    try {
-        return require('.zenstack/policy').default;
-    } catch {
-        if (process.env.ZENSTACK_TEST === '1') {
-            try {
-                // special handling for running as tests, try resolving relative to CWD
-                return require(path.join(process.cwd(), 'node_modules', '.zenstack', 'policy')).default;
-            } catch {
-                throw new Error(
-                    'Policy definition cannot be loaded from default location. Please make sure "zenstack generate" has been run.'
-                );
-            }
-        }
-        throw new Error(
-            'Policy definition cannot be loaded from default location. Please make sure "zenstack generate" has been run.'
-        );
-    }
-}
-
-function getDefaultZodSchemas(): ZodSchemas | undefined {
-    try {
-        // eslint-disable-next-line @typescript-eslint/no-var-requires
-        return require('.zenstack/zod');
-    } catch {
-        if (process.env.ZENSTACK_TEST === '1') {
-            try {
-                // special handling for running as tests, try resolving relative to CWD
-                return require(path.join(process.cwd(), 'node_modules', '.zenstack', 'zod'));
-            } catch {
-                return undefined;
-            }
-        }
-        return undefined;
-    }
-}