src/ExtendZodSchemaClass.ts

@@ -0,0 +1,56 @@
+import { TraitInstance, TraitStaticMembers, expression } from "@thilawyn/traitify-ts"
+import { AbstractClass } from "type-fest"
+import { z } from "zod"
+import { ZodSchemaObject, ZodSchemaObjectTrait } from "./lib"
+import { Extend, StaticMembers } from "./util"
+
+
+export function ExtendZodSchemaClass<
+    Self                  extends (
+        AbstractClass<
+            TraitInstance<ZodSchemaObjectTrait<T, Catchall, Values, PartialValues>> & Values,
+            [values: Values]
+        > &
+        TraitStaticMembers<ZodSchemaObjectTrait<T, Catchall, Values, PartialValues>>
+    ),
+
+    T                     extends z.ZodRawShape,
+    Catchall              extends z.ZodTypeAny,
+    Values                extends object,
+    PartialValues         extends Partial<Values>,
+
+    ExtendedT             extends z.ZodRawShape,
+    ExtendedCatchall      extends z.ZodTypeAny,
+    ExtendedValues        extends Values,
+    ExtendedPartialValues extends Partial<ExtendedValues>,
+>(
+    class_: Self | (
+        AbstractClass<
+            TraitInstance<ZodSchemaObjectTrait<T, Catchall, Values, PartialValues>> & Values,
+            [values: Values]
+        > &
+        TraitStaticMembers<ZodSchemaObjectTrait<T, Catchall, Values, PartialValues>>
+    ),
+
+    schemaWithDefaults: (
+        schemaWithDefaults: z.ZodObject<T, "strip", Catchall, Values, PartialValues>
+    ) => z.ZodObject<ExtendedT, "strip", ExtendedCatchall, ExtendedValues, ExtendedPartialValues>,
+) {
+    return expression
+        .extends(
+            class_ as unknown as (
+                AbstractClass<
+                    Omit<
+                        Extend<[InstanceType<Self>, ExtendedValues]>,
+                        keyof TraitInstance<ZodSchemaObjectTrait<T, Catchall, Values, PartialValues>>
+                    >,
+                    [values: ExtendedValues]
+                > &
+                Omit<
+                    StaticMembers<Self>,
+                    keyof TraitStaticMembers<ZodSchemaObjectTrait<T, Catchall, Values, PartialValues>>
+                >
+            )
+        )
+        .expresses(ZodSchemaObject(schemaWithDefaults((class_ as Self).schemaWithDefaults)))
+}

src/ZodSchemaClass.ts

@@ -1,7 +1,7 @@
 import { expression } from "@thilawyn/traitify-ts"
 import { Class } from "type-fest"
 import { z } from "zod"
-import { ExtendableZodSchemaObject } from "./traits/ExtendableZodSchemaObject"
+import { ZodSchemaObject } from "./traits/ZodSchemaObject"
 
 
 export function ZodSchemaClass<
@@ -20,5 +20,5 @@ export function ZodSchemaClass<
                 }
             } as Class<Values, [values: Values]>
         )
-        .expresses(ExtendableZodSchemaObject(schemaWithDefaults))
+        .expresses(ZodSchemaObject(schemaWithDefaults))
 }

src/lib.ts

@@ -1,6 +1,6 @@
+export { ExtendZodSchemaClass } from "./ExtendZodSchemaClass"
 export { JsonifiedZodSchemaClass } from "./JsonifiedZodSchemaClass"
 export { ZodSchemaClass } from "./ZodSchemaClass"
-export { ExtendableZodSchemaObject } from "./traits/ExtendableZodSchemaObject"
 export { JsonifiedZodSchemaObject } from "./traits/JsonifiedZodSchemaObject"
 export { MobXObservableZodSchemaObject } from "./traits/MobXObservableZodSchemaObject"
 export { ZodSchemaObject, ZodSchemaObjectTrait } from "./traits/ZodSchemaObject"

src/tests.ts

@@ -1,6 +1,7 @@
 import { Implements } from "@thilawyn/traitify-ts"
 import { Option } from "effect"
 import { z } from "zod"
+import { ExtendZodSchemaClass } from "./ExtendZodSchemaClass"
 import { JsonifiedZodSchemaClass } from "./JsonifiedZodSchemaClass"
 import { ZodSchemaClass } from "./ZodSchemaClass"
 import { s } from "./schema/lib"
@@ -47,9 +48,11 @@ console.log(JSON.stringify(jsonifiedUserInst))
 // jsonifiedUserInst.dejsonify()
 
 
-const adminUserExp = User.extend(s => s.extend({
+const adminUserExp = ExtendZodSchemaClass(User,
+    schema => schema.extend({
         role: z.literal("Admin").default("Admin")
-})).build()
+    })
+).build()
 
 @adminUserExp.staticImplements
 class AdminUser extends adminUserExp.extends implements Implements<typeof adminUserExp> {}

src/traits/MobXObservableZodSchemaObject.ts

@@ -8,13 +8,15 @@ export const MobXObservableZodSchemaObject = trait
     .staticAbstract(Super => class extends Super {
         declare readonly schema: z.ZodObject<z.ZodRawShape, z.UnknownKeysParam, z.ZodTypeAny, object, object>
     })
-    .implement(Super => class ObservableZodSchemaObject extends Super {
+    .implement(Super => class MobXObservableZodSchemaObjectImpl extends Super {
+        declare ["constructor"]: typeof MobXObservableZodSchemaObjectImpl
+
         constructor(...args: any[]) {
             super(...args)
 
             makeObservable(this,
                 mapValues(
-                    (this.constructor as typeof ObservableZodSchemaObject).schema.shape,
+                    this.constructor.schema.shape,
                     () => observable,
                 )
             )