Merge pull request #3555 from hey-api/feat/zod-processor-export

feat: add export flag to zod processor

Author: mrlubos

packages/openapi-ts/src/plugins/zod/mini/processor.ts

@@ -33,13 +33,13 @@ export function createProcessor(plugin: ZodPlugin['Instance']): ProcessorResult
     return ctx.schema;
   }
 
-  function process(ctx: ProcessorContext): void {
+  function process(ctx: ProcessorContext): ZodFinal | void {
     if (!processor.markEmitted(ctx.path)) return;
 
+    const shouldExport = ctx.export !== false;
+
     return processor.withContext({ anchor: ctx.namingAnchor, tags: ctx.tags }, () => {
-      const visitor = createVisitor({
-        schemaExtractor: extractor,
-      });
+      const visitor = createVisitor({ schemaExtractor: extractor });
       const walk = createSchemaWalker(visitor);
 
       const result = walk(ctx.schema, {
@@ -52,7 +52,12 @@ export function createProcessor(plugin: ZodPlugin['Instance']): ProcessorResult
         plugin,
       }) as ZodFinal;
 
-      exportAst({ ...ctx, final, plugin });
+      if (shouldExport) {
+        exportAst({ ...ctx, final, plugin });
+        return;
+      }
+
+      return final;
     });
   }
 

packages/openapi-ts/src/plugins/zod/shared/processor.ts

@@ -1,18 +1,17 @@
-import type {
-  IR,
-  NamingConfig,
-  SchemaProcessorContext,
-  SchemaProcessorResult,
-} from '@hey-api/shared';
+import type { IR, NamingConfig, SchemaProcessorContext } from '@hey-api/shared';
 
 import type { ZodPlugin } from '../types';
-import type { TypeOptions } from './types';
+import type { TypeOptions, ZodFinal } from './types';
 
 export type ProcessorContext = SchemaProcessorContext & {
+  /** Whether to export the result (default: true) */
+  export?: boolean;
   naming: NamingConfig & TypeOptions;
   /** The plugin instance. */
   plugin: ZodPlugin['Instance'];
   schema: IR.SchemaObject;
 };
 
-export type ProcessorResult = SchemaProcessorResult<ProcessorContext>;
+export type ProcessorResult = {
+  process: (ctx: ProcessorContext) => ZodFinal | void;
+};

packages/openapi-ts/src/plugins/zod/v3/processor.ts

@@ -33,13 +33,13 @@ export function createProcessor(plugin: ZodPlugin['Instance']): ProcessorResult
     return ctx.schema;
   }
 
-  function process(ctx: ProcessorContext): void {
+  function process(ctx: ProcessorContext): ZodFinal | void {
     if (!processor.markEmitted(ctx.path)) return;
 
+    const shouldExport = ctx.export !== false;
+
     return processor.withContext({ anchor: ctx.namingAnchor, tags: ctx.tags }, () => {
-      const visitor = createVisitor({
-        schemaExtractor: extractor,
-      });
+      const visitor = createVisitor({ schemaExtractor: extractor });
       const walk = createSchemaWalker(visitor);
 
       const result = walk(ctx.schema, {
@@ -52,7 +52,12 @@ export function createProcessor(plugin: ZodPlugin['Instance']): ProcessorResult
         plugin,
       }) as ZodFinal;
 
-      exportAst({ ...ctx, final, plugin });
+      if (shouldExport) {
+        exportAst({ ...ctx, final, plugin });
+        return;
+      }
+
+      return final;
     });
   }
 

packages/openapi-ts/src/plugins/zod/v4/processor.ts

@@ -33,13 +33,13 @@ export function createProcessor(plugin: ZodPlugin['Instance']): ProcessorResult
     return ctx.schema;
   }
 
-  function process(ctx: ProcessorContext): void {
+  function process(ctx: ProcessorContext): ZodFinal | void {
     if (!processor.markEmitted(ctx.path)) return;
 
+    const shouldExport = ctx.export !== false;
+
     return processor.withContext({ anchor: ctx.namingAnchor, tags: ctx.tags }, () => {
-      const visitor = createVisitor({
-        schemaExtractor: extractor,
-      });
+      const visitor = createVisitor({ schemaExtractor: extractor });
       const walk = createSchemaWalker(visitor);
 
       const result = walk(ctx.schema, {
@@ -52,7 +52,12 @@ export function createProcessor(plugin: ZodPlugin['Instance']): ProcessorResult
         plugin,
       }) as ZodFinal;
 
-      exportAst({ ...ctx, final, plugin });
+      if (shouldExport) {
+        exportAst({ ...ctx, final, plugin });
+        return;
+      }
+
+      return final;
     });
   }