prisma-flare 1.3.0 → 1.3.2

package/dist/cli/db-reset.cjs

@@ -233,11 +233,32 @@ function getPrismaProvider(rootDir) {
 }
 function parseModelRelations(schemaContent) {
   const models = [];
-  const modelRegex = /model\s+(\w+)\s*\{([^}]+)\}/g;
+  const modelStartRegex = /model\s+(\w+)\s*\{/g;
   let modelMatch;
-  while ((modelMatch = modelRegex.exec(schemaContent)) !== null) {
+  while ((modelMatch = modelStartRegex.exec(schemaContent)) !== null) {
     const modelName = modelMatch[1];
-    const modelBody = modelMatch[2];
+    const bodyStart = modelMatch.index + modelMatch[0].length;
+    let depth = 1;
+    let i = bodyStart;
+    let inString = false;
+    let stringChar = "";
+    while (i < schemaContent.length && depth > 0) {
+      const ch = schemaContent[i];
+      if (inString) {
+        if (ch === stringChar && schemaContent[i - 1] !== "\\") inString = false;
+      } else {
+        if (ch === '"' || ch === "'") {
+          inString = true;
+          stringChar = ch;
+        } else if (ch === "{") {
+          depth++;
+        } else if (ch === "}") {
+          depth--;
+        }
+      }
+      i++;
+    }
+    const modelBody = schemaContent.slice(bodyStart, i - 1);
     const relations = [];
     const lines = modelBody.split("\n");
     for (const line of lines) {
@@ -297,13 +318,33 @@ ${entries.join(",\n")}
 };`;
 }
 function getRelationModelMap(rootDir) {
+  const dmmfResult = getRelationModelMapFromDMMF(rootDir);
+  if (dmmfResult) return dmmfResult;
   const resolution = resolveSchemaPath(rootDir);
-  if (!resolution) {
-    return null;
-  }
+  if (!resolution) return null;
   const models = parseModelRelations(resolution.content);
   return generateRelationModelMap(models);
 }
+function getRelationModelMapFromDMMF(rootDir) {
+  try {
+    const clientPath = getPrismaClientPath(rootDir);
+    const resolvedPath = clientPath.startsWith("/") ? clientPath : require.resolve(clientPath, { paths: [rootDir] });
+    const prismaModule = require(resolvedPath);
+    const dmmf = prismaModule?.Prisma?.dmmf ?? prismaModule?.dmmf;
+    if (!dmmf?.datamodel?.models) return null;
+    const models = dmmf.datamodel.models.map((model) => ({
+      name: model.name,
+      relations: model.fields.filter((f) => f.kind === "object").map((f) => ({
+        fieldName: f.name,
+        targetModel: f.type,
+        isArray: f.isList
+      }))
+    }));
+    return generateRelationModelMap(models);
+  } catch {
+    return null;
+  }
+}
 
 // src/cli/templates/type-helpers.ts
 function generateTypeHelpers(options = {}) {
@@ -549,6 +590,11 @@ ${exportKeyword}type ColumnChangeCallback<T extends ModelName = ModelName> = (
 /**
  * Options for column change hooks (afterChange)
  */
+${exportKeyword}interface HookOptions {
+  /** Tag to group hooks. Tagged hooks can be disabled/enabled via hookRegistry.disable(tag)/enable(tag). */
+  tag?: string;
+}
+
 ${exportKeyword}interface ColumnChangeOptions<T extends ModelName = ModelName> {
   /**
    * Additional fields to include when fetching records for this hook.
@@ -556,6 +602,8 @@ ${exportKeyword}interface ColumnChangeOptions<T extends ModelName = ModelName> {
    * Use this when your callback needs access to other fields.
    */
   includeFields?: FieldName<T>[];
+  /** Tag to group hooks. Tagged hooks can be disabled/enabled via hookRegistry.disable(tag)/enable(tag). */
+  tag?: string;
 }
 `.trimStart();
 }
@@ -565,53 +613,53 @@ var FLARE_BUILDER_METHODS = {
   whereConditions: [
     {
       name: "where",
-      signature: "(condition: WhereInput<T>): FlareBuilder<T, Args & { where: WhereInput<T> }>"
+      signature: "(condition: WhereInput<T>): this"
     },
     {
       name: "andWhere",
-      signature: "(condition: WhereInput<T>): FlareBuilder<T, Args & { where: WhereInput<T> }>"
+      signature: "(condition: WhereInput<T>): this"
     },
     {
       name: "orWhere",
-      signature: "(condition: WhereInput<T>): FlareBuilder<T, Args & { where: WhereInput<T> }>"
+      signature: "(condition: WhereInput<T>): this"
     },
     {
       name: "whereGroup",
-      signature: "(callback: (builder: FlareBuilder<T, Record<string, never>>) => FlareBuilder<T, any>, mode?: 'AND' | 'OR'): FlareBuilder<T, Args & { where: WhereInput<T> }>"
+      signature: "(callback: (builder: FlareBuilder<T, Record<string, never>>) => FlareBuilder<T, any>, mode?: 'AND' | 'OR'): this"
     },
     {
       name: "orWhereGroup",
-      signature: "(callback: (builder: FlareBuilder<T, Record<string, never>>) => FlareBuilder<T, any>): FlareBuilder<T, Args & { where: WhereInput<T> }>"
+      signature: "(callback: (builder: FlareBuilder<T, Record<string, never>>) => FlareBuilder<T, any>): this"
     },
     {
       name: "withId",
-      signature: "(id: number | string): FlareBuilder<T, Args & { where: { id: number | string } }>"
+      signature: "(id: number | string): this"
     }
   ],
   orderingAndLimiting: [
     {
       name: "order",
-      signature: "(orderBy: OrderByInput<T>): FlareBuilder<T, Args & { orderBy: OrderByInput<T> }>"
+      signature: "(orderBy: OrderByInput<T>): this"
     },
     {
       name: "first",
-      signature: "(key?: keyof RecordType<T> | string): FlareBuilder<T, Args & { orderBy: any; take: number }>"
+      signature: "(key?: keyof RecordType<T> | string): this"
     },
     {
       name: "last",
-      signature: "(key?: keyof RecordType<T> | string): FlareBuilder<T, Args & { orderBy: any; take: number }>"
+      signature: "(key?: keyof RecordType<T> | string): this"
     },
     {
       name: "limit",
-      signature: "(count: number): FlareBuilder<T, Args & { take: number }>"
+      signature: "(count: number): this"
     },
     {
       name: "skip",
-      signature: "(count: number): FlareBuilder<T, Args & { skip: number }>"
+      signature: "(count: number): this"
     },
     {
       name: "distinct",
-      signature: "(fields: DistinctInput<T>): FlareBuilder<T, Args & { distinct: DistinctInput<T> }>"
+      signature: "(fields: DistinctInput<T>): this"
     }
   ],
   selection: [
@@ -723,7 +771,7 @@ var FLARE_BUILDER_METHODS = {
   pagination: [
     {
       name: "paginate",
-      signature: "(page?: number, perPage?: number): Promise<PaginatedResult<RecordType<T>>>"
+      signature: (ns, useFlareResult) => useFlareResult ? `(page?: number, perPage?: number): Promise<PaginatedResult<FlareResult<T, Args>>>` : `(page?: number, perPage?: number): Promise<PaginatedResult<${ns}.Result<ModelDelegate<T>, Args, 'findFirstOrThrow'>>>`
     }
   ],
   existence: [
@@ -735,11 +783,11 @@ var FLARE_BUILDER_METHODS = {
   utilities: [
     {
       name: "when",
-      signature: "(condition: boolean | (() => boolean), callback: (qb: FlareBuilder<T, Args>) => void): FlareBuilder<T, Args>"
+      signature: "(condition: boolean | (() => boolean), callback: (qb: this) => void): this"
     },
     {
       name: "chunk",
-      signature: "(size: number, callback: (results: RecordType<T>[]) => Promise<void> | void): Promise<void>"
+      signature: (ns, useFlareResult) => useFlareResult ? `(size: number, callback: (results: FlareResultMany<T, Args>) => Promise<void> | void): Promise<void>` : `(size: number, callback: (results: ${ns}.Result<ModelDelegate<T>, Args, 'findMany'>) => Promise<void> | void): Promise<void>`
     },
     {
       name: "clone",
@@ -938,9 +986,10 @@ import {
  */
 export function beforeCreate<T extends ModelName>(
   model: T,
-  callback: BeforeHookCallback<T>
+  callback: BeforeHookCallback<T>,
+  options?: HookOptions
 ): void {
-  _beforeCreate(model as any, callback as any);
+  _beforeCreate(model as any, callback as any, options);
 }
 
 /**
@@ -948,9 +997,10 @@ export function beforeCreate<T extends ModelName>(
  */
 export function afterCreate<T extends ModelName>(
   model: T,
-  callback: AfterHookCallback<T>
+  callback: AfterHookCallback<T>,
+  options?: HookOptions
 ): void {
-  _afterCreate(model as any, callback as any);
+  _afterCreate(model as any, callback as any, options);
 }
 
 /**
@@ -958,9 +1008,10 @@ export function afterCreate<T extends ModelName>(
  */
 export function beforeUpdate<T extends ModelName>(
   model: T,
-  callback: BeforeHookCallback<T>
+  callback: BeforeHookCallback<T>,
+  options?: HookOptions
 ): void {
-  _beforeUpdate(model as any, callback as any);
+  _beforeUpdate(model as any, callback as any, options);
 }
 
 /**
@@ -968,9 +1019,10 @@ export function beforeUpdate<T extends ModelName>(
  */
 export function afterUpdate<T extends ModelName>(
   model: T,
-  callback: AfterHookCallback<T>
+  callback: AfterHookCallback<T>,
+  options?: HookOptions
 ): void {
-  _afterUpdate(model as any, callback as any);
+  _afterUpdate(model as any, callback as any, options);
 }
 
 /**
@@ -978,9 +1030,10 @@ export function afterUpdate<T extends ModelName>(
  */
 export function beforeDelete<T extends ModelName>(
   model: T,
-  callback: BeforeHookCallback<T>
+  callback: BeforeHookCallback<T>,
+  options?: HookOptions
 ): void {
-  _beforeDelete(model as any, callback as any);
+  _beforeDelete(model as any, callback as any, options);
 }
 
 /**
@@ -988,9 +1041,10 @@ export function beforeDelete<T extends ModelName>(
  */
 export function afterDelete<T extends ModelName>(
   model: T,
-  callback: AfterHookCallback<T>
+  callback: AfterHookCallback<T>,
+  options?: HookOptions
 ): void {
-  _afterDelete(model as any, callback as any);
+  _afterDelete(model as any, callback as any, options);
 }
 
 /**
@@ -1010,9 +1064,10 @@ export function afterChange<T extends ModelName>(
  */
 export function afterUpsert<T extends ModelName>(
   model: T,
-  callback: AfterHookCallback<T>
+  callback: AfterHookCallback<T>,
+  options?: HookOptions
 ): void {
-  _afterUpsert(model as any, callback as any);
+  _afterUpsert(model as any, callback as any, options);
 }
 
 // Re-export hookRegistry for advanced use cases
@@ -1092,7 +1147,8 @@ import {
  */
 export declare function beforeCreate<T extends ModelName>(
   model: T,
-  callback: BeforeHookCallback<T>
+  callback: BeforeHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 /**
@@ -1100,7 +1156,8 @@ export declare function beforeCreate<T extends ModelName>(
  */
 export declare function afterCreate<T extends ModelName>(
   model: T,
-  callback: AfterHookCallback<T>
+  callback: AfterHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 /**
@@ -1108,7 +1165,8 @@ export declare function afterCreate<T extends ModelName>(
  */
 export declare function beforeUpdate<T extends ModelName>(
   model: T,
-  callback: BeforeHookCallback<T>
+  callback: BeforeHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 /**
@@ -1116,7 +1174,8 @@ export declare function beforeUpdate<T extends ModelName>(
  */
 export declare function afterUpdate<T extends ModelName>(
   model: T,
-  callback: AfterHookCallback<T>
+  callback: AfterHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 /**
@@ -1124,7 +1183,8 @@ export declare function afterUpdate<T extends ModelName>(
  */
 export declare function beforeDelete<T extends ModelName>(
   model: T,
-  callback: BeforeHookCallback<T>
+  callback: BeforeHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 /**
@@ -1132,7 +1192,8 @@ export declare function beforeDelete<T extends ModelName>(
  */
 export declare function afterDelete<T extends ModelName>(
   model: T,
-  callback: AfterHookCallback<T>
+  callback: AfterHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 /**
@@ -1150,7 +1211,8 @@ export declare function afterChange<T extends ModelName>(
  */
 export declare function afterUpsert<T extends ModelName>(
   model: T,
-  callback: AfterHookCallback<T>
+  callback: AfterHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 // Re-export hookRegistry for advanced use cases
@@ -1274,6 +1336,7 @@ import type {
   AfterHookCallback,
   ColumnChangeCallback,
   ColumnChangeOptions,
+  HookOptions,
   FieldName,
   HookConfig
 } from 'prisma-flare';
@@ -1322,7 +1385,8 @@ export declare class FlareClient extends BasePrismaClient {
  */
 export declare function beforeCreate<T extends ModelName>(
   model: T,
-  callback: BeforeHookCallback<T>
+  callback: BeforeHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 /**
@@ -1330,7 +1394,8 @@ export declare function beforeCreate<T extends ModelName>(
  */
 export declare function afterCreate<T extends ModelName>(
   model: T,
-  callback: AfterHookCallback<T>
+  callback: AfterHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 /**
@@ -1338,7 +1403,8 @@ export declare function afterCreate<T extends ModelName>(
  */
 export declare function beforeUpdate<T extends ModelName>(
   model: T,
-  callback: BeforeHookCallback<T>
+  callback: BeforeHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 /**
@@ -1346,7 +1412,8 @@ export declare function beforeUpdate<T extends ModelName>(
  */
 export declare function afterUpdate<T extends ModelName>(
   model: T,
-  callback: AfterHookCallback<T>
+  callback: AfterHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 /**
@@ -1354,7 +1421,8 @@ export declare function afterUpdate<T extends ModelName>(
  */
 export declare function beforeDelete<T extends ModelName>(
   model: T,
-  callback: BeforeHookCallback<T>
+  callback: BeforeHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 /**
@@ -1362,7 +1430,8 @@ export declare function beforeDelete<T extends ModelName>(
  */
 export declare function afterDelete<T extends ModelName>(
   model: T,
-  callback: AfterHookCallback<T>
+  callback: AfterHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 /**
@@ -1380,7 +1449,8 @@ export declare function afterChange<T extends ModelName>(
  */
 export declare function afterUpsert<T extends ModelName>(
   model: T,
-  callback: AfterHookCallback<T>
+  callback: AfterHookCallback<T>,
+  options?: HookOptions
 ): void;
 
 // Re-export hookRegistry for advanced use cases