diff --git a/src/Node/index.tsx b/src/Node/index.tsx index c2c05db5..6fb1f56e 100644 --- a/src/Node/index.tsx +++ b/src/Node/index.tsx @@ -120,10 +120,15 @@ export default class Node extends React.Component { renderNodeElement = () => { const { data, renderCustomNodeElement } = this.props; if (typeof renderCustomNodeElement === 'function') { - return renderCustomNodeElement({ nodeDatum: data, toggleNode: this.handleNodeToggle }); + return renderCustomNodeElement({ + hierarchyPointNode: this.props.hierarchyPointNode, + nodeDatum: data, + toggleNode: this.handleNodeToggle, + }); } return DefaultNodeElement({ + hierarchyPointNode: this.props.hierarchyPointNode, nodeDatum: data, toggleNode: this.handleNodeToggle, onNodeClick: this.handleOnClick, diff --git a/src/types/common.ts b/src/types/common.ts index 441a2048..c129431a 100644 --- a/src/types/common.ts +++ b/src/types/common.ts @@ -42,6 +42,7 @@ export interface CustomNodeElementProps { * The full datum of the node that is being rendered. */ nodeDatum: TreeNodeDatum; + hierarchyPointNode: HierarchyPointNode; /** * Toggles the expanded/collapsed state of the node. *