mod

ThinAirThings · May 3, 2024 · 6f21eba · 6f21eba
1 parent 60e4592
commit 6f21eba
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/base/defineNode.ts b/src/base/defineNode.ts
@@ -13,6 +13,8 @@ type Concrete<T extends Record<string, any>> = {
     [P in keyof T]: NonNullable<T[P]>;
 };
 
+
+type DefaultType<T> = T extends ZodTypeAny ? InferZodSchema<T> : never;
 export class NodeDefinition<T extends string, StateDefinition extends ZodObject<any>, StateDefaults extends ZodObject<any> = ZodObject<{}>> {
     nodeType: T;
     stateDefinition: StateDefinition;
@@ -26,7 +28,9 @@ export class NodeDefinition<T extends string, StateDefinition extends ZodObject<
 
     defaults<Defaults extends { [K in keyof TypeOf<StateDefinition>]?: TypeOf<StateDefinition>[K] }>(
         defaults: Defaults
-    ): NodeDefinition<T, StateDefinition, ZodObject<Concrete<Defaults>, any, any, Concrete<Defaults>>> {
+    ): NodeDefinition<T, StateDefinition, ZodObject<{
+        [K in keyof Defaults]: StateDefinition['shape'][K]
+    }>> {
         const defaultsDefinition = Object.entries(this.stateDefinition.shape).reduce((acc, [key, value]) => ({
             ...acc,
             [key]: (value as any).default(defaults[key])