feat: provide HierarchyPointNode for renderCustomNodeElement (bkrem#355)

Co-authored-by: Jordan Rosenberg <[email protected]>

Author: bkrem

src/Node/index.tsx

@@ -118,12 +118,17 @@ export default class Node extends React.Component<NodeProps, NodeState> {
 
   // TODO: needs tests
   renderNodeElement = () => {
-    const { data, renderCustomNodeElement } = this.props;
+    const { data, hierarchyPointNode, renderCustomNodeElement } = this.props;
     if (typeof renderCustomNodeElement === 'function') {
-      return renderCustomNodeElement({ nodeDatum: data, toggleNode: this.handleNodeToggle });
+      return renderCustomNodeElement({
+        hierarchyPointNode: hierarchyPointNode,
+        nodeDatum: data,
+        toggleNode: this.handleNodeToggle,
+      });
     }
 
     return DefaultNodeElement({
+      hierarchyPointNode: hierarchyPointNode,
       nodeDatum: data,
       toggleNode: this.handleNodeToggle,
       onNodeClick: this.handleOnClick,

src/types/common.ts

@@ -42,6 +42,11 @@ export interface CustomNodeElementProps {
    * The full datum of the node that is being rendered.
    */
   nodeDatum: TreeNodeDatum;
+  /**
+   * The D3 `HierarchyPointNode` representation of the node, which wraps `nodeDatum`
+   * with additional properties.
+   */
+  hierarchyPointNode: HierarchyPointNode<TreeNodeDatum>;
   /**
    * Toggles the expanded/collapsed state of the node.
    *