feat: Add css class override for node text

ErikGartner · Dec 27, 2015 · 21b8677 · 21b8677
1 parent c995eb5
commit 21b8677
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -27,6 +27,7 @@ The data object should have the following structure:
 [{
   name: "Father",
   class: "overriding-css-class",
+  textClass: "overriding-css-class",
   marriage: {
     spouse: {
       name: "Mother",

diff --git a/src/builder.js b/src/builder.js
@@ -78,11 +78,7 @@ const treeBuilder = {
     // Create the node rectangles.
     nodes.append('rect')
       .attr('class', function(d) {
-        if (d.class) {
-          return d.class;
-        } else {
-          return opts.styles.node;
-        }
+        return d.class ? d.class : opts.styles.nodes;
       })
       .attr('width', nodeSize[0] / 2)
       .attr('height', nodeSize[1] / 3)
@@ -104,7 +100,9 @@ const treeBuilder = {
       .text(function(d) {
         return d.name;
       })
-      .attr('class', opts.styles.text)
+      .attr('class', function(d) {
+        return d.textClass ? d.textClass : opts.styles.text;
+      })
       .attr('x', tx)
       .attr('y', ty);
   },