Add background color style to output svg's

mermaid-js · Jan 14, 2022 · 46e3b70 · 46e3b70
1 parent e371751
commit 46e3b70
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/src-test/test.js b/src-test/test.js
@@ -41,7 +41,9 @@ async function compileDiagram(file, format) {
       "-o",
       out + "/" + result,
       "-c",
-      workflows + "/config.json"
+      workflows + "/config.json",
+      "-b",
+      "lightgray"
     ], { timeout: 5000 });
 
     const stdout = child.stdout.toString('utf8').trim()

diff --git a/src/index.js b/src/index.js
@@ -13,6 +13,10 @@ const error = message => {
   process.exit(1)
 }
 
+const warn = message => {
+  console.log(chalk.yellow(`\n${message}\n`))
+}
+
 const checkConfigFile = file => {
   if (!fs.existsSync(file)) {
     error(`Configuration file "${file}" doesn't exist`)
@@ -167,7 +171,15 @@ const parseMMD = async (browser, definition, output) => {
   }
 
   if (output.endsWith('svg')) {
-    const svg = await page.$eval('#container', container => container.innerHTML)
+    const svg = await page.$eval('#container', (container, backgroundColor) => {
+      const svg = container.getElementsByTagName?.('svg')?.[0]
+      if (svg.style) {
+        svg.style.backgroundColor = backgroundColor
+      } else {
+        warn("svg not found. Not applying background color.")
+      }
+      return container.innerHTML
+    }, backgroundColor)
     const svg_xml = convertToValidXML(svg)
     fs.writeFileSync(output, svg_xml)
   } else if (output.endsWith('png')) {