fix(zod): DateTime, Bytes, and Decimal field inference for z.input (#2685)

Author: sanny-io

packages/zod/src/types.ts

@@ -106,10 +106,10 @@ type FieldTypeZodMap = {
     Int: z.ZodNumber;
     BigInt: z.ZodBigInt;
     Float: z.ZodNumber;
-    Decimal: z.ZodType<Decimal>;
+    Decimal: z.ZodType<Decimal, Decimal>;
     Boolean: z.ZodBoolean;
-    DateTime: z.ZodType<Date>;
-    Bytes: z.ZodType<Uint8Array>;
+    DateTime: z.ZodDate;
+    Bytes: z.ZodType<Uint8Array, Uint8Array>;
     Json: JsonZodType;
 };
 

packages/zod/test/factory.test.ts

@@ -210,6 +210,14 @@ describe('SchemaFactory - makeModelSchema', () => {
             expect(userSchema.safeParse({ ...validUser, metadata: { key: BigInt(1) } }).success).toBe(false);
             expect(userSchema.safeParse({ ...validUser, metadata: [BigInt(1)] }).success).toBe(false);
         });
+
+        it('infers correct input types for fields', () => {
+            const _userSchema = factory.makeModelSchema('User');
+            type UserInput = z.input<typeof _userSchema>;
+            expectTypeOf<UserInput['birthdate']>().toEqualTypeOf<Date | null | undefined>();
+            expectTypeOf<UserInput['balance']>().toEqualTypeOf<Decimal>();
+            expectTypeOf<UserInput['avatar']>().toEqualTypeOf<Uint8Array | null | undefined>();
+        });
     });
 
     describe('string validation attributes', () => {