src/ZodSchemaClass.ts

@@ -6,7 +6,7 @@ import { Extend, StaticMembers } from "./util"
 
 
 export function ZodSchemaClassOf<
-    Superclass        extends AbstractClass<{}, []>,
+    Super             extends AbstractClass<{}, []>,
 
     SchemaT           extends z.ZodRawShape,
     SchemaUnknownKeys extends z.UnknownKeysParam,
@@ -15,7 +15,7 @@ export function ZodSchemaClassOf<
     Values            extends {},
     DefaultValues     extends Partial<Values>,
 >(
-    of: Superclass,
+    of: Super,
 
     { schema, defaultValues }: {
         schema:         z.ZodObject<SchemaT, SchemaUnknownKeys, SchemaCatchall, Values, Values>
@@ -37,7 +37,7 @@ export function ZodSchemaClassOf<
 
     return exp.extends as AbstractClass<
         Extend<[
-            InstanceType<Superclass>,
+            InstanceType<Super>,
             InstanceType<typeof exp.extends>,
             Values,
         ]>,
@@ -45,7 +45,7 @@ export function ZodSchemaClassOf<
         ConstructorParameters<typeof exp.extends>
     > &
     Extend<[
-        StaticMembers<Superclass>,
+        StaticMembers<Super>,
         StaticMembers<typeof exp.extends>,
     ]>
 }

src/traits/ExtendableZodSchemaObject.ts

@@ -0,0 +1,39 @@
+import { abstract, trait } from "@thilawyn/traitify-ts"
+import { z } from "zod"
+import { ZodSchemaAbstractClass } from "../types/ZodSchemaClass"
+
+
+export const ExtendableZodSchemaObject = trait(
+    abstract(),
+
+    Super => class ExtendableZodSchemaObject extends Super {
+        static extend<
+            Super                  extends ZodSchemaAbstractClass<SuperSchemaT, SuperSchemaUnknownKeys, SuperSchemaCatchall, SuperValues, SuperDefaultValues>,
+
+            SuperSchemaT           extends z.ZodRawShape,
+            SuperSchemaUnknownKeys extends z.UnknownKeysParam,
+            SuperSchemaCatchall    extends z.ZodTypeAny,
+            SuperValues            extends {},
+            SuperDefaultValues     extends Partial<SuperValues>,
+
+            SchemaT                extends z.ZodRawShape,
+            SchemaUnknownKeys      extends z.UnknownKeysParam,
+            SchemaCatchall         extends z.ZodTypeAny,
+            Values                 extends SuperValues,
+            DefaultValues          extends Partial<Values>,
+        >(
+            this: Super | ZodSchemaAbstractClass<SuperSchemaT, SuperSchemaUnknownKeys, SuperSchemaCatchall, SuperValues, SuperDefaultValues>,
+
+            props: {
+                schema: (props: {
+                    schema:  Super["schema"]
+                    shape:   Super["schema"]["shape"]
+                }) => z.ZodObject<SchemaT, SchemaUnknownKeys, SchemaCatchall, Values, Values>
+
+                defaultValues: (defaultValues: SuperDefaultValues) => DefaultValues
+            },
+        ) {
+
+        }
+    },
+)