Support custom model name normalization in handler

Added an optional normalize function to collectSchemas and updated the handler to use a modelName provider from context if available. Extended types to support passing context with casing options to the handler.

Author: JeffreyNijs

package.json

@@ -1,6 +1,6 @@
 {
   "name": "hey-api-builders",
-  "version": "0.8.1",
+  "version": "1.0.1",
   "description": "A custom plugin for @hey-api/openapi-ts that generates mock data builders with a lightweight custom runtime, Zod integration, or static mock generation.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -61,4 +61,4 @@
     "vitest": "^3.2.4",
     "zod": "^3.23.8"
   }
-}
+}

src/core/schema-transformer.ts

@@ -360,11 +360,14 @@ export function sanitizeSchema(node: NormalizedSchemaNode): Schema {
  * @param all - All IR schemas
  * @returns Array of schema metadata
  */
-export function collectSchemas(all: Record<string, IR.SchemaObject>): GeneratedSchemaMeta[] {
+export function collectSchemas(
+  all: Record<string, IR.SchemaObject>,
+  normalize?: (name: string) => string
+): GeneratedSchemaMeta[] {
   const metas: GeneratedSchemaMeta[] = [];
   for (const [name, irSchema] of Object.entries(all)) {
     let typeName = name.replace(/Schema$/, '');
-    typeName = normalizeTypeName(typeName);
+    typeName = normalize ? normalize(typeName) : normalizeTypeName(typeName);
 
     const jsf = sanitizeSchema(normalizeSchema(irToSchema(irSchema as IR.SchemaObject, all)));
     const schemaWithType = jsf as ExtendedSchema;

src/plugin/handler.ts

@@ -37,12 +37,16 @@ function resolveMockStrategy(config: {
 /**
  * Main plugin handler for generating builder classes
  */
-export const handler: BuildersHandler = ({ plugin }) => {
+export const handler: BuildersHandler = ({ plugin, context }) => {
   const rawSchemas: Record<string, IR.SchemaObject> = {};
   plugin.forEach('schema', (event) => {
     rawSchemas[event.name] = event.schema;
   });
-  const metas = collectSchemas(rawSchemas);
+
+  const modelNameProvider = context?.casing?.modelName;
+  const normalize = modelNameProvider ? (name: string) => modelNameProvider(name) : undefined;
+
+  const metas = collectSchemas(rawSchemas, normalize);
 
   const file = plugin.createFile({ id: plugin.name, path: plugin.output });
 

src/types/index.ts

@@ -111,10 +111,22 @@ export interface Config {
  */
 export type BuildersPlugin = DefinePlugin<Config>;
 
+/**
+ * Arguments for the plugin handler
+ */
+export interface HandlerArgs {
+  plugin: Parameters<BuildersPlugin['Handler']>[0]['plugin'];
+  context: IR.Context & {
+    casing?: {
+      modelName?: (name: string) => string;
+    };
+  };
+}
+
 /**
  * Builder handler type
  */
-export type BuildersHandler = BuildersPlugin['Handler'];
+export type BuildersHandler = (args: HandlerArgs) => void;
 
 /**
  * Options for builder instances