From 5aa852acbc52cc1b97a18d540e9d987ca890f620 Mon Sep 17 00:00:00 2001
From: yuetsh <517252939@qq.com>
Date: Thu, 7 May 2026 06:44:56 -0600
Subject: [PATCH] fix

---
 src/shared/composables/useMermaid.ts | 239 +++++++++++++++++++--------
 1 file changed, 168 insertions(+), 71 deletions(-)

diff --git a/src/shared/composables/useMermaid.ts b/src/shared/composables/useMermaid.ts
index e6b8a7c..d568f50 100644
--- a/src/shared/composables/useMermaid.ts
+++ b/src/shared/composables/useMermaid.ts
@@ -1,118 +1,205 @@
 import { getRandomId } from "utils/functions"
 
 const mermaidThemeVariables = {
-  primaryColor: "#eff6ff",
-  primaryTextColor: "#1d4ed8",
-  primaryBorderColor: "#3b82f6",
-  lineColor: "#94a3b8",
+  primaryColor: "#e0f2fe",
+  primaryTextColor: "#0f172a",
+  primaryBorderColor: "#0284c7",
+  lineColor: "#64748b",
+  secondaryColor: "#f5f3ff",
+  tertiaryColor: "#ecfdf5",
   background: "#ffffff",
-  mainBkg: "#eff6ff",
+  mainBkg: "#f8fafc",
+  secondBkg: "#eef2ff",
+  tertiaryBkg: "#f0fdfa",
+  nodeBorder: "#2563eb",
+  clusterBkg: "#f8fafc",
+  clusterBorder: "#cbd5e1",
+  edgeLabelBackground: "#ffffff",
   fontFamily:
     'Inter, ui-sans-serif, system-ui, -apple-system, BlinkMacSystemFont, "Segoe UI", sans-serif',
 }
 
+const semanticNodeClasses = [
+  "startNode",
+  "endNode",
+  "startEnd",
+  "input",
+  "output",
+  "process",
+  "decision",
+  "loop",
+]
+
 const displayStyleId = "oj-mermaid-display-style"
 
-const shapes = ["rect", "polygon", "ellipse", "circle", "path"]
-
-function nodeShapeRule(cls: string, fill: string, stroke: string) {
-  const sel = shapes
-    .map((s) => `.oj-mermaid-flowchart g.node.${cls} ${s}`)
-    .join(",\n")
-  return `${sel} { fill: ${fill} !important; stroke: ${stroke} !important; }`
-}
-
-function nodeLabelRule(cls: string, color: string) {
-  const bases = [".label", ".nodeLabel", ".nodeLabel p", ".label span"]
-  const sel = bases
-    .map((b) => `.oj-mermaid-flowchart g.node.${cls} ${b}`)
-    .join(",\n")
-  return `${sel} { color: ${color} !important; fill: ${color} !important; }`
-}
-
 const mermaidDisplayStyle = `
 .oj-mermaid-flowchart {
   max-width: 100%;
   height: auto;
 }
 
-/* Default node */
 .oj-mermaid-flowchart g.node rect,
 .oj-mermaid-flowchart g.node polygon,
 .oj-mermaid-flowchart g.node ellipse,
 .oj-mermaid-flowchart g.node circle,
 .oj-mermaid-flowchart g.node path {
-  fill: #eff6ff !important;
-  stroke: #3b82f6 !important;
-  stroke-width: 1.8px !important;
+  stroke-width: 2.5px !important;
+  filter: drop-shadow(0 6px 12px rgba(15, 23, 42, 0.12));
+}
+
+.oj-mermaid-flowchart g.node.startNode rect,
+.oj-mermaid-flowchart g.node.startNode polygon,
+.oj-mermaid-flowchart g.node.startNode ellipse,
+.oj-mermaid-flowchart g.node.startNode circle,
+.oj-mermaid-flowchart g.node.startNode path,
+.oj-mermaid-flowchart g.node.startEnd rect,
+.oj-mermaid-flowchart g.node.startEnd polygon,
+.oj-mermaid-flowchart g.node.startEnd ellipse,
+.oj-mermaid-flowchart g.node.startEnd circle,
+.oj-mermaid-flowchart g.node.startEnd path {
+  fill: #dcfce7 !important;
+  stroke: #16a34a !important;
+}
+
+.oj-mermaid-flowchart g.node.endNode rect,
+.oj-mermaid-flowchart g.node.endNode polygon,
+.oj-mermaid-flowchart g.node.endNode ellipse,
+.oj-mermaid-flowchart g.node.endNode circle,
+.oj-mermaid-flowchart g.node.endNode path {
+  fill: #fee2e2 !important;
+  stroke: #dc2626 !important;
+}
+
+.oj-mermaid-flowchart g.node.input rect,
+.oj-mermaid-flowchart g.node.input polygon,
+.oj-mermaid-flowchart g.node.input ellipse,
+.oj-mermaid-flowchart g.node.input circle,
+.oj-mermaid-flowchart g.node.input path {
+  fill: #dbeafe !important;
+  stroke: #2563eb !important;
+}
+
+.oj-mermaid-flowchart g.node.output rect,
+.oj-mermaid-flowchart g.node.output polygon,
+.oj-mermaid-flowchart g.node.output ellipse,
+.oj-mermaid-flowchart g.node.output circle,
+.oj-mermaid-flowchart g.node.output path {
+  fill: #ede9fe !important;
+  stroke: #7c3aed !important;
+}
+
+.oj-mermaid-flowchart g.node.process rect,
+.oj-mermaid-flowchart g.node.process polygon,
+.oj-mermaid-flowchart g.node.process ellipse,
+.oj-mermaid-flowchart g.node.process circle,
+.oj-mermaid-flowchart g.node.process path {
+  fill: #f0f9ff !important;
+  stroke: #0284c7 !important;
+}
+
+.oj-mermaid-flowchart g.node.decision rect,
+.oj-mermaid-flowchart g.node.decision polygon,
+.oj-mermaid-flowchart g.node.decision ellipse,
+.oj-mermaid-flowchart g.node.decision circle,
+.oj-mermaid-flowchart g.node.decision path {
+  fill: #fef3c7 !important;
+  stroke: #d97706 !important;
+}
+
+.oj-mermaid-flowchart g.node.loop rect,
+.oj-mermaid-flowchart g.node.loop polygon,
+.oj-mermaid-flowchart g.node.loop ellipse,
+.oj-mermaid-flowchart g.node.loop circle,
+.oj-mermaid-flowchart g.node.loop path {
+  fill: #fae8ff !important;
+  stroke: #c026d3 !important;
+}
+
+.oj-mermaid-flowchart g.node.oj-node-palette-0 rect,
+.oj-mermaid-flowchart g.node.oj-node-palette-0 polygon,
+.oj-mermaid-flowchart g.node.oj-node-palette-0 ellipse,
+.oj-mermaid-flowchart g.node.oj-node-palette-0 circle,
+.oj-mermaid-flowchart g.node.oj-node-palette-0 path {
+  fill: #dbeafe !important;
+  stroke: #2563eb !important;
+}
+
+.oj-mermaid-flowchart g.node.oj-node-palette-1 rect,
+.oj-mermaid-flowchart g.node.oj-node-palette-1 polygon,
+.oj-mermaid-flowchart g.node.oj-node-palette-1 ellipse,
+.oj-mermaid-flowchart g.node.oj-node-palette-1 circle,
+.oj-mermaid-flowchart g.node.oj-node-palette-1 path {
+  fill: #ccfbf1 !important;
+  stroke: #0d9488 !important;
+}
+
+.oj-mermaid-flowchart g.node.oj-node-palette-2 rect,
+.oj-mermaid-flowchart g.node.oj-node-palette-2 polygon,
+.oj-mermaid-flowchart g.node.oj-node-palette-2 ellipse,
+.oj-mermaid-flowchart g.node.oj-node-palette-2 circle,
+.oj-mermaid-flowchart g.node.oj-node-palette-2 path {
+  fill: #ede9fe !important;
+  stroke: #7c3aed !important;
+}
+
+.oj-mermaid-flowchart g.node.oj-node-palette-3 rect,
+.oj-mermaid-flowchart g.node.oj-node-palette-3 polygon,
+.oj-mermaid-flowchart g.node.oj-node-palette-3 ellipse,
+.oj-mermaid-flowchart g.node.oj-node-palette-3 circle,
+.oj-mermaid-flowchart g.node.oj-node-palette-3 path {
+  fill: #ffe4e6 !important;
+  stroke: #e11d48 !important;
+}
+
+.oj-mermaid-flowchart g.node.oj-node-palette-4 rect,
+.oj-mermaid-flowchart g.node.oj-node-palette-4 polygon,
+.oj-mermaid-flowchart g.node.oj-node-palette-4 ellipse,
+.oj-mermaid-flowchart g.node.oj-node-palette-4 circle,
+.oj-mermaid-flowchart g.node.oj-node-palette-4 path {
+  fill: #fef3c7 !important;
+  stroke: #d97706 !important;
+}
+
+.oj-mermaid-flowchart g.node.oj-node-palette-5 rect,
+.oj-mermaid-flowchart g.node.oj-node-palette-5 polygon,
+.oj-mermaid-flowchart g.node.oj-node-palette-5 ellipse,
+.oj-mermaid-flowchart g.node.oj-node-palette-5 circle,
+.oj-mermaid-flowchart g.node.oj-node-palette-5 path {
+  fill: #dcfce7 !important;
+  stroke: #16a34a !important;
 }
 
-/* Default node text */
 .oj-mermaid-flowchart g.node .label,
 .oj-mermaid-flowchart g.node .nodeLabel,
 .oj-mermaid-flowchart g.node .nodeLabel p,
 .oj-mermaid-flowchart g.node .label span {
-  color: #1d4ed8 !important;
-  fill: #1d4ed8 !important;
-  font-weight: 600 !important;
+  color: #0f172a !important;
+  fill: #0f172a !important;
+  font-weight: 650 !important;
 }
 
-/* startNode / startEnd */
-${nodeShapeRule("startNode", "#dcfce7", "#16a34a")}
-${nodeShapeRule("startEnd", "#dcfce7", "#16a34a")}
-${nodeLabelRule("startNode", "#166534")}
-${nodeLabelRule("startEnd", "#166534")}
-
-/* endNode */
-${nodeShapeRule("endNode", "#fee2e2", "#dc2626")}
-${nodeLabelRule("endNode", "#991b1b")}
-
-/* input */
-${nodeShapeRule("input", "#dbeafe", "#2563eb")}
-${nodeLabelRule("input", "#1e40af")}
-
-/* output */
-${nodeShapeRule("output", "#ede9fe", "#7c3aed")}
-${nodeLabelRule("output", "#5b21b6")}
-
-/* process */
-${nodeShapeRule("process", "#f1f5f9", "#64748b")}
-${nodeLabelRule("process", "#334155")}
-
-/* decision */
-${nodeShapeRule("decision", "#fef9c3", "#ca8a04")}
-${nodeLabelRule("decision", "#92400e")}
-
-/* loop */
-${nodeShapeRule("loop", "#fae8ff", "#c026d3")}
-${nodeLabelRule("loop", "#7e22ce")}
-
-/* Edges */
 .oj-mermaid-flowchart .edgePaths path.path,
 .oj-mermaid-flowchart .flowchart-link {
-  stroke: #94a3b8 !important;
-  stroke-width: 1.8px !important;
+  stroke: #64748b !important;
+  stroke-width: 2.4px !important;
 }
 
-/* Arrowheads */
 .oj-mermaid-flowchart marker path,
 .oj-mermaid-flowchart .marker {
-  fill: #94a3b8 !important;
-  stroke: #94a3b8 !important;
+  fill: #64748b !important;
+  stroke: #64748b !important;
 }
 
-/* Edge label background */
 .oj-mermaid-flowchart .edgeLabel rect,
 .oj-mermaid-flowchart .edgeLabel .labelBkg {
-  fill: rgba(255, 255, 255, 0.9) !important;
-  stroke: #e2e8f0 !important;
+  fill: rgba(255, 255, 255, 0.94) !important;
+  stroke: #cbd5e1 !important;
 }
 
-/* Edge label text */
 .oj-mermaid-flowchart .edgeLabel,
 .oj-mermaid-flowchart .edgeLabel span,
 .oj-mermaid-flowchart .edgeLabel p {
-  color: #475569 !important;
+  color: #334155 !important;
   font-weight: 600 !important;
 }
 `
@@ -127,6 +214,16 @@ function applyFlowchartDisplayStyle(container: HTMLElement) {
 
   svg.classList.add("oj-mermaid-flowchart")
 
+  const nodes = Array.from(svg.querySelectorAll<SVGGElement>("g.node"))
+  nodes.forEach((node, index) => {
+    const hasSemanticClass = semanticNodeClasses.some((className) =>
+      node.classList.contains(className),
+    )
+    if (!hasSemanticClass) {
+      node.classList.add(`oj-node-palette-${index % 6}`)
+    }
+  })
+
   svg.querySelector(`#${displayStyleId}`)?.remove()
   const style = document.createElementNS(svgNamespace, "style")
   style.setAttribute("id", displayStyleId)