🪲 Fix silly cases for UIntNumberSchema (#608)

LayerZero-Labs · May 9, 2024 · f3ab6c6 · f3ab6c6
1 parent 1e9b652
commit f3ab6c6
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/.changeset/slow-walls-impress.md b/.changeset/slow-walls-impress.md
@@ -0,0 +1,5 @@
+---
+"@layerzerolabs/devtools": patch
+---
+
+Update UIntNumberSchema to handle silly cases (like invalid toString() implementation)
diff --git a/packages/devtools/src/omnigraph/schema.ts b/packages/devtools/src/omnigraph/schema.ts
@@ -27,7 +27,21 @@ export const UIntBigIntSchema = z
     })
     .pipe(z.bigint().nonnegative())
 
-export const UIntNumberSchema = z.coerce.number().nonnegative().int()
+export const UIntNumberSchema = z
+    .unknown()
+    .transform((value, ctx): number => {
+        try {
+            return Number(value)
+        } catch {
+            ctx.addIssue({
+                code: z.ZodIssueCode.custom,
+                message: `Invalid Number-like value`,
+            })
+
+            return z.NEVER
+        }
+    })
+    .pipe(z.number().nonnegative().int())
 
 export const AddressSchema = z.string()