Merge pull request #41 from iway1/bug/refine-transform

fix typing issues with refine and transform usage

src/__tests__/createSchemaForm.test.tsx

@@ -22,6 +22,7 @@ import {
   useReqDescription,
   useTsController,
 } from "../FieldContext";
+import { expectTypeOf } from "expect-type";
 import { createUniqueFieldSchema } from "../createFieldSchema";
 
 const testIds = {
@@ -767,7 +768,8 @@ describe("createSchemaForm", () => {
       render(<Form schema={Schema} onSubmit={() => {}} />);
     }).toThrow();
   });
-  it("should correctly type the form with multiple unique field schemas.", () => {
+
+  it("should have correct typings with multiple unique field schemas when transform and refine are used.", () => {
     const A = createUniqueFieldSchema(z.string(), "one");
     const B = createUniqueFieldSchema(z.string(), "two");
     function In1(_: { req: string }) {
@@ -784,11 +786,16 @@ describe("createSchemaForm", () => {
     const Form = createTsForm(mapping);
 
     <Form
-      schema={z.object({
-        a: A,
-        b: B,
-      })}
-      onSubmit={() => {}}
+      schema={z
+        .object({
+          a: A,
+          b: B,
+        })
+        .refine((_) => true)
+        .transform((a) => a.a)}
+      onSubmit={(data) => {
+        expectTypeOf(data).toBeString();
+      }}
       props={{
         a: {
           req: "One",

src/createSchemaForm.tsx

@@ -90,8 +90,10 @@ export type ExtraProps = {
 type UnwrapEffects<T extends AnyZodObject | ZodEffects<any, any>> =
   T extends AnyZodObject
     ? T
-    : T extends ZodEffects<any, any>
-    ? T["_def"]["schema"]
+    : T extends ZodEffects<infer EffectsSchema, any>
+    ? EffectsSchema extends ZodEffects<infer EffectsSchemaInner, any>
+      ? EffectsSchemaInner
+      : EffectsSchema
     : never;
 
 function checkForDuplicateTypes(array: RTFSupportedZodTypes[]) {
@@ -237,7 +239,7 @@ export function createTsForm<
     /**
      * A callback function that will be called with the data once the form has been submitted and validated successfully.
      */
-    onSubmit: (values: z.infer<UnwrapEffects<SchemaType>>) => void;
+    onSubmit: (values: z.infer<SchemaType>) => void;
     /**
      * Initializes your form with default values. Is a deep partial, so all properties and nested properties are optional.
      */
@@ -296,7 +298,7 @@ export function createTsForm<
      */
     props?: RequireKeysWithRequiredChildren<
       Partial<{
-        [key in keyof z.infer<SchemaType>]: Mapping[IndexOf<
+        [key in keyof z.infer<UnwrapEffects<SchemaType>>]: Mapping[IndexOf<
           UnwrapMapping<Mapping>,
           readonly [
             IndexOfUnwrapZodType<