fix: type (#577)

* docs: update

* fix: type

* test: transform

Author: soc221b

README.md

@@ -25,17 +25,18 @@ Built-in zod types:
 ```ts
 import * as z from 'zod' // or import * as z from 'zod/mini'
 import { faker } from '@faker-js/faker'
-import { fake, fakeSchema, setFaker } from 'zod-schema-faker'
+import { fake, setFaker } from 'zod-schema-faker'
 
-const User = z.interface({
-  name: z.string(),
-  age: z.uint32(),
+const Player = z.object({
+  username: z.string(),
+  xp: z.number(),
 })
 
 // enable tree shaking
 if (process.env.NODE_ENV === 'development') {
-  setFaker(faker) // or setFaker(your faker instance)
-  fake(z.string()) // { name: 'lorem', age: 42 }
+  setFaker(faker)
+  const data = fake(Player)
+  console.log(data) // { username: "billie", xp: 100 }
 }
 ```
 
@@ -88,11 +89,9 @@ fake(px) // '100px'
 ## Unsupported
 
 - .file 🚧
-- .function 🚧
 - .intersection 🚧
 - .refine ❌
 - .superRefine ❌
-- .transform 🚧
 
 ## About
 

e2e/issue-189/src/index.ts

@@ -3,15 +3,17 @@ import { expectType, TypeEqual } from 'ts-expect'
 import { z } from 'zod'
 import { fake, setFaker } from 'zod-schema-faker'
 
-const schema = z.number()
+const Player = z.object({
+  username: z.string(),
+  xp: z.number(),
+})
 
 // enable tree shaking
 if (process.env.NODE_ENV === 'development') {
   setFaker(faker)
-  const data = fake(schema)
-
-  console.log(data) // => -2556.9
+  const data = fake(Player)
+  console.log(data) // { username: "billie", xp: 100 }
 
   expectType<TypeEqual<typeof data, any>>(false)
-  expectType<TypeEqual<typeof data, number>>(true)
+  expectType<TypeEqual<typeof data, { username: string; xp: number }>>(true)
 }

src/fake.ts

@@ -1,6 +1,6 @@
+import * as core from 'zod/v4/core'
 import { rootFake } from './internals/fake'
-import { ZodType } from './internals/type'
 
-export function fake<T extends ZodType>(schema: T): T['_zod']['output'] {
+export function fake<T extends core.$ZodType>(schema: T): core.infer<T> {
   return rootFake(schema, { depth: 0 })
 }

src/internals/schemas/custom.ts

@@ -1,9 +1,9 @@
 import * as core from 'zod/v4/core'
 import { Context } from '../context'
-import { Fake, Infer, RootFake, ZodType } from '../type'
+import { Fake, Infer, RootFake } from '../type'
 
-const customs = new Map<ZodType, Fake<ZodType>>()
-export function custom<T extends ZodType>(schema: T, fake: Fake<T>): void {
+const customs = new Map<core.$ZodType, Fake<core.$ZodType>>()
+export function custom<T extends core.$ZodType>(schema: T, fake: Fake<T>): void {
   customs.set(schema, fake as any)
 }
 

src/internals/type.ts

@@ -1,9 +1,8 @@
+import * as core from 'zod/v4/core'
 import { Context } from './context'
 
-export type ZodType<O = unknown> = { _zod: { output: O } } // TODO: use core.$ZodType
+export type Infer<T extends core.$ZodType> = T['_zod']['output']
 
-export type Infer<T extends ZodType> = T['_zod']['output']
+export type Fake<T extends core.$ZodType> = (schema: T, context: Context, fake: RootFake) => core.infer<T>
 
-export type Fake<T extends ZodType> = (schema: T, context: Context, fake: RootFake) => Infer<T>
-
-export type RootFake = <T extends ZodType>(schema: T, context: Context) => Infer<T>
+export type RootFake = <T extends core.$ZodType>(schema: T, context: Context) => core.infer<T>

tests/mini.test.ts

@@ -596,8 +596,14 @@ const validSuits: { schema: z.ZodMiniType; description?: string; only?: boolean;
     description: 'refinement',
   },
 
-  // TODO: transform
-  // { schema: z.transform(val => String(val)) },
+  // transform
+  {
+    schema: z.pipe(
+      z.string(),
+      z.transform(val => val.length),
+    ),
+    description: 'transform',
+  },
 
   // tuple
   {

tests/zod.test.ts

@@ -598,7 +598,10 @@ const validSuits: { schema: z.ZodType; description?: string; only?: boolean; asy
   },
 
   // TODO: transform
-  // { schema: z.transform(val => String(val)) },
+  {
+    schema: z.string().pipe(z.transform(val => val.length)),
+    description: 'transform',
+  },
 
   // tuple
   {