change type definitions to account for transform and preprocess

asteasolutions · Apr 3, 2024 · da04076 · da04076
1 parent 56995bf
commit da04076
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/src/openapi-generator.ts b/src/openapi-generator.ts
@@ -17,6 +17,7 @@ import {
   OpenAPIDefinitions,
   ResponseConfig,
   RouteConfig,
+  RouteParameter,
   ZodContentObject,
   ZodRequestBody,
 } from './openapi-registry';
@@ -543,9 +544,7 @@ export class OpenAPIGenerator {
     ];
   }
 
-  private cleanParameter(
-    schema: AnyZodObject | ZodEffects<AnyZodObject> | undefined
-  ) {
+  private cleanParameter(schema: RouteParameter) {
     if (!schema) {
       return undefined;
     }

diff --git a/src/openapi-registry.ts b/src/openapi-registry.ts
@@ -102,15 +102,20 @@ export interface ResponseConfig {
   content?: ZodContentObject;
 }
 
+export type RouteParameter =
+  | AnyZodObject
+  | ZodEffects<AnyZodObject, unknown, unknown>
+  | undefined;
+
 export type RouteConfig = Omit<OperationObject, 'responses'> & {
   method: Method;
   path: string;
   request?: {
     body?: ZodRequestBody;
-    params?: AnyZodObject | ZodEffects<AnyZodObject>;
-    query?: AnyZodObject | ZodEffects<AnyZodObject>;
-    cookies?: AnyZodObject | ZodEffects<AnyZodObject>;
-    headers?: AnyZodObject | ZodEffects<AnyZodObject> | ZodType<unknown>[];
+    params?: RouteParameter;
+    query?: RouteParameter;
+    cookies?: RouteParameter;
+    headers?: RouteParameter | ZodType<unknown>[];
   };
   responses: {
     [statusCode: string]: ResponseConfig;