From 35a8037f5a378cbafa136021b1e4858e5612a927 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Julien=20Valverd=C3=A9?= <julien.valverde@mailo.com>
Date: Thu, 17 Apr 2025 02:30:23 +0200
Subject: [PATCH] Form work

---
 .../extension-form/src/internal/FormField.ts  | 39 ++++++++-----------
 .../extension-form/src/internal/FormTree.ts   | 12 +++---
 2 files changed, 22 insertions(+), 29 deletions(-)

diff --git a/packages/extension-form/src/internal/FormField.ts b/packages/extension-form/src/internal/FormField.ts
index fce5fa2..b9320f0 100644
--- a/packages/extension-form/src/internal/FormField.ts
+++ b/packages/extension-form/src/internal/FormField.ts
@@ -11,34 +11,27 @@ export interface GenericFormField<S extends Schema.Schema.Any> extends FormField
     readonly value: S["Type"]
 }
 
-export interface TupleFormField<S extends Schema.Tuple<readonly Schema.Schema.Any[]>> extends FormField<S> {
+export interface TupleFormField<
+    S extends Schema.TupleType<Elements, Rest>,
+    Elements extends Schema.TupleType.Elements,
+    Rest extends Schema.TupleType.Rest,
+> extends FormField<S> {
     readonly _tag: "TupleFormField"
-    readonly elements: { readonly [K in keyof S["elements"]]: FormTree.FormTree<
-        S["elements"][K] extends Schema.Schema.Any
-            ? S["elements"][K]
-            : never
-    > }
+    readonly elements: [...{ readonly [K in keyof Elements]: FormTree.FormTree<Elements[K]> }]
 }
 
-export interface Tuple2FormField<S extends Schema.Tuple2<Schema.Schema.Any, Schema.Schema.Any>> extends FormField<S> {
-    readonly _tag: "Tuple2FormField"
-    readonly elements: { readonly [K in keyof S["elements"]]: FormTree.FormTree<
-        S["elements"][K] extends Schema.Schema.Any
-            ? S["elements"][K]
-            : never
-    > }
-}
-
-export interface ArrayFormField<S extends Schema.Array$<Schema.Schema.AnyNoContext>> extends FormField<S> {
+export interface ArrayFormField<
+    S extends Schema.Array$<Value>,
+    Value extends Schema.Schema.Any,
+> extends FormField<S> {
     readonly _tag: "ArrayFormField"
-    readonly elements: readonly FormTree.FormTree<S["value"]>[]
+    readonly elements: readonly FormTree.FormTree<Value>[]
 }
 
-export interface StructFormField<S extends Schema.Struct<{
-    readonly [x: string]: Schema.Schema.AnyNoContext
-    readonly [x: number]: Schema.Schema.AnyNoContext
-    readonly [x: symbol]: Schema.Schema.AnyNoContext
-}>> extends FormField<S> {
+export interface StructFormField<
+    S extends Schema.Struct<Fields>,
+    Fields extends Schema.Struct.Fields,
+> extends FormField<S> {
     readonly _tag: "StructFormField"
-    readonly fields: { readonly [K in keyof S["fields"]]: FormTree.FormTree<S["fields"][K]> }
+    readonly fields: { readonly [K in keyof Fields]: FormTree.FormTree<Fields[K]> }
 }
diff --git a/packages/extension-form/src/internal/FormTree.ts b/packages/extension-form/src/internal/FormTree.ts
index e33ef0b..ec7f4ef 100644
--- a/packages/extension-form/src/internal/FormTree.ts
+++ b/packages/extension-form/src/internal/FormTree.ts
@@ -2,12 +2,12 @@ import { Schema } from "effect"
 import type * as FormField from "./FormField.js"
 
 
-export type FormTree<S extends Schema.Schema.Any> = (
-    S extends Schema.Tuple<any> ? FormField.TupleFormField<S> :
-    S extends Schema.Tuple2<any, any> ? FormField.Tuple2FormField<S> :
-    S extends Schema.Array$<any> ? FormField.ArrayFormField<S> :
-    S extends Schema.Struct<any> ? FormField.StructFormField<S> :
-    FormField.GenericFormField<S>
+export type FormTree<S> = (
+    S extends Schema.TupleType<infer Elements, infer Rest> ? FormField.TupleFormField<S, Elements, Rest> :
+    S extends Schema.Array$<infer Value> ? FormField.ArrayFormField<S, Value> :
+    S extends Schema.Struct<infer Fields> ? FormField.StructFormField<S, Fields> :
+    S extends Schema.Schema.Any ? FormField.GenericFormField<S> :
+    never
 )