feat: pipeline support

lib/MockGenerator.ts

@@ -28,6 +28,7 @@ import PromiseGenerator from './generators/PromiseGenerator';
 import MapGenerator from './generators/MapGenerator';
 import SetGenerator from './generators/SetGenerator';
 import RecordGenerator from './generators/RecordGenerator';
+import PipelineGenerator from './generators/PipelineGenerator';
 
 export default class MockGenerator<T extends z.ZodTypeAny> {
   private generator: BaseGenerator<TypeOf<T>>;
@@ -67,6 +68,7 @@ export default class MockGenerator<T extends z.ZodTypeAny> {
       ZodMap: MapGenerator,
       ZodSet: SetGenerator,
       ZodRecord: RecordGenerator,
+      ZodPipeline: PipelineGenerator,
     };
     if (this.schema._def.typeName in generatorMap) {
       this.generator = new generatorMap[this.schema._def.typeName as ZodFirstPartyTypeKind]();

lib/generators/PipelineGenerator.ts

@@ -0,0 +1,11 @@
+import type { z } from 'zod';
+import MockGenerator from '../MockGenerator';
+import type BaseGenerator from './BaseGenerator';
+
+export default class PipelineGenerator<T extends z.ZodPipeline<z.ZodTypeAny, z.ZodTypeAny>> implements BaseGenerator<T> {
+  public generate(schema: T): z.TypeOf<T> {
+    const { out: outSchema } = schema._def;
+    const outGenerated = new MockGenerator(outSchema).generate();
+    return outGenerated;
+  }
+}

tests/pipeline.test.ts

@@ -0,0 +1,25 @@
+import { describe, test } from 'vitest';
+import { z } from 'zod';
+import { expect } from './utils/expect';
+
+describe('Pipeline', () => {
+  test('with transform', () => {
+    const schema = z.string()
+      .transform(val => val.length)
+      .pipe(z.number().min(10));
+
+    expect(schema._def.out);
+  });
+  test('with coerse', () => {
+    const dateLike = z.union([z.number(), z.string(), z.date()]);
+    const schema = dateLike.pipe(z.coerce.date());
+
+    expect(schema._def.out);
+  });
+  test('with chain coerse', () => {
+    const toNumber = z.number().or(z.string()).pipe(z.coerce.number());
+    const toBigInt = z.bigint().or(toNumber).pipe(z.coerce.bigint());
+
+    expect(toBigInt._def.out);
+  });
+});