prisma-generator-express 1.40.0 → 1.41.0

package/src/generators/generateRouteConfigType.ts

@@ -1,26 +1,22 @@
+import { ImportStyle } from '../utils/resolveImportStyle'
+import { importExt } from '../utils/importExt'
+import type { Target } from '../constants'
+
 const ROUTER_OPERATIONS = [
-  'findUnique',
-  'findUniqueOrThrow',
-  'findFirst',
-  'findFirstOrThrow',
-  'findMany',
-  'findManyPaginated',
-  'count',
-  'aggregate',
-  'groupBy',
-  'create',
-  'createMany',
-  'createManyAndReturn',
-  'update',
-  'updateMany',
-  'updateManyAndReturn',
-  'upsert',
-  'delete',
-  'deleteMany',
+  'findUnique', 'findUniqueOrThrow', 'findFirst', 'findFirstOrThrow',
+  'findMany', 'findManyPaginated', 'count', 'aggregate', 'groupBy',
+  'create', 'createMany', 'createManyAndReturn',
+  'update', 'updateMany', 'updateManyAndReturn',
+  'upsert', 'delete', 'deleteMany',
 ] as const
 
 type RouterOperation = (typeof ROUTER_OPERATIONS)[number]
 
+const READ_OPERATIONS: ReadonlySet<RouterOperation> = new Set<RouterOperation>([
+  'findUnique', 'findUniqueOrThrow', 'findFirst', 'findFirstOrThrow',
+  'findMany', 'findManyPaginated', 'count', 'aggregate', 'groupBy',
+])
+
 const ROUTER_OP_TO_SHAPE_OP: Record<RouterOperation, string> = {
   findUnique: 'findUnique',
   findUniqueOrThrow: 'findUniqueOrThrow',
@@ -50,40 +46,51 @@ export function generateRouteConfigType(
   modelName: string,
   hookHandlerType: string,
   guardShapesImport: string | null,
+  importStyle: ImportStyle,
+  target: Target,
 ): string {
+  const ext = importExt(importStyle)
   const m = modelName
+  const supportsProgressive = target === 'express'
 
   if (!guardShapesImport) {
-    return `export type ${m}RouteConfig<TCtx = unknown> = RouteConfig\n`
+    return `export type ${m}RouteConfig<TCtx = unknown> = RouteConfig<Record<string, any>, TCtx>\n`
   }
 
-  const shapeOps = Object.values(ROUTER_OP_TO_SHAPE_OP).filter(
-    (v, i, a) => a.indexOf(v) === i,
-  )
-
-  const opShapeImports = shapeOps
-    .map((op) => `${m}${capitalize(op)}ShapeInput`)
-    .join(',\n  ')
+  const shapeOps = Object.values(ROUTER_OP_TO_SHAPE_OP).filter((v, i, a) => a.indexOf(v) === i)
+  const opShapeImports = shapeOps.map((op) => `${m}${capitalize(op)}ShapeInput`).join(',\n  ')
 
   const overrides = ROUTER_OPERATIONS.map((routerOp) => {
     const shapeOp = ROUTER_OP_TO_SHAPE_OP[routerOp]
     const c = capitalize(shapeOp)
-    return (
-      `  ${routerOp}?: {\n` +
-      `    before?: ${hookHandlerType}[]\n` +
-      `    after?: ${hookHandlerType}[]\n` +
-      `    shape?: ${m}${c}ShapeInput<TCtx>\n` +
-      `  }`
-    )
+    const isRead = READ_OPERATIONS.has(routerOp)
+    const lines = [
+      `    before?: ${hookHandlerType}[]`,
+      `    after?: ${hookHandlerType}[]`,
+      `    shape?: ${m}${c}ShapeInput<TCtx>`,
+    ]
+    if (isRead && supportsProgressive) {
+      lines.push(`    progressive?: Record<string, ProgressiveVariantConfig>`)
+      lines.push(`    progressiveStages?: Record<string, ProgressiveStage<TCtx>>`)
+    }
+    return `  ${routerOp}?: {\n${lines.join('\n')}\n  }`
   }).join('\n')
 
   const omitKeys = ROUTER_OPERATIONS.map((k) => `'${k}'`).join('\n  | ')
 
+  const progressiveTypeImport = supportsProgressive
+    ? `import type { ProgressiveVariantConfig, ProgressiveStage } from '../routeConfig.target${ext}'\n\n`
+    : ''
+
   return (
+    progressiveTypeImport +
     `import type {\n  ${opShapeImports}\n} from '${guardShapesImport}'\n\n` +
     `export type ${m}RouteConfig<TCtx = unknown> = Omit<\n` +
-    `  RouteConfig,\n` +
+    `  RouteConfig<Record<string, any>, TCtx>,\n` +
     `  | ${omitKeys}\n` +
-    `> & {\n${overrides}\n}\n`
+    `  | 'resolveContext'\n` +
+    `> & {\n` +
+    `  resolveContext?: (request: import('express').Request) => TCtx | Promise<TCtx>\n` +
+    `${overrides}\n}\n`
   )
 }

package/src/generators/generateRouter.ts

@@ -1,18 +1,22 @@
 import { DMMF } from '@prisma/generator-helper'
 import { toCamelCase } from '../utils/strings'
 import { generateRouteConfigType } from './generateRouteConfigType'
+import { ImportStyle } from '../utils/resolveImportStyle'
+import { importExt } from '../utils/importExt'
 
 export function generateRouterFunction({
   model,
   enums,
-  relativeClientPath,
   guardShapesImport,
+  importStyle,
 }: {
   model: DMMF.Model
   enums: DMMF.DatamodelEnum[]
-  relativeClientPath: string
+  relativeClientPath?: string
   guardShapesImport: string | null
+  importStyle: ImportStyle
 }): string {
+  const ext = importExt(importStyle)
   const modelName = model.name
   const prefix = toCamelCase(modelName)
   const modelNameLower = modelName.toLowerCase()
@@ -41,8 +45,8 @@ export function generateRouterFunction({
       values: e.values.map((v) => ({ name: v.name })),
     }))
 
-  return `import express, { Request, Response, NextFunction, RequestHandler } from 'express'
-import type { PrismaClient } from '${relativeClientPath}'
+  return `import express from 'express'
+import type { Request, Response, NextFunction, RequestHandler } from 'express'
 import {
   ${prefix}FindUnique,
   ${prefix}FindUniqueOrThrow,
@@ -61,19 +65,25 @@ import {
   ${prefix}DeleteMany,
   ${prefix}Aggregate,
   ${prefix}Count,
-  ${prefix}GroupBy
-} from './${modelName}Handlers'
-import type { RouteConfig } from '../routeConfig.target'
-import { parseQueryParams } from '../parseQueryParams'
-import { sanitizeKeys } from '../misc'
-import { buildModelOpenApi } from '../buildModelOpenApi'
-import { transformResult } from '../operationRuntime'
-
-${generateRouteConfigType(modelName, 'RequestHandler', guardShapesImport)}
+  ${prefix}GroupBy,
+} from './${modelName}Handlers${ext}'
+import * as core from './${modelName}Core${ext}'
+import type { RouteConfig } from '../routeConfig.target${ext}'
+import { parseQueryParams } from '../parseQueryParams${ext}'
+import { sanitizeKeys } from '../misc${ext}'
+import { buildModelOpenApi } from '../buildModelOpenApi${ext}'
+import type { OperationContext } from '../operationRuntime${ext}'
+import {
+  transformResult,
+  acceptsEventStream,
+  runProgressiveEndpoint,
+  runSingleResultSSE,
+} from '../operationRuntime${ext}'
+
+${generateRouteConfigType(modelName, 'RequestHandler', guardShapesImport, importStyle, 'express')}
 const _env = typeof process !== 'undefined' && process.env ? process.env : {} as Record<string, string | undefined>
 
 const MODEL_FIELDS = ${JSON.stringify(fieldsMeta, null, 2)} as const
-
 const MODEL_ENUMS = ${JSON.stringify(enumsMeta, null, 2)} as const
 
 const defaultOpConfig = {
@@ -105,7 +115,6 @@ function getQueryBuilderConfig(config: RouteConfig) {
 
 export function ${routerFunctionName}<TCtx = unknown>(config: ${modelName}RouteConfig<TCtx> = {}) {
   const router = express.Router()
-
   router.use(express.json())
 
   const customPrefix = normalizePrefix(config.customUrlPrefix || '')
@@ -113,22 +122,29 @@ export function ${routerFunctionName}<TCtx = unknown>(config: ${modelName}RouteC
   const basePath = customPrefix + modelPrefix
 
   const openApiDisabled = config.disableOpenApi === true
-    || (config.disableOpenApi !== false && (
-      _env.DISABLE_OPENAPI === 'true'
-      || _env.NODE_ENV === 'production'
-    ))
+    || (config.disableOpenApi !== false && (_env.DISABLE_OPENAPI === 'true' || _env.NODE_ENV === 'production'))
 
   const postReadsEnabled = !config.disablePostReads
 
   const qbEnabled = isQueryBuilderEnabled(config)
-
   if (qbEnabled) {
     const qbConfig = getQueryBuilderConfig(config)
     if (qbConfig) {
-      try { require('../queryBuilder').startQueryBuilder(qbConfig) } catch (err) { if (_env.NODE_ENV !== 'production') console.warn('[query-builder]', err) }
+      try { require('../queryBuilder${ext}').startQueryBuilder(qbConfig) } catch (err) { if (_env.NODE_ENV !== 'production') console.warn('[query-builder]', err) }
     }
   }
 
+  const buildContext = (req: Request, res: Response): OperationContext => ({
+    prisma: (req as any).prisma,
+    postgres: (req as any).postgres,
+    sqlite: (req as any).sqlite,
+    parsedQuery: res.locals.parsedQuery,
+    body: req.body,
+    guardShape: res.locals.guardShape,
+    guardCaller: res.locals.guardCaller,
+    paginationConfig: res.locals.routeConfig?.pagination,
+  })
+
   const parseQuery: RequestHandler = (req, res, next) => {
     const rawQuery = req.query
     if (rawQuery && Object.keys(rawQuery).length > 0) {
@@ -148,147 +164,154 @@ export function ${routerFunctionName}<TCtx = unknown>(config: ${modelName}RouteC
   const setShape = (opConfig: any): RequestHandler => {
     return (req, res, next) => {
       res.locals.routeConfig = config
-      if (opConfig.shape) {
-        res.locals.guardShape = opConfig.shape
-        const caller = config.guard?.resolveVariant?.(req)
-          ?? req.get(config.guard?.variantHeader || 'x-api-variant')
-          ?? undefined
-        if (caller) {
-          res.locals.guardCaller = caller
+      const caller = config.guard?.resolveVariant?.(req)
+        ?? req.get(config.guard?.variantHeader || 'x-api-variant')
+        ?? undefined
+      if (caller) res.locals.guardCaller = caller
+      if (opConfig.shape) res.locals.guardShape = opConfig.shape
+      next()
+    }
+  }
+
+  const maybeProgressiveSSE = (opConfig: any, coreFn: (ctx: OperationContext) => Promise<unknown>): RequestHandler => {
+    return async (req, res, next) => {
+      if (res.headersSent || res.writableEnded) return next()
+      if (req.method !== 'GET') return next()
+      if (!acceptsEventStream(req.headers.accept)) return next()
+
+      const variant = res.locals.guardCaller as string | undefined
+      const progressiveConfig = variant ? opConfig.progressive?.[variant] : undefined
+
+      try {
+        if (!progressiveConfig || progressiveConfig.enabled === false) {
+          await runSingleResultSSE({
+            req,
+            res,
+            coreQueryFn: () => coreFn(buildContext(req, res)),
+          })
+          return
+        }
+
+        if (!Array.isArray(progressiveConfig.stages)) {
+          return next({ status: 500, message: 'Progressive endpoint requires stages array' })
+        }
+
+        const stageRegistry = opConfig.progressiveStages ?? {}
+        const missingStage = progressiveConfig.stages.find(
+          (name: string) => typeof stageRegistry[name] !== 'function',
+        )
+        if (missingStage) {
+          return next({ status: 500, message: 'Missing progressive stage: ' + missingStage })
+        }
+
+        if (typeof config.resolveContext !== 'function') {
+          return next({ status: 500, message: 'Progressive endpoint requires config.resolveContext' })
+        }
+
+        const ctx = await config.resolveContext(req)
+        await runProgressiveEndpoint({
+          req,
+          res,
+          ctx,
+          prisma: (req as any).prisma,
+          variant: variant as string,
+          stages: progressiveConfig.stages,
+          stageRegistry,
+        })
+      } catch (err) {
+        if (!res.headersSent) {
+          return next({ status: 500, message: 'Internal server error' })
         }
       }
-      next()
     }
   }
 
   const respond: RequestHandler = (_req, res) => {
     const data = res.locals.data
-    if (data === undefined) {
-      return res.status(500).json({ message: 'No data set by handler' })
-    }
+    if (data === undefined) return res.status(500).json({ message: 'No data set by handler' })
     return res.json(transformResult(data))
   }
 
   const respondCreated: RequestHandler = (_req, res) => {
     const data = res.locals.data
-    if (data === undefined) {
-      return res.status(500).json({ message: 'No data set by handler' })
-    }
+    if (data === undefined) return res.status(500).json({ message: 'No data set by handler' })
     return res.status(201).json(transformResult(data))
   }
 
   if (!openApiDisabled) {
     const openapiJsonPath = basePath ? \`\${basePath}/openapi.json\` : '/openapi.json'
     const openapiYamlPath = basePath ? \`\${basePath}/openapi.yaml\` : '/openapi.yaml'
-
     router.get(openapiJsonPath, (_req, res) => {
-      const spec = buildModelOpenApi(
-        '${modelName}',
-        MODEL_FIELDS as any,
-        MODEL_ENUMS as any,
-        config,
-        { format: 'json' }
-      )
+      const spec = buildModelOpenApi('${modelName}', MODEL_FIELDS as any, MODEL_ENUMS as any, config, { format: 'json' })
       res.json(spec)
     })
-
     router.get(openapiYamlPath, (_req, res) => {
-      const spec = buildModelOpenApi(
-        '${modelName}',
-        MODEL_FIELDS as any,
-        MODEL_ENUMS as any,
-        config,
-        { format: 'yaml' }
-      )
+      const spec = buildModelOpenApi('${modelName}', MODEL_FIELDS as any, MODEL_ENUMS as any, config, { format: 'yaml' })
       res.type('application/yaml').send(spec as string)
     })
   }
 
   if (config.enableAll || config.findFirst) {
-    const opConfig = config.findFirst || defaultOpConfig
+    const opConfig: any = config.findFirst || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath ? \`\${basePath}/first\` : '/first'
-    router.get(path, parseQuery, setShape(opConfig), ...before, ${prefix}FindFirst as RequestHandler, ...after, respond)
-    if (postReadsEnabled) {
-      router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}FindFirst as RequestHandler, ...after, respond)
-    }
+    router.get(path, parseQuery, setShape(opConfig), ...before, maybeProgressiveSSE(opConfig, core.findFirst), ${prefix}FindFirst as RequestHandler, ...after, respond)
+    if (postReadsEnabled) router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}FindFirst as RequestHandler, ...after, respond)
   }
-
   if (config.enableAll || config.findFirstOrThrow) {
-    const opConfig = config.findFirstOrThrow || defaultOpConfig
+    const opConfig: any = config.findFirstOrThrow || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath ? \`\${basePath}/first/strict\` : '/first/strict'
-    router.get(path, parseQuery, setShape(opConfig), ...before, ${prefix}FindFirstOrThrow as RequestHandler, ...after, respond)
-    if (postReadsEnabled) {
-      router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}FindFirstOrThrow as RequestHandler, ...after, respond)
-    }
+    router.get(path, parseQuery, setShape(opConfig), ...before, maybeProgressiveSSE(opConfig, core.findFirstOrThrow), ${prefix}FindFirstOrThrow as RequestHandler, ...after, respond)
+    if (postReadsEnabled) router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}FindFirstOrThrow as RequestHandler, ...after, respond)
   }
-
   if (config.enableAll || config.findManyPaginated) {
-    const opConfig = config.findManyPaginated || defaultOpConfig
+    const opConfig: any = config.findManyPaginated || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath ? \`\${basePath}/paginated\` : '/paginated'
-    router.get(path, parseQuery, setShape(opConfig), ...before, ${prefix}FindManyPaginated as RequestHandler, ...after, respond)
-    if (postReadsEnabled) {
-      router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}FindManyPaginated as RequestHandler, ...after, respond)
-    }
+    router.get(path, parseQuery, setShape(opConfig), ...before, maybeProgressiveSSE(opConfig, core.findManyPaginated), ${prefix}FindManyPaginated as RequestHandler, ...after, respond)
+    if (postReadsEnabled) router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}FindManyPaginated as RequestHandler, ...after, respond)
   }
-
   if (config.enableAll || config.aggregate) {
-    const opConfig = config.aggregate || defaultOpConfig
+    const opConfig: any = config.aggregate || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath ? \`\${basePath}/aggregate\` : '/aggregate'
-    router.get(path, parseQuery, setShape(opConfig), ...before, ${prefix}Aggregate as RequestHandler, ...after, respond)
-    if (postReadsEnabled) {
-      router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}Aggregate as RequestHandler, ...after, respond)
-    }
+    router.get(path, parseQuery, setShape(opConfig), ...before, maybeProgressiveSSE(opConfig, core.aggregate), ${prefix}Aggregate as RequestHandler, ...after, respond)
+    if (postReadsEnabled) router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}Aggregate as RequestHandler, ...after, respond)
   }
-
   if (config.enableAll || config.count) {
-    const opConfig = config.count || defaultOpConfig
+    const opConfig: any = config.count || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath ? \`\${basePath}/count\` : '/count'
-    router.get(path, parseQuery, setShape(opConfig), ...before, ${prefix}Count as RequestHandler, ...after, respond)
-    if (postReadsEnabled) {
-      router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}Count as RequestHandler, ...after, respond)
-    }
+    router.get(path, parseQuery, setShape(opConfig), ...before, maybeProgressiveSSE(opConfig, core.count), ${prefix}Count as RequestHandler, ...after, respond)
+    if (postReadsEnabled) router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}Count as RequestHandler, ...after, respond)
   }
-
   if (config.enableAll || config.groupBy) {
-    const opConfig = config.groupBy || defaultOpConfig
+    const opConfig: any = config.groupBy || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath ? \`\${basePath}/groupby\` : '/groupby'
-    router.get(path, parseQuery, setShape(opConfig), ...before, ${prefix}GroupBy as RequestHandler, ...after, respond)
-    if (postReadsEnabled) {
-      router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}GroupBy as RequestHandler, ...after, respond)
-    }
+    router.get(path, parseQuery, setShape(opConfig), ...before, maybeProgressiveSSE(opConfig, core.groupBy), ${prefix}GroupBy as RequestHandler, ...after, respond)
+    if (postReadsEnabled) router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}GroupBy as RequestHandler, ...after, respond)
   }
-
   if (config.enableAll || config.findUniqueOrThrow) {
-    const opConfig = config.findUniqueOrThrow || defaultOpConfig
+    const opConfig: any = config.findUniqueOrThrow || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath ? \`\${basePath}/unique/strict\` : '/unique/strict'
-    router.get(path, parseQuery, setShape(opConfig), ...before, ${prefix}FindUniqueOrThrow as RequestHandler, ...after, respond)
-    if (postReadsEnabled) {
-      router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}FindUniqueOrThrow as RequestHandler, ...after, respond)
-    }
+    router.get(path, parseQuery, setShape(opConfig), ...before, maybeProgressiveSSE(opConfig, core.findUniqueOrThrow), ${prefix}FindUniqueOrThrow as RequestHandler, ...after, respond)
+    if (postReadsEnabled) router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}FindUniqueOrThrow as RequestHandler, ...after, respond)
   }
-
   if (config.enableAll || config.findUnique) {
-    const opConfig = config.findUnique || defaultOpConfig
+    const opConfig: any = config.findUnique || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath ? \`\${basePath}/unique\` : '/unique'
-    router.get(path, parseQuery, setShape(opConfig), ...before, ${prefix}FindUnique as RequestHandler, ...after, respond)
-    if (postReadsEnabled) {
-      router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}FindUnique as RequestHandler, ...after, respond)
-    }
+    router.get(path, parseQuery, setShape(opConfig), ...before, maybeProgressiveSSE(opConfig, core.findUnique), ${prefix}FindUnique as RequestHandler, ...after, respond)
+    if (postReadsEnabled) router.post(path, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}FindUnique as RequestHandler, ...after, respond)
   }
-
   if (config.enableAll || config.findMany) {
-    const opConfig = config.findMany || defaultOpConfig
+    const opConfig: any = config.findMany || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath || '/'
-    router.get(path, parseQuery, setShape(opConfig), ...before, ${prefix}FindMany as RequestHandler, ...after, respond)
+    router.get(path, parseQuery, setShape(opConfig), ...before, maybeProgressiveSSE(opConfig, core.findMany), ${prefix}FindMany as RequestHandler, ...after, respond)
     if (postReadsEnabled) {
       const postPath = basePath ? \`\${basePath}/read\` : '/read'
       router.post(postPath, parseBodyAsQuery, setShape(opConfig), ...before, ${prefix}FindMany as RequestHandler, ...after, respond)
@@ -296,63 +319,55 @@ export function ${routerFunctionName}<TCtx = unknown>(config: ${modelName}RouteC
   }
 
   if (config.enableAll || config.createManyAndReturn) {
-    const opConfig = config.createManyAndReturn || defaultOpConfig
+    const opConfig: any = config.createManyAndReturn || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath ? \`\${basePath}/many/return\` : '/many/return'
     router.post(path, setShape(opConfig), ...before, ${prefix}CreateManyAndReturn as RequestHandler, ...after, respondCreated)
   }
-
   if (config.enableAll || config.createMany) {
-    const opConfig = config.createMany || defaultOpConfig
+    const opConfig: any = config.createMany || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath ? \`\${basePath}/many\` : '/many'
     router.post(path, setShape(opConfig), ...before, ${prefix}CreateMany as RequestHandler, ...after, respondCreated)
   }
-
   if (config.enableAll || config.create) {
-    const opConfig = config.create || defaultOpConfig
+    const opConfig: any = config.create || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath || '/'
     router.post(path, setShape(opConfig), ...before, ${prefix}Create as RequestHandler, ...after, respondCreated)
   }
-
   if (config.enableAll || config.updateManyAndReturn) {
-    const opConfig = config.updateManyAndReturn || defaultOpConfig
+    const opConfig: any = config.updateManyAndReturn || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath ? \`\${basePath}/many/return\` : '/many/return'
     router.put(path, setShape(opConfig), ...before, ${prefix}UpdateManyAndReturn as RequestHandler, ...after, respond)
   }
-
   if (config.enableAll || config.updateMany) {
-    const opConfig = config.updateMany || defaultOpConfig
+    const opConfig: any = config.updateMany || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath ? \`\${basePath}/many\` : '/many'
     router.put(path, setShape(opConfig), ...before, ${prefix}UpdateMany as RequestHandler, ...after, respond)
   }
-
   if (config.enableAll || config.update) {
-    const opConfig = config.update || defaultOpConfig
+    const opConfig: any = config.update || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath || '/'
     router.put(path, setShape(opConfig), ...before, ${prefix}Update as RequestHandler, ...after, respond)
   }
-
   if (config.enableAll || config.upsert) {
-    const opConfig = config.upsert || defaultOpConfig
+    const opConfig: any = config.upsert || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath || '/'
     router.patch(path, setShape(opConfig), ...before, ${prefix}Upsert as RequestHandler, ...after, respond)
   }
-
   if (config.enableAll || config.deleteMany) {
-    const opConfig = config.deleteMany || defaultOpConfig
+    const opConfig: any = config.deleteMany || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath ? \`\${basePath}/many\` : '/many'
     router.delete(path, setShape(opConfig), ...before, ${prefix}DeleteMany as RequestHandler, ...after, respond)
   }
-
   if (config.enableAll || config.delete) {
-    const opConfig = config.delete || defaultOpConfig
+    const opConfig: any = config.delete || defaultOpConfig
     const { before = [], after = [] } = opConfig
     const path = basePath || '/'
     router.delete(path, setShape(opConfig), ...before, ${prefix}Delete as RequestHandler, ...after, respond)
@@ -361,9 +376,7 @@ export function ${routerFunctionName}<TCtx = unknown>(config: ${modelName}RouteC
   router.use((err: any, _req: Request, res: Response, next: NextFunction) => {
     const status = typeof err.status === 'number' ? err.status : 500
     const message = err.message || 'Internal server error'
-    if (!res.headersSent) {
-      return res.status(status).json({ message })
-    }
+    if (!res.headersSent) return res.status(status).json({ message })
     next(err)
   })
 

package/src/generators/generateRouterFastify.ts

@@ -1,16 +1,21 @@
 import { DMMF } from '@prisma/generator-helper'
 import { toCamelCase } from '../utils/strings'
 import { generateRouteConfigType } from './generateRouteConfigType'
+import { ImportStyle } from '../utils/resolveImportStyle'
+import { importExt } from '../utils/importExt'
 
 export function generateFastifyRouterFunction({
   model,
   enums,
   guardShapesImport,
+  importStyle,
 }: {
   model: DMMF.Model
   enums: DMMF.DatamodelEnum[]
   guardShapesImport: string | null
+  importStyle: ImportStyle
 }): string {
+  const ext = importExt(importStyle)
   const modelName = model.name
   const prefix = toCamelCase(modelName)
   const modelNameLower = modelName.toLowerCase()
@@ -59,14 +64,14 @@ import {
   ${prefix}Aggregate,
   ${prefix}Count,
   ${prefix}GroupBy,
-} from './${modelName}Handlers'
-import type { RouteConfig, FastifyHookHandler } from '../routeConfig.target'
-import { parseQueryParams } from '../parseQueryParams'
-import { sanitizeKeys } from '../misc'
-import { buildModelOpenApi } from '../buildModelOpenApi'
-import { mapError, transformResult, HttpError } from '../operationRuntime'
-
-${generateRouteConfigType(modelName, 'FastifyHookHandler', guardShapesImport)}
+} from './${modelName}Handlers${ext}'
+import type { RouteConfig, FastifyHookHandler } from '../routeConfig.target${ext}'
+import { parseQueryParams } from '../parseQueryParams${ext}'
+import { sanitizeKeys } from '../misc${ext}'
+import { buildModelOpenApi } from '../buildModelOpenApi${ext}'
+import { mapError, transformResult, HttpError } from '../operationRuntime${ext}'
+
+${generateRouteConfigType(modelName, 'FastifyHookHandler', guardShapesImport, importStyle, 'fastify')}
 const _env = typeof process !== 'undefined' && process.env ? process.env : {} as Record<string, string | undefined>
 
 const MODEL_FIELDS = ${JSON.stringify(fieldsMeta, null, 2)} as const
@@ -181,7 +186,7 @@ export async function ${routerFunctionName}<TCtx = unknown>(
   if (qbEnabled) {
     const qbConfig = getQueryBuilderConfig(config)
     if (qbConfig) {
-      try { require('../queryBuilder').startQueryBuilder(qbConfig) } catch (err) { if (_env.NODE_ENV !== 'production') console.warn('[query-builder]', err) }
+      try { require('../queryBuilder${ext}').startQueryBuilder(qbConfig) } catch (err) { if (_env.NODE_ENV !== 'production') console.warn('[query-builder]', err) }
     }
   }
 

package/src/generators/generateRouterHono.ts

@@ -1,16 +1,21 @@
 import { DMMF } from '@prisma/generator-helper'
 import { toCamelCase } from '../utils/strings'
 import { generateRouteConfigType } from './generateRouteConfigType'
+import { ImportStyle } from '../utils/resolveImportStyle'
+import { importExt } from '../utils/importExt'
 
 export function generateHonoRouterFunction({
   model,
   enums,
   guardShapesImport,
+  importStyle,
 }: {
   model: DMMF.Model
   enums: DMMF.DatamodelEnum[]
   guardShapesImport: string | null
+  importStyle: ImportStyle
 }): string {
+  const ext = importExt(importStyle)
   const modelName = model.name
   const prefix = toCamelCase(modelName)
   const modelNameLower = modelName.toLowerCase()
@@ -61,14 +66,15 @@ import {
   ${prefix}Aggregate,
   ${prefix}Count,
   ${prefix}GroupBy,
-} from './${modelName}Handlers'
-import type { RouteConfig, HonoHookHandler } from '../routeConfig.target'
-import { parseQueryParams } from '../parseQueryParams'
-import { sanitizeKeys } from '../misc'
-import { buildModelOpenApi } from '../buildModelOpenApi'
-import { mapError, transformResult, HttpError } from '../operationRuntime'
-
- ${generateRouteConfigType(modelName, 'HonoHookHandler', guardShapesImport)}
+} from './${modelName}Handlers${ext}'
+import type { RouteConfig, HonoHookHandler } from '../routeConfig.target${ext}'
+import { parseQueryParams } from '../parseQueryParams${ext}'
+import { sanitizeKeys } from '../misc${ext}'
+import { buildModelOpenApi } from '../buildModelOpenApi${ext}'
+import { mapError, transformResult, HttpError } from '../operationRuntime${ext}'
+
+${generateRouteConfigType(modelName, 'HonoHookHandler', guardShapesImport, importStyle, 'hono')}
+
 type HonoVariables = {
   prisma: any
   postgres?: any