feat(chat): add Mermaid diagram zoom controls (#2100)

* feat(chat): add mermaid diagram zoom controls * fix(chat): handle malformed mermaid data urls * fix(chat): preserve mermaid load error stack --------- Co-authored-by: Bohdan Triapitsyn <artmore@protonmail.com>
2026-07-08 20:43:23 +03:00
parent dfed121bf1
commit 3d32ac7989
18 changed files with 1069 additions and 115 deletions
@@ -0,0 +1,37 @@
+import type { I18nKey, I18nParams } from '@/lib/i18n';
+
+export class MermaidLoadFailure extends Error {
+    key: I18nKey;
+    params?: I18nParams;
+
+    constructor(key: I18nKey, params?: I18nParams) {
+        super(key);
+        this.name = 'MermaidLoadFailure';
+        this.key = key;
+        this.params = params;
+    }
+}
+
+const mermaidLoadFailure = (key: I18nKey, params?: I18nParams): MermaidLoadFailure => new MermaidLoadFailure(key, params);
+
+export const isMermaidLoadFailure = (value: unknown): value is MermaidLoadFailure => value instanceof MermaidLoadFailure;
+
+export const nextMermaidLoadRequestId = (current: number): number => current + 1;
+
+export const isCurrentMermaidLoadRequest = (current: number, requestId: number): boolean => current === requestId;
+
+const decodeMermaidDataUrl = (value: string): string => {
+    const commaIndex = value.indexOf(',');
+    if (commaIndex < 0) {
+        throw mermaidLoadFailure('chat.toolOutputDialog.mermaid.dataUrlMalformed');
+    }
+
+    const metadata = value.slice(0, commaIndex).toLowerCase();
+    const payload = value.slice(commaIndex + 1);
+    if (metadata.includes(';base64')) {
+        return atob(payload);
+    }
+    return decodeURIComponent(payload);
+};
+
+export const getMermaidDataUrlSourcePromise = (value: string): Promise<string> => Promise.resolve().then(() => decodeMermaidDataUrl(value));