fix: schemaOf handles Dates

jquense · Feb 12, 2021 · c1fc816 · c1fc816
1 parent 0f12920
commit c1fc816
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -40,6 +40,8 @@ type ObjectSchemaOf<T extends AnyObject> = ObjectSchema<
   {
     [k in keyof T]-?: T[k] extends Array<infer E>
       ? ArraySchema<SchemaOf<E> | Lazy<SchemaOf<E>>>
+      : T[k] extends Date
+      ? DateSchema<T[k]>
       : T[k] extends AnyObject
       ? // we can't use  ObjectSchema<{ []: SchemaOf<T[k]> }> b/c TS produces a union of two schema
         ObjectSchemaOf<T[k] | Lazy<T[k]>>
@@ -49,6 +51,8 @@ type ObjectSchemaOf<T extends AnyObject> = ObjectSchema<
 
 type SchemaOf<T> = T extends Array<infer E>
   ? ArraySchema<SchemaOf<E> | Lazy<SchemaOf<E>>>
+  : T extends Date
+  ? DateSchema<T>
   : T extends AnyObject
   ? ObjectSchemaOf<T>
   : BaseSchema<T, Config>;

diff --git a/test/types.ts b/test/types.ts
@@ -10,6 +10,7 @@ import {
   ref,
   lazy,
   SchemaOf,
+  date,
 } from '../src';
 import type {
   AssertsShape,
@@ -220,6 +221,7 @@ SchemaOf: {
     title: string | undefined;
     age?: number;
     colors: string[];
+    createdAt: Date;
   };
 
   type PersonSchema = SchemaOf<Person>;
@@ -229,6 +231,7 @@ SchemaOf: {
     title: string(),
     age: number(),
     colors: array(string().defined()),
+    createdAt: date().defined(),
   });
 }