marcel/mermaid
1
0
Fork 0
mirror of https://github.com/mermaid-js/mermaid.git synced 2025-10-10 17:49:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update selectSvgElement.ts

Browse Source
This commit is contained in:
Reda Al Sulais
2023-08-05 20:45:23 +03:00
committed by GitHub
parent ef29388d68
commit 1802f5d0c0
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
packages/mermaid/src/rendering-util/selectSvgElement.ts
View File

@@ -11,14 +11,12 @@ import type { HTML, SVG } from '../diagram-api/types.js';
export const selectSvgElement = (id: string): SVG => { export const selectSvgElement = (id: string): SVG => {
const { securityLevel } = getConfig(); const { securityLevel } = getConfig();
// handle root and document for when rendering in sandbox mode // handle root and document for when rendering in sandbox mode
let doc: Document = document; let root: HTML = select('body');
if (securityLevel === 'sandbox') { if (securityLevel === 'sandbox') {
const sandboxElement: HTML = select(`#i${id}`); const sandboxElement: HTML = select(`#i${id}`);
doc = sandboxElement.node()?.contentDocument ?? doc; const doc: Document = sandboxElement.node()?.contentDocument ?? document;
root = select(doc.body as HTMLIFrameElement);
} }
const root: HTML =
securityLevel === 'sandbox' ? select(doc.body as HTMLIFrameElement) : select('body');
const svg: SVG = root.select(`#${id}`); const svg: SVG = root.select(`#${id}`);
return svg; return svg;
}; };