TZodSchemaClass work

src/ZodSchemaClass.ts

@@ -1,7 +1,8 @@
+import { Effect, pipe } from "effect"
 import { AbstractClass, Class as ConcreteClass, Opaque } from "type-fest"
 import { z } from "zod"
 import { DefinedDefaultValuesTag, NewZodSchemaInstanceArgs, NewZodSchemaInstanceInput, TZodSchemaClass } from "."
-import { StaticMembers } from "./util"
+import { StaticMembers, parseZodTypeEffect } from "./util"
 
 
 type ExtendableClass = AbstractClass<any, []> & {
@@ -12,7 +13,6 @@ type ExtendableClass = AbstractClass<any, []> & {
 export function ZodSchemaClassOf<
     Parent            extends ExtendableClass,
 
-    Schema            extends z.ZodObject<SchemaT, SchemaUnknownKeys, SchemaCatchall, Values, Values>,
     SchemaT           extends z.ZodRawShape,
     SchemaUnknownKeys extends z.UnknownKeysParam,
     SchemaCatchall    extends z.ZodTypeAny,
@@ -23,7 +23,7 @@ export function ZodSchemaClassOf<
     of: Parent,
 
     { schema, defaultValues }: {
-        schema:         Schema | z.ZodObject<SchemaT, SchemaUnknownKeys, SchemaCatchall, Values, Values>
+        schema:         z.ZodObject<SchemaT, SchemaUnknownKeys, SchemaCatchall, Values, Values>
         defaultValues:  Opaque<DefaultValues, DefinedDefaultValuesTag>
     },
 ) {
@@ -34,7 +34,6 @@ export function ZodSchemaClassOf<
     )
 
     type TZodSchemaClassImpl = TZodSchemaClass<
-        Schema,
         SchemaT,
         SchemaUnknownKeys,
         SchemaCatchall,
@@ -44,7 +43,7 @@ export function ZodSchemaClassOf<
     >
 
     return class extends (of as unknown as ConcreteClass<any, []>) {
-        static readonly schema = schema
+        static readonly schema        = schema
         static readonly defaultValues = defaultValues
 
         constructor(values: Values) {
@@ -52,7 +51,7 @@ export function ZodSchemaClassOf<
             Object.assign(this, values)
         }
 
-        static async instantiate(
+        static instantiate(
             ...[values, params]: NewZodSchemaInstanceArgs<
                 NewZodSchemaInstanceInput<Values, DefaultValues>
             >
@@ -71,6 +70,52 @@ export function ZodSchemaClassOf<
                 await this.schema.parseAsync({ ...this.defaultValues, ...values }, params)
             )
         }
+
+        static instantiateEffect(
+            ...[values, params]: NewZodSchemaInstanceArgs<
+                NewZodSchemaInstanceInput<Values, DefaultValues>
+            >
+        ) {
+            return pipe(
+                parseZodTypeEffect(
+                    this.schema,
+                    { ...this.defaultValues, ...values },
+                    params,
+                ),
+
+                Effect.map(values => new this(values)),
+            )
+        }
+
+        static extend<
+            ExtendedSchemaT           extends z.ZodRawShape,
+            ExtendedSchemaUnknownKeys extends z.UnknownKeysParam,
+            ExtendedSchemaCatchall    extends z.ZodTypeAny,
+
+            ExtendedValues            extends Values,
+            ExtendedDefaultValues     extends Partial<ExtendedValues>,
+        >(
+            props: {
+                schema: (props: {
+                    schema:  z.ZodObject<SchemaT, SchemaUnknownKeys, SchemaCatchall, Values, Values>
+                    shape:   SchemaT
+                }) => z.ZodObject<ExtendedSchemaT, ExtendedSchemaUnknownKeys, ExtendedSchemaCatchall, ExtendedValues, ExtendedValues>
+
+                defaultValues: (defaultValues: DefaultValues) => Opaque<ExtendedDefaultValues, DefinedDefaultValuesTag>
+            },
+        ) {
+            const schema = props.schema({
+                schema:  this.schema,
+                shape:   this.schema.shape,
+            })
+
+            const defaultValues = props.defaultValues(this.defaultValues)
+
+            return class extends (this as unknown as ConcreteClass<any, []>) {
+                static readonly schema        = schema
+                static readonly defaultValues = defaultValues
+            }
+        }
     } as unknown as (
         Class<
             InstanceType<Parent> &