added zod type - uniforms for props

vazco · Apr 20, 2024 · 028672a · 028672a
1 parent 49715f4
commit 028672a
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 1 deletion.
diff --git a/packages/uniforms-bridge-zod/src/ZodBridge.ts b/packages/uniforms-bridge-zod/src/ZodBridge.ts
@@ -155,6 +155,14 @@ export default class ZodBridge<T extends ZodRawShape> extends Bridge {
     };
 
     let field = this.getField(name);
+
+    const uniforms = field._def.uniforms;
+    if (typeof uniforms === 'function') {
+      props.component = uniforms;
+    } else {
+      Object.assign(props, uniforms);
+    }
+
     if (field instanceof ZodDefault) {
       field = field.removeDefault();
       props.required = false;

diff --git a/packages/uniforms-bridge-zod/src/register.ts b/packages/uniforms-bridge-zod/src/register.ts
@@ -1,4 +1,5 @@
-import { filterDOMProps } from 'uniforms';
+import { ConnectedField, filterDOMProps, UnknownObject } from 'uniforms';
+import { z } from 'zod';
 
 // There's no possibility to retrieve them at runtime.
 declare module 'uniforms' {
@@ -9,3 +10,22 @@ declare module 'uniforms' {
 }
 
 filterDOMProps.register('minCount', 'maxCount');
+
+declare module 'zod' {
+  interface ZodType {
+    uniforms(uniforms: UnknownObject | ConnectedField<any>): this;
+  }
+
+  interface ZodTypeDef {
+    uniforms: UnknownObject | ConnectedField<any>;
+  }
+}
+
+z.ZodType.prototype.uniforms = function extend(uniforms) {
+  const This = (this as any).constructor;
+
+  return new This({
+    ...this._def,
+    uniforms,
+  });
+};