added custom css class on edge labels

It should fix dagrejs#195
kaelhem · Mar 27, 2020 · b84438c · b84438c
1 parent 58c0378
commit b84438c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/create-edge-labels.js b/lib/create-edge-labels.js
@@ -23,7 +23,12 @@ function createEdgeLabels(selection, g) {
     var root = d3.select(this);
     root.select(".label").remove();
     var edge = g.edge(e);
-    var label = addLabel(root, g.edge(e), 0, 0).classed("label", true);
+    var className = "label";
+    if (edge.className) {
+      className += " " + edge.className;
+    }
+    var label = addLabel(root, g.edge(e), 0, 0);
+    label.attr("class", className);
     var bbox = label.node().getBBox();
 
     if (edge.labelId) { label.attr("id", edge.labelId); }