From 4672a7e0b8b296da0f49349080761b983f7f8e71 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Julien=20Valverd=C3=A9?= <julien.valverde@mailo.com>
Date: Sat, 24 Feb 2024 23:37:42 +0100
Subject: [PATCH] ZodSchemaClassBuilder work

---
 src/ZodSchemaClass.ts | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/src/ZodSchemaClass.ts b/src/ZodSchemaClass.ts
index 1bc86c4..bb9e065 100644
--- a/src/ZodSchemaClass.ts
+++ b/src/ZodSchemaClass.ts
@@ -140,7 +140,7 @@ export interface ZodSchemaClassBuilder<
         //                                        \/ Ensures `extends` can only be called once at the beginning
         this:        ZodSchemaClassBuilder<typeof TraitExpression.NullSuperclass, Traits>,
         superclass:  Super,
-    ): InstanceType<typeof this.constructor<Super, Traits>>
+    ): ZodSchemaClassBuilder<Super, Traits>
 
     expresses<
         const T extends readonly Trait<
@@ -158,4 +158,21 @@ export interface ZodSchemaClassBuilder<
         Superclass,
         TraitExpressionBuilder.ExpressesReturnTypeTraits<Traits, T>
     >
+
+    expressesFirst<
+        const T extends readonly Trait<
+            TraitExpression<
+                typeof TraitExpression.NullSuperclass,
+                readonly Trait<any, any, any, any>[]
+            >,
+            any,
+            any,
+            any
+        >[]
+    >(
+        ...traits: T
+    ): ZodSchemaClassBuilder<
+        Superclass,
+        TraitExpressionBuilder.ExpressesFirstReturnTypeTraits<Traits, T>
+    >
 }