feat: allow drilling through optional and nullable

atmina · Feb 8, 2024 · f518691 · f518691
1 parent 3084bce
commit f518691
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/src/formbuilder.tsx b/src/formbuilder.tsx
@@ -81,7 +81,9 @@ export type FormBuilder<T> = FormBuilderRegisterFn<T> & {
         ): $UseFieldArrayReturn<TItem>;
       }
     : {
-        [K in Exclude<keyof T, `${"__"}${string}`>]-?: FormBuilder<T[K]>;
+        [K in Exclude<keyof T, `${"__"}${string}`>]-?: FormBuilder<
+          T[K] extends Primitive ? T[K] : NonNullable<T[K]>
+        >;
       });
 
 /*

diff --git a/src/types.test.tsx b/src/types.test.tsx
@@ -38,6 +38,14 @@ describe("Types", () => {
       FormBuilder<{ foo?: string }>
     >();
 
+    // Nested optional
+    type NestedOptional = FormBuilder<{
+      foo?: { bar?: { baz?: string } };
+    }>;
+    expectTypeOf<NestedOptional["foo"]["bar"]["baz"]>().toMatchTypeOf<
+      FormBuilder<string>
+    >();
+
     // Nullable
     expectTypeOf<FormBuilder<string | null>>().toMatchTypeOf<
       FormBuilder<string>
@@ -46,6 +54,14 @@ describe("Types", () => {
       FormBuilder<string | null>
     >();
 
+    // Nested nullable
+    type NestedNull = FormBuilder<{
+      foo: { bar: { baz: string | null } | null } | null;
+    }>;
+    expectTypeOf<NestedNull["foo"]["bar"]["baz"]>().toMatchTypeOf<
+      FormBuilder<string>
+    >();
+
     // Undefined
     expectTypeOf<FormBuilder<string | undefined>>().toMatchTypeOf<
       FormBuilder<string>