feat: adding support to dates

yUnreal · May 25, 2024 · 840dacd · 840dacd
1 parent 5eb7923
commit 840dacd
Show file tree

Hide file tree

Showing 6 changed files with 57 additions and 2 deletions.
diff --git a/lib/structs/Document.ts b/lib/structs/Document.ts
@@ -1,3 +1,4 @@
+import isPlainObject from 'is-plain-obj';
 import { AnyObject } from '../types/utils';
 import { Model } from './Model';
 import { Schema } from './schema/Schema';
@@ -6,7 +7,16 @@ export class Document<T extends AnyObject & { id?: string }> {
     public constructor(
         public data: T,
         public model: Model<Schema<T>>
-    ) {}
+    ) {
+        for (const [key, value] of Object.entries(data)) {
+            const DATE_KEY = '$date';
+
+            if (isPlainObject(value) && value[DATE_KEY])
+                Object.defineProperty(data, key, {
+                    value: new Date(<number>value[DATE_KEY]),
+                });
+        }
+    }
 
     public fetch() {
         return this.model.findById(this.id);

diff --git a/lib/structs/Model.ts b/lib/structs/Model.ts
@@ -24,6 +24,14 @@ export class Model<S extends Schema<AnyObject>> {
 
         this.schema.parse(data);
 
+        for (const [key, value] of Object.entries(data)) {
+            if (value instanceof Date)
+                Object.defineProperty(data, key, {
+                    value: { $date: value.getTime() },
+                    enumerable: true,
+                });
+        }
+
         data.id ??= uuid();
 
         this.driver.update((crrData) =>

diff --git a/lib/structs/schema/DateSchemaKey.ts b/lib/structs/schema/DateSchemaKey.ts
@@ -0,0 +1,24 @@
+import { Expression, Types } from '../../types/schema';
+import { SchemaKey } from './SchemaKey';
+
+export class DateSchemaKey extends SchemaKey<Types.Date> {
+    public min(
+        date: Date,
+        message = `${Expression.Key} must be older than ${date.toISOString()}`
+    ) {
+        return this.effect(
+            (crrData) => crrData.getTime() > date.getTime(),
+            message
+        );
+    }
+
+    public max(
+        date: Date,
+        message = `${Expression.Key} must be newer than ${date.toISOString()}`
+    ) {
+        return this.effect(
+            (crrData) => crrData.getTime() < date.getTime(),
+            message
+        );
+    }
+}
diff --git a/lib/structs/schema/Schema.ts b/lib/structs/schema/Schema.ts
@@ -5,6 +5,7 @@ import { StringSchemaKey } from './StringSchemaKey';
 import { NumberSchemaKey } from './NumberSchemaKey';
 import { BooleanSchemaKey } from './BooleanSchemaKey';
 import { BigIntSchemaKey } from './BigIntSchemaKey';
+import { DateSchemaKey } from './DateSchemaKey';
 
 export class Schema<S extends AnyObject> {
     public constructor(public shape: Required<InferShape<S>>) {}
@@ -25,6 +26,10 @@ export class Schema<S extends AnyObject> {
         return new BigIntSchemaKey({ type: Types.BigInt, flags: [] });
     }
 
+    public static date() {
+        return new DateSchemaKey({ type: Types.Date, flags: [] });
+    }
+
     public parse<O>(object: O) {
         if (!isPlainObject(object))
             throw new Error('Invalid value when parsing schema');

diff --git a/lib/types/schema.ts b/lib/types/schema.ts
@@ -1,5 +1,6 @@
 import { BigIntSchemaKey } from '../structs/schema/BigIntSchemaKey';
 import { BooleanSchemaKey } from '../structs/schema/BooleanSchemaKey';
+import { DateSchemaKey } from '../structs/schema/DateSchemaKey';
 import { NumberSchemaKey } from '../structs/schema/NumberSchemaKey';
 import { Schema } from '../structs/schema/Schema';
 import { StringSchemaKey } from '../structs/schema/StringSchemaKey';
@@ -10,6 +11,7 @@ export enum Types {
     Number,
     Boolean,
     BigInt,
+    Date,
 }
 
 export type SchemaDefinition = {
@@ -31,13 +33,15 @@ export interface MappedSchemaTypes {
     [Types.Number]: number;
     [Types.Boolean]: boolean;
     [Types.BigInt]: bigint;
+    [Types.Date]: Date;
 }
 
 export interface MappedSchemaKeys {
     [Types.String]: StringSchemaKey;
     [Types.Number]: NumberSchemaKey;
     [Types.Boolean]: BooleanSchemaKey;
     [Types.BigInt]: BigIntSchemaKey;
+    [Types.Date]: DateSchemaKey;
 }
 
 export type InferShape<S extends AnyObject> = {
@@ -52,7 +56,9 @@ export type ExtractType<S> = S extends string
         ? Types.Boolean
         : S extends bigint
           ? Types.BigInt
-          : never;
+          : S extends Date
+            ? Types.Date
+            : never;
 
 export interface Constraint<Type extends Types> {
     effect(value: MappedSchemaTypes[Type]): unknown;

diff --git a/lib/utils/getType.ts b/lib/utils/getType.ts
@@ -1,6 +1,8 @@
 import { Types } from '../types/schema';
 
 export const getType = (value: unknown) => {
+    if (value instanceof Date) return Types.Date;
+
     switch (typeof value) {
     case 'number':
         return Types.Number;