InstantiableZodSchemaObject

2024-02-07 00:32:23 +01:00
parent 6ee9337426
commit f40bebbc9b
2 changed files with 99 additions and 50 deletions
--- a/src/traits/InstantiableZodSchemaObject.ts
+++ b/src/traits/InstantiableZodSchemaObject.ts
@@ -0,0 +1,95 @@
+import { abstract, trait } from "@thilawyn/traitify-ts"
+import { Effect, pipe } from "effect"
+import { Class } from "type-fest"
+import { z } from "zod"
+import { NewZodSchemaInstanceArgs, NewZodSchemaInstanceInput } from ".."
+import { parseZodTypeEffect } from "../util"
+
+
+type ZodSchemaClassStatic<
+    SchemaT           extends z.ZodRawShape,
+    SchemaUnknownKeys extends z.UnknownKeysParam,
+    SchemaCatchall    extends z.ZodTypeAny,
+
+    Values            extends {},
+    DefaultValues     extends Partial<Values>,
+> = {
+    readonly schema:         z.ZodObject<SchemaT, SchemaUnknownKeys, SchemaCatchall, Values, Values>
+    readonly defaultValues:  DefaultValues
+}
+
+
+export const InstantiableZodSchemaObject = trait(
+    abstract(),
+
+    Super => class InstantiableZodSchemaObject extends Super {
+        static create<
+            Self              extends Class<any, any[]> & ZodSchemaClassStatic<SchemaT, SchemaUnknownKeys, SchemaCatchall, Values, DefaultValues>,
+
+            SchemaT           extends z.ZodRawShape,
+            SchemaUnknownKeys extends z.UnknownKeysParam,
+            SchemaCatchall    extends z.ZodTypeAny,
+
+            Values            extends {},
+            DefaultValues     extends Partial<Values>,
+        >(
+            this: Self,
+
+            ...[values, params]: NewZodSchemaInstanceArgs<
+                NewZodSchemaInstanceInput<Values, DefaultValues>
+            >
+        ) {
+            return new this(
+                this.schema.parse({ ...this.defaultValues, ...values }, params)
+            ) as InstanceType<Self>
+        }
+
+        static async createPromise<
+            Self              extends Class<any, any[]> & ZodSchemaClassStatic<SchemaT, SchemaUnknownKeys, SchemaCatchall, Values, DefaultValues>,
+
+            SchemaT           extends z.ZodRawShape,
+            SchemaUnknownKeys extends z.UnknownKeysParam,
+            SchemaCatchall    extends z.ZodTypeAny,
+
+            Values            extends {},
+            DefaultValues     extends Partial<Values>,
+        >(
+            this: Self,
+
+            ...[values, params]: NewZodSchemaInstanceArgs<
+                NewZodSchemaInstanceInput<Values, DefaultValues>
+            >
+        ) {
+            return new this(
+                await this.schema.parseAsync({ ...this.defaultValues, ...values }, params)
+            ) as Promise<InstanceType<Self>>
+        }
+
+        static createEffect<
+            Self              extends Class<any, any[]> & ZodSchemaClassStatic<SchemaT, SchemaUnknownKeys, SchemaCatchall, Values, DefaultValues>,
+
+            SchemaT           extends z.ZodRawShape,
+            SchemaUnknownKeys extends z.UnknownKeysParam,
+            SchemaCatchall    extends z.ZodTypeAny,
+
+            Values            extends {},
+            DefaultValues     extends Partial<Values>,
+        >(
+            this: Self,
+
+            ...[values, params]: NewZodSchemaInstanceArgs<
+                NewZodSchemaInstanceInput<Values, DefaultValues>
+            >
+        ) {
+            return pipe(
+                parseZodTypeEffect(
+                    this.schema,
+                    { ...this.defaultValues, ...values },
+                    params,
+                ),
+
+                Effect.map(values => new this(values) as InstanceType<Self>),
+            )
+        }
+    },
+)