chore: 排除mermaid渲染的svg标签净化

src/components/markdown-render/utils/markdown-parser.ts

@@ -17,6 +17,7 @@ const katexOptions: MarkedKatexOptions = {
   output: 'html',
 }
 
+// Mermaid初始化
 mermaid.initialize({
   startOnLoad: false,
   htmlLabels: true,
@@ -167,12 +168,40 @@ export function createMarkedInst() {
         },
         postprocess(html: string) {
           initMermaidPanzoom()
-          return DOMPurify.sanitize(html)
+          return getSanitizedHtml(html)
         },
       },
     })
 }
 
+// 获取净化HTML, 排除Mermaid render的svg标签
+function getSanitizedHtml(html: string) {
+  const mermaidElementRegex = /<div class="mermaid-viewer"><svg(.*?)>(.*?)<\/svg><\/div>/gs
+  const mermaidObjects: { placeholder: string; content: string }[] = []
+  let processedHtml = html
+  let match: RegExpExecArray | null = null
+  let i = 0
+
+  while ((match = mermaidElementRegex.exec(html)) !== null) {
+    const fullMatch = match[0]
+
+    const placeholder = `MERMAID_PLACEHOLDER_${i}`
+    mermaidObjects.push({ placeholder, content: fullMatch })
+
+    processedHtml = processedHtml.replace(fullMatch, placeholder)
+    i++
+  }
+
+  const sanitizedHtmlWithoutMermaid = DOMPurify.sanitize(processedHtml)
+
+  let sanitizedHtml = sanitizedHtmlWithoutMermaid
+  for (const mermaidItem of mermaidObjects) {
+    sanitizedHtml = sanitizedHtml.replace(mermaidItem.placeholder, mermaidItem.content)
+  }
+
+  return sanitizedHtml
+}
+
 export function markdownParser(markdown: string) {
   return createMarkedInst().parse(markdown)
 }