fixes bug where hidden ids weren't found when using optional etc

iway1 · Mar 7, 2023 · 16f6f46 · 16f6f46
1 parent a2f3962
commit 16f6f46
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 5 deletions.
diff --git a/src/__tests__/isZodTypeEqual.test.ts b/src/__tests__/isZodTypeEqual.test.ts
@@ -259,4 +259,11 @@ describe("isZodTypeEqual", () => {
 
     expect(isZodTypeEqual(A, B)).toStrictEqual(true);
   });
+  it("should return true if a unique field schema is compared with its optional", () => {
+    const A = createUniqueFieldSchema(z.string(), "text");
+    console.log("A def: ");
+    console.log(A._def);
+
+    expect(isZodTypeEqual(A, A.optional())).toStrictEqual(true);
+  });
 });
diff --git a/src/unwrap.tsx b/src/unwrap.tsx
@@ -25,12 +25,12 @@ export function unwrap(type: RTFSupportedZodTypes): {
   // Realized zod has a built in "unwrap()" function after writing this.
   // Not sure if it's super necessary.
   let r = type;
-  let hiddenId: null | string = null;
-  if (isSchemaWithHiddenProperties(type)) {
-    hiddenId = type._def[HIDDEN_ID_PROPERTY];
-  }
+  let unwrappedHiddenId: null | string = null;
 
   while (unwrappable.has(r._def.typeName)) {
+    if (isSchemaWithHiddenProperties(r)) {
+      unwrappedHiddenId = r._def[HIDDEN_ID_PROPERTY];
+    }
     switch (r._def.typeName) {
       case z.ZodFirstPartyTypeKind.ZodOptional:
         r = r._def.innerType;
@@ -49,9 +49,15 @@ export function unwrap(type: RTFSupportedZodTypes): {
     }
   }
 
+  let innerHiddenId: null | string = null;
+
+  if (isSchemaWithHiddenProperties(r)) {
+    innerHiddenId = r._def[HIDDEN_ID_PROPERTY];
+  }
+
   return {
     type: r,
-    [HIDDEN_ID_PROPERTY]: hiddenId,
+    [HIDDEN_ID_PROPERTY]: innerHiddenId || unwrappedHiddenId,
   };
 }