Merge pull request #144 from asteasolutions/fix/deep-partial-loosing-…

…metadata

#132 preserve schema metadata for deepPartial

spec/routes/parameters.spec.ts

@@ -21,6 +21,37 @@ describe('parameters', () => {
       ]);
     });
 
+    it('generates a deepPartial object query parameter for route', () => {
+      const { parameters } = generateDataForRoute({
+        request: {
+          query: z
+            .object({
+              filter: z
+                .object({ test: z.string() })
+                .openapi({ param: { style: 'deepObject' } }),
+            })
+            .deepPartial(),
+        },
+      });
+
+      expect(parameters).toEqual([
+        {
+          in: 'query',
+          name: 'filter',
+          required: false,
+          style: 'deepObject',
+          schema: {
+            type: 'object',
+            properties: {
+              test: {
+                type: 'string',
+              },
+            },
+          },
+        },
+      ]);
+    });
+
     it('generates a reference query parameter for route', () => {
       const TestQuery = registerParameter(
         'TestQuery',

src/zod-extensions.ts

@@ -125,6 +125,21 @@ export function extendZodWithOpenApi(zod: typeof z) {
     return result;
   };
 
+  const zodDeepPartial = zod.ZodObject.prototype.deepPartial;
+  zod.ZodObject.prototype.deepPartial = function (this: any) {
+    const initialShape = this._def.shape();
+
+    const result = zodDeepPartial.apply(this);
+
+    const resultShape = result._def.shape();
+
+    Object.entries(resultShape).forEach(([key, value]) => {
+      value._def.openapi = initialShape[key]?._def?.openapi;
+    });
+
+    return result;
+  };
+
   const zodOptional = zod.ZodType.prototype.optional as any;
   (zod.ZodType.prototype.optional as any) = function (
     this: any,