From c1fc816cdb03f7c9ff2e6745ff38a2b4f119d556 Mon Sep 17 00:00:00 2001
From: Jason Quense <monastic.panic@gmail.com>
Date: Fri, 12 Feb 2021 17:15:43 -0500
Subject: [PATCH] fix: schemaOf handles Dates

---
 src/index.ts  | 4 ++++
 test/types.ts | 3 +++
 2 files changed, 7 insertions(+)

diff --git a/src/index.ts b/src/index.ts
index 5c86cee62..c8b73406b 100644
--- 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
index dfe20c555..d2d297764 100644
--- 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(),
   });
 }