zodselect 1.0.14 → 1.0.15

package/dist/build-select-object.d.ts

@@ -6,9 +6,9 @@ export type ZodSelectAll<T extends ZodType<object> | ZodUnion> = {
 /**
  * Produces a runtime select-all object for the given Zod schema.
  *
- * Every top-level key is set to `true`, which tells the server to include
- * that field in the response.  For union schemas the result is the merged
- * set of keys across all variants.
+ * Every scalar key is set to `true`. Nested object fields are recursively
+ * expanded into nested select objects. Array fields remain `true`.
+ * For union schemas the result is the merged set of keys across all variants.
  *
  * @example
  * ```ts

package/dist/build-select-object.js

@@ -1,10 +1,11 @@
+import { ZodCatch, ZodDefault, ZodLazy, ZodNullable, ZodOptional } from 'zod';
 import { getInnerType, isZodObject, isZodUnion } from './utilities.js';
 /**
  * Produces a runtime select-all object for the given Zod schema.
  *
- * Every top-level key is set to `true`, which tells the server to include
- * that field in the response.  For union schemas the result is the merged
- * set of keys across all variants.
+ * Every scalar key is set to `true`. Nested object fields are recursively
+ * expanded into nested select objects. Array fields remain `true`.
+ * For union schemas the result is the merged set of keys across all variants.
  *
  * @example
  * ```ts
@@ -17,20 +18,47 @@ export function buildSelectObject(schema) {
     collectKeys(schema, result);
     return result;
 }
-function collectKeys(schema, result) {
+/**
+ * Unwraps optional/nullable/default/catch/lazy wrappers but NOT arrays.
+ */
+function unwrapField(schema) {
+    if (schema instanceof ZodOptional
+        || schema instanceof ZodNullable
+        || schema instanceof ZodDefault
+        || schema instanceof ZodCatch) {
+        return unwrapField(schema.def.innerType);
+    }
+    if (schema instanceof ZodLazy) {
+        return unwrapField(schema.def.getter());
+    }
+    return schema;
+}
+const MAX_DEPTH = 10;
+function collectKeys(schema, result, depth = 0) {
+    if (depth >= MAX_DEPTH) {
+        return;
+    }
     schema = getInnerType(schema);
     if (isZodObject(schema)) {
         for (const key in schema.shape) {
-            result[key] = true;
+            const fieldInner = unwrapField(schema.shape[key]);
+            if (isZodObject(fieldInner)) {
+                const nested = {};
+                collectKeys(fieldInner, nested, depth + 1);
+                result[key] = nested;
+            }
+            else {
+                result[key] = true;
+            }
         }
     }
     else if (isZodUnion(schema)) {
         for (const option of schema.options) {
-            collectKeys(option, result);
+            collectKeys(option, result, depth);
         }
     }
     else if (schema.def.type === 'intersection') {
-        collectKeys(schema.def.left, result);
-        collectKeys(schema.def.right, result);
+        collectKeys(schema.def.left, result, depth);
+        collectKeys(schema.def.right, result, depth);
     }
 }

package/dist/build-select-object.spec.js

@@ -45,7 +45,7 @@ describe('.buildSelectObject()', () => {
             c: true
         });
     });
-    it('should only collect top-level keys (not nested)', () => {
+    it('should recursively expand nested objects', () => {
         const schema = z.object({
             active: z.boolean(),
             user: z.object({
@@ -53,6 +53,69 @@ describe('.buildSelectObject()', () => {
                 name: z.string()
             })
         });
-        assert.deepStrictEqual(buildSelectObject(schema), { active: true, user: true });
+        assert.deepStrictEqual(buildSelectObject(schema), {
+            active: true,
+            user: { age: true, name: true }
+        });
+    });
+    it('should recursively expand deeply nested objects', () => {
+        const schema = z.object({
+            name: z.string(),
+            organization: z.object({
+                _id: z.string(),
+                address: z.object({
+                    city: z.string(),
+                    country: z.string()
+                })
+            })
+        });
+        assert.deepStrictEqual(buildSelectObject(schema), {
+            name: true,
+            organization: { _id: true, address: { city: true, country: true } }
+        });
+    });
+    it('should expand optional nested objects', () => {
+        const schema = z.object({
+            name: z.string(),
+            organization: z.object({
+                _id: z.string(),
+                name: z.string()
+            }).optional()
+        });
+        assert.deepStrictEqual(buildSelectObject(schema), {
+            name: true,
+            organization: { _id: true, name: true }
+        });
+    });
+    it('should not expand arrays', () => {
+        const schema = z.object({
+            name: z.string(),
+            tags: z.array(z.string())
+        });
+        assert.deepStrictEqual(buildSelectObject(schema), { name: true, tags: true });
+    });
+    it('should not expand arrays of objects', () => {
+        const schema = z.object({
+            items: z.array(z.object({ id: z.string(), value: z.number() })),
+            name: z.string()
+        });
+        assert.deepStrictEqual(buildSelectObject(schema), { items: true, name: true });
+    });
+    it('should handle recursive schemas without infinite loop', () => {
+        ;
+        const node = z.lazy(() => z.object({
+            children: z.array(node).optional(),
+            name: z.string(),
+            parent: node.optional()
+        }));
+        const schema = z.object({
+            name: z.string(),
+            root: node
+        });
+        // Should not throw or hang — recursive objects hit depth limit
+        const result = buildSelectObject(schema);
+        assert.strictEqual(result.name, true);
+        // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
+        assert.ok(typeof result.root === 'object' && result.root !== null);
     });
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zodselect",
-  "version": "1.0.14",
+  "version": "1.0.15",
   "description": "Type-safe selection utility built on top of Zod. Allows for strongly typed field selection, schema merging, and runtime refinement for frontend or backend use.",
   "author": "SecureDataSystems",
   "license": "MIT",