zod-nest 2.1.2 → 2.1.3

package/dist/index.mjs

@@ -434,6 +434,60 @@ var rewriteRefs = /* @__PURE__ */ __name((node, selfRef) => {
     rewriteRefs(value, selfRef);
   }
 }, "rewriteRefs");
+var refsIn = /* @__PURE__ */ __name(function* (node) {
+  if (Array.isArray(node)) {
+    for (const item of node) {
+      yield* refsIn(item);
+    }
+    return;
+  }
+  if (node === null || typeof node !== "object") {
+    return;
+  }
+  const obj = node;
+  if (typeof obj.$ref === "string") {
+    yield obj.$ref;
+  }
+  for (const value of Object.values(obj)) {
+    yield* refsIn(value);
+  }
+}, "refsIn");
+var rootDefId = /* @__PURE__ */ __name((raw) => {
+  const ref = raw.$ref;
+  if (typeof ref !== "string" || !ref.startsWith(DEFS_PREFIX)) {
+    return void 0;
+  }
+  return ref.slice(DEFS_PREFIX.length);
+}, "rootDefId");
+var isReferenced = /* @__PURE__ */ __name((defs, id) => {
+  const ownRef = `${DEFS_PREFIX}${id}`;
+  for (const ref of refsIn(defs)) {
+    if (ref === "#" || ref === ownRef) {
+      return true;
+    }
+  }
+  return false;
+}, "isReferenced");
+var liftRoot = /* @__PURE__ */ __name((raw, refs) => {
+  const id = rootDefId(raw);
+  if (id === void 0) {
+    return void 0;
+  }
+  const body = refs.get(id);
+  if (body === void 0 || isReferenced(raw.$defs, id)) {
+    return void 0;
+  }
+  refs.delete(id);
+  return body;
+}, "liftRoot");
+var stripEnvelope = /* @__PURE__ */ __name((raw) => {
+  const root = {
+    ...raw
+  };
+  delete root.$schema;
+  delete root.$defs;
+  return root;
+}, "stripEnvelope");
 var postProcess = /* @__PURE__ */ __name((raw) => {
   const refs = /* @__PURE__ */ new Map();
   const rawDefs = raw.$defs;
@@ -442,11 +496,7 @@ var postProcess = /* @__PURE__ */ __name((raw) => {
       refs.set(id, body);
     }
   }
-  const root = {
-    ...raw
-  };
-  delete root.$schema;
-  delete root.$defs;
+  const root = liftRoot(raw, refs) ?? stripEnvelope(raw);
   rewriteRefs(root, void 0);
   for (const [id, body] of refs) {
     rewriteRefs(body, `${COMPONENTS_SCHEMAS_PREFIX}${id}`);