Merge pull request #2916 from gwincr11/cg-sanitize

feat: add sanitization consistently
mermaid-js · Apr 12, 2022 · c13f6f7 · c13f6f7
2 parents 7d2e77a + 6052636
commit c13f6f7
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/src/diagrams/er/erDb.js b/src/diagrams/er/erDb.js
@@ -1,6 +1,7 @@
 import { log } from '../../logger';
 import mermaidAPI from '../../mermaidAPI';
 import * as configApi from '../../config';
+import common from '../common/common';
 
 let entities = {};
 let relationships = [];
@@ -69,15 +70,17 @@ const getRelationships = () => relationships;
 
 // Keep this - TODO: revisit...allow the diagram to have a title
 const setTitle = function (txt) {
-  title = txt;
+  let sanitizedText = common.sanitizeText(txt, configApi.getConfig());
+  title = sanitizedText;
 };
 
 const getTitle = function () {
   return title;
 };
 
 const setAccDescription = function (txt) {
-  description = txt;
+  let sanitizedText = common.sanitizeText(txt, configApi.getConfig());
+  description = sanitizedText;
 };
 
 const getAccDescription = function () {

diff --git a/src/diagrams/pie/pieDb.js b/src/diagrams/pie/pieDb.js
@@ -2,7 +2,6 @@ import { log } from '../../logger';
 import mermaidAPI from '../../mermaidAPI';
 import * as configApi from '../../config';
 import common from '../common/common';
-const sanitizeText = (txt) => common.sanitizeText(txt, configApi.getConfig());
 
 let sections = {};
 let title = '';

diff --git a/src/diagrams/user-journey/journeyDb.js b/src/diagrams/user-journey/journeyDb.js
@@ -1,5 +1,6 @@
 import mermaidAPI from '../../mermaidAPI';
 import * as configApi from '../../config';
+import common from '../common/common';
 
 let title = '';
 let currentSection = '';
@@ -21,7 +22,9 @@ export const clear = function () {
 };
 
 export const setTitle = function (txt) {
-  title = txt;
+  let sanitizedText = common.sanitizeText(txt, configApi.getConfig());
+  title = sanitizedText;
+
 };
 
 export const getTitle = function () {