@ktrysmt/beautiful-mermaid 1.5.0 → 1.5.1

package/dist/index.js

@@ -481,8 +481,43 @@ function parseFlowchart(lines) {
     }
     parseEdgeLine(line, graph, subgraphStack);
   }
+  resolveSubgraphEdgeEndpoints(graph);
   return graph;
 }
+function resolveSubgraphEdgeEndpoints(graph) {
+  const representatives = /* @__PURE__ */ new Map();
+  function pickRepresentative(sg) {
+    for (const id of sg.nodeIds) {
+      if (graph.nodes.has(id)) return id;
+    }
+    for (const child of sg.children) {
+      const childRep = representatives.get(child.id) ?? pickRepresentative(child);
+      if (childRep) return childRep;
+    }
+    return void 0;
+  }
+  function visit(sg) {
+    for (const child of sg.children) visit(child);
+    const rep = pickRepresentative(sg);
+    if (rep) representatives.set(sg.id, rep);
+  }
+  for (const sg of graph.subgraphs) visit(sg);
+  if (representatives.size === 0) return;
+  for (const edge of graph.edges) {
+    const newSource = representatives.get(edge.source);
+    if (newSource) edge.source = newSource;
+    const newTarget = representatives.get(edge.target);
+    if (newTarget) edge.target = newTarget;
+  }
+  for (const sgId of representatives.keys()) {
+    graph.nodes.delete(sgId);
+  }
+  function scrubMembership(sg) {
+    sg.nodeIds = sg.nodeIds.filter((id) => !representatives.has(id));
+    for (const child of sg.children) scrubMembership(child);
+  }
+  for (const sg of graph.subgraphs) scrubMembership(sg);
+}
 function parseStateDiagram(lines) {
   const graph = {
     direction: "TD",