From 54ef5b0beed4ede0eaeab96a09ecdb95c1d45915 Mon Sep 17 00:00:00 2001
From: Ashish Jain <ashishjain0512@gmail.com>
Date: Thu, 19 Aug 2021 21:15:21 +0200
Subject: [PATCH] Fix for classDiagram-v2 support for cardinality where HTML
 label is true

---
 src/dagre-wrapper/edges.js | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/dagre-wrapper/edges.js b/src/dagre-wrapper/edges.js
index 6970133fe..9f3b00c72 100644
--- a/src/dagre-wrapper/edges.js
+++ b/src/dagre-wrapper/edges.js
@@ -56,7 +56,7 @@ export const insertEdgeLabel = (elem, edge) => {
       terminalLabels[edge.id] = {};
     }
     terminalLabels[edge.id].startLeft = startEdgeLabelLeft;
-    setTerminalWidth(fo, bbox);
+    setTerminalWidth(fo, edge.startLabelLeft);
   }
   if (edge.startLabelRight) {
     // Create the actual text element
@@ -72,7 +72,7 @@ export const insertEdgeLabel = (elem, edge) => {
       terminalLabels[edge.id] = {};
     }
     terminalLabels[edge.id].startRight = startEdgeLabelRight;
-    setTerminalWidth(fo, bbox);
+    setTerminalWidth(fo, edge.startLabelRight);
   }
   if (edge.endLabelLeft) {
     // Create the actual text element
@@ -89,7 +89,7 @@ export const insertEdgeLabel = (elem, edge) => {
       terminalLabels[edge.id] = {};
     }
     terminalLabels[edge.id].endLeft = endEdgeLabelLeft;
-    setTerminalWidth(fo, bbox);
+    setTerminalWidth(fo, edge.endLabelLeft);
   }
   if (edge.endLabelRight) {
     // Create the actual text element
@@ -106,14 +106,14 @@ export const insertEdgeLabel = (elem, edge) => {
       terminalLabels[edge.id] = {};
     }
     terminalLabels[edge.id].endRight = endEdgeLabelRight;
-    setTerminalWidth(fo, bbox);
+    setTerminalWidth(fo, edge.endLabelRight);
   }
 };
 
-function setTerminalWidth(fo, box) {
+function setTerminalWidth(fo, value) {
   if (getConfig().flowchart.htmlLabels && fo) {
-    fo.style.width = box.width;
-    fo.style.height = box.height;
+    fo.style.width = value.length * 9 + 'px';
+    fo.style.height = '12px';
   }
 }