From 147034e95d443ea2f86c4f65a93b6139bf6563ba Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Julien=20Valverd=C3=A9?= <julien.valverde@mailo.com>
Date: Wed, 24 Jan 2024 21:49:19 +0100
Subject: [PATCH] ZodSchemaClassOf refactoring

---
 src/ZodSchemaClass.ts | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/src/ZodSchemaClass.ts b/src/ZodSchemaClass.ts
index 9886e1f..d2dbf62 100644
--- a/src/ZodSchemaClass.ts
+++ b/src/ZodSchemaClass.ts
@@ -2,7 +2,7 @@ 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, parseZodTypeEffect } from "./util"
+import { Class, GetClassType, StaticMembers, parseZodTypeEffect } from "./util"
 
 
 type ExtendableClass = AbstractClass<any, []> & {
@@ -27,12 +27,6 @@ export function ZodSchemaClassOf<
         defaultValues:  Opaque<DefaultValues, DefinedDefaultValuesTag>
     },
 ) {
-    type Class<T, Arguments extends unknown[]> = (
-        Parent extends ConcreteClass<any>
-            ? ConcreteClass<T, Arguments>
-            : AbstractClass<T, Arguments>
-    )
-
     type TZodSchemaClassImpl = TZodSchemaClass<
         SchemaT,
         SchemaUnknownKeys,
@@ -118,6 +112,8 @@ export function ZodSchemaClassOf<
         }
     } as unknown as (
         Class<
+            GetClassType<Parent>,
+
             InstanceType<Parent> &
             InstanceType<TZodSchemaClassImpl>,