resolves #1641 Mermaid background on span elements were missing

yuzutech · Sep 23, 2023 · c411e24 · c411e24
1 parent e229e29
commit c411e24
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 3 deletions.
diff --git a/diagrams.net/src/worker.js b/diagrams.net/src/worker.js
@@ -64,7 +64,16 @@ export class Worker {
       // const pageCount = parseInt(await page.mainFrame().$eval('#LoadingComplete', div => div.getAttribute('pageCount')))
 
       if (task.isPng) {
-        await page.setContent(evalResult.svg)
+        await page.setContent(`<!DOCTYPE html>  
+<html>
+<head>  
+<meta name="viewport" content="initial-scale=1.0, user-scalable=no" />  
+<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />  
+</head>  
+<body> 
+${evalResult.svg}
+</body>
+</html>`)
         const container = await page.$('svg')
         return await container.screenshot({
           type: 'png',

diff --git a/mermaid/src/worker.js b/mermaid/src/worker.js
@@ -191,11 +191,20 @@ async function toPNG (page, svg) {
     'puppeteer'
   )
   try {
-    await page.setContent(svg)
+    await page.setContent(`<!DOCTYPE html>  
+<html>
+<head>  
+<meta name="viewport" content="initial-scale=1.0, user-scalable=no" />  
+<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />  
+</head>  
+<body> 
+${svg}
+</body>
+</html>`)
     const container = await page.$('#container')
     const result = await container.screenshot({
       type: 'png',
-      omitBackground: true
+      omitBackground: true,
     })
     successfulSpan(span)
     return result