feat: parse ZodPipeline

anatine · Aug 8, 2023 · b3e6cb7 · b3e6cb7
1 parent 547e9a8
commit b3e6cb7
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 4 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/packages/zod-openapi/src/lib/zod-openapi.spec.ts b/packages/zod-openapi/src/lib/zod-openapi.spec.ts
@@ -931,4 +931,34 @@ describe('zodOpenapi', () => {
       ]
     });
   })
+
+  it('should work with ZodPipeline', () => {
+    expect(
+      generateSchema(
+        z
+          .string()
+          .regex(/^\d+$/)
+          .transform(Number)
+          .pipe(z.number().min(0).max(10))
+      )
+    ).toEqual({
+      type: 'string',
+      pattern: '^\\d+$',
+    } satisfies SchemaObject);
+
+    expect(
+      generateSchema(
+        z
+          .string()
+          .regex(/^\d+$/)
+          .transform(Number)
+          .pipe(z.number().min(0).max(10)),
+        true
+      )
+    ).toEqual({
+      type: 'number',
+      minimum: 0,
+      maximum: 10,
+    } satisfies SchemaObject);
+  });
 });
diff --git a/packages/zod-openapi/src/lib/zod-openapi.ts b/packages/zod-openapi/src/lib/zod-openapi.ts
@@ -511,6 +511,16 @@ function catchAllParser({
   );
 }
 
+function parsePipeline({
+  zodRef,
+  useOutput,
+}: ParsingArgs<z.ZodPipeline<never, never>>): SchemaObject {
+  if (useOutput) {
+    return generateSchema(zodRef._def.out, useOutput);
+  }
+  return generateSchema(zodRef._def.in, useOutput);
+}
+
 const workerMap = {
   ZodObject: parseObject,
   ZodRecord: parseRecord,
@@ -545,6 +555,7 @@ const workerMap = {
   ZodAny: catchAllParser,
   ZodUnknown: catchAllParser,
   ZodVoid: catchAllParser,
+  ZodPipeline: parsePipeline,
 };
 type WorkerKeys = keyof typeof workerMap;