@kubb/core 5.0.0-beta.61 → 5.0.0-beta.62

package/dist/mocks.d.ts

@@ -1,5 +1,5 @@
 import { t as __name } from "./chunk-C0LytTxp.js";
-import { At as Adapter, G as Generator, J as KubbDriver, U as Parser, h as Config, jt as AdapterFactoryOptions, st as PluginFactoryOptions, tt as NormalizedPlugin } from "./diagnostics-DiaUv_iK.js";
+import { At as Adapter, G as Generator, J as KubbDriver, U as Parser, h as Config, jt as AdapterFactoryOptions, st as PluginFactoryOptions, tt as NormalizedPlugin } from "./diagnostics-D0G07LHG.js";
 import { FileNode, InputMeta, Macro, OperationNode, SchemaNode } from "@kubb/ast";
 
 //#region src/mocks.d.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kubb/core",
-  "version": "5.0.0-beta.61",
+  "version": "5.0.0-beta.62",
   "description": "Core engine for Kubb's plugin-based code generation system. Provides the plugin driver, file manager, defineConfig, and build orchestration used by every Kubb plugin.",
   "keywords": [
     "code-generator",
@@ -56,14 +56,14 @@
     "registry": "https://registry.npmjs.org/"
   },
   "dependencies": {
-    "@kubb/ast": "5.0.0-beta.61"
+    "@kubb/ast": "5.0.0-beta.62"
   },
   "devDependencies": {
     "@internals/utils": "0.0.0",
-    "@kubb/renderer-jsx": "5.0.0-beta.61"
+    "@kubb/renderer-jsx": "5.0.0-beta.62"
   },
   "peerDependencies": {
-    "@kubb/renderer-jsx": "5.0.0-beta.61"
+    "@kubb/renderer-jsx": "5.0.0-beta.62"
   },
   "engines": {
     "node": ">=22"

package/src/KubbDriver.ts

@@ -2,7 +2,7 @@ import { resolve } from 'node:path'
 import { arrayToAsyncIterable, type AsyncEventEmitter, forBatches, getElapsedMs, isPromise, memoize, Url } from '@internals/utils'
 import * as factory from '@kubb/ast/factory'
 import { collectUsedSchemaNames } from '@kubb/ast/utils'
-import type { FileNode, InputMeta, InputNode, OperationNode, SchemaNode } from '@kubb/ast'
+import type { Enforce, FileNode, InputMeta, InputNode, OperationNode, SchemaNode } from '@kubb/ast'
 import { OPERATION_FILTER_TYPES, SCHEMA_PARALLEL } from './constants.ts'
 import { type Diagnostic, Diagnostics, type ProblemDiagnostic } from './diagnostics.ts'
 import type { RendererFactory } from './createRenderer.ts'
@@ -45,7 +45,7 @@ type RequirePluginContext = {
   requiredBy?: string
 }
 
-function enforceOrder(enforce: 'pre' | 'post' | undefined): number {
+function enforceOrder(enforce: Enforce | undefined): number {
   return enforce === 'pre' ? -1 : enforce === 'post' ? 1 : 0
 }
 
@@ -282,37 +282,21 @@ export class KubbDriver {
    * Call this method inside `addGenerator()` (in `kubb:plugin:setup`) to wire up a generator.
    */
   registerGenerator(pluginName: string, generator: Generator): void {
-    if (generator.schema) {
-      const schemaHandler = async (node: SchemaNode, ctx: GeneratorContext) => {
-        if (ctx.plugin.name !== pluginName) return
-        const result = await generator.schema!(node, ctx)
-
-        await this.dispatch({ result, renderer: generator.renderer })
-      }
+    const register = <TNode>(event: keyof KubbHooks & string, method: ((node: TNode, ctx: GeneratorContext) => unknown) | undefined): void => {
+      if (!method) return
 
-      this.#trackListener('kubb:generate:schema', schemaHandler)
-    }
-
-    if (generator.operation) {
-      const operationHandler = async (node: OperationNode, ctx: GeneratorContext) => {
+      const handler = async (node: TNode, ctx: GeneratorContext) => {
         if (ctx.plugin.name !== pluginName) return
-
-        const result = await generator.operation!(node, ctx)
+        const result = await method(node, ctx)
         await this.dispatch({ result, renderer: generator.renderer })
       }
 
-      this.#trackListener('kubb:generate:operation', operationHandler)
+      this.#trackListener(event, handler as HookListener<KubbHooks[typeof event], unknown>)
     }
 
-    if (generator.operations) {
-      const operationsHandler = async (nodes: Array<OperationNode>, ctx: GeneratorContext) => {
-        if (ctx.plugin.name !== pluginName) return
-        const result = await generator.operations!(nodes, ctx)
-        await this.dispatch({ result, renderer: generator.renderer })
-      }
-
-      this.#trackListener('kubb:generate:operations', operationsHandler)
-    }
+    register('kubb:generate:schema', generator.schema)
+    register('kubb:generate:operation', generator.operation)
+    register('kubb:generate:operations', generator.operations)
 
     this.#eventGeneratorPlugins.add(pluginName)
   }

package/src/definePlugin.ts

@@ -1,4 +1,4 @@
-import type { FileNode, HttpMethod, Macro, UserFileNode } from '@kubb/ast'
+import type { Enforce, FileNode, HttpMethod, Macro, UserFileNode } from '@kubb/ast'
 import { diagnosticCode } from './constants.ts'
 import type { Generator } from './defineGenerator.ts'
 import type { BannerMeta, Resolver } from './defineResolver.ts'
@@ -351,7 +351,7 @@ export type Plugin<TFactory extends PluginFactoryOptions = PluginFactoryOptions>
    *
    * Dependency constraints always take precedence over `enforce`.
    */
-  enforce?: 'pre' | 'post'
+  enforce?: Enforce
   /**
    * The options passed by the user when calling the plugin factory.
    */