diff --git a/src/dagre-wrapper/nodes.js b/src/dagre-wrapper/nodes.js index 0857a7a2a..652861afa 100644 --- a/src/dagre-wrapper/nodes.js +++ b/src/dagre-wrapper/nodes.js @@ -672,7 +672,10 @@ const class_box = (parent, node) => { } const classAttributes = []; node.classData.members.forEach((str) => { - const parsedText = parseMember(str).displayText; + let parsedText = parseMember(str).displayText; + if (getConfig().flowchart.htmlLabels) { + parsedText = parsedText.replace(//g, '>'); + } const lbl = labelContainer .node() .appendChild(createLabel(parsedText, node.labelStyle, true, true)); @@ -695,10 +698,21 @@ const class_box = (parent, node) => { const classMethods = []; node.classData.methods.forEach((str) => { - const parsedText = parseMember(str).displayText; + const parsedInfo = parseMember(str); + let displayText =parsedInfo.displayText; + if (getConfig().flowchart.htmlLabels) { + displayText = displayText.replace(//g, '>'); + } const lbl = labelContainer .node() - .appendChild(createLabel(parsedText, node.labelStyle, true, true)); + .appendChild( + createLabel( + displayText, + parsedInfo.cssStyle ? parsedInfo.cssStyle : node.labelStyle, + true, + true + ) + ); let bbox = lbl.getBBox(); if (evaluate(getConfig().flowchart.htmlLabels)) { const div = lbl.children[0];