How to have a reference to a tree node with current item

Here I have a tree select that I have implement using antd 3.x and now I'm trying to migrate rc-tree. So I rendered my tree using a list with the type of TreeNodeType[] type as follow

type.ts

export interface TreeItem {
  compositeKey: string;
  title?: string;
  totalLeafs?: number;
  level: number;
  totalChildren?: number;
}

export interface TreeNodeType extends TreeItem {
  title: string;
  parent?: TreeNodeType;
  children?: TreeNodeType[];
  allChildrenLoaded?: boolean;
  hasChildren?: boolean;
}

TreeView.ts

const renderTreeNodes = useCallback(
    (nodes: TreeNodeType[], level: number = 1): React.ReactNode =>
      nodes
        .filter(item => !hiddenKeys.includes(item.compositeKey))
        .map(item => {
          return (
            <TreeNode
              title={treeNodeRenderFn ? treeNodeRenderFn(item) : item.title}
              key={item.compositeKey}
              isLeaf={
                level === treeDepth ||
                (item.allChildrenLoaded && isEmpty(item.children)) || !item.hasChildren
              }
              // here we can't have itemRef with rc-tree but we can in antd3.x
              itemRef={item}
            >
              {item.children && renderTreeNodes(item.children, level + 1)}
            </TreeNode>
          );
        }),
    [treeDepth, hiddenKeys, treeNodeRenderFn, loading]
  );

But thing is we can keep a reference to each node like itemRef. But rc-tree not support to that. But rendering as following I need to have a reference to each and every tree node(like itemRef in the above code) as I have to use it as following.

const handleExpand = (_: string[], info: AntTreeNodeExpandedEvent) => {
    dispatch({
      type: ActionType.changeExpandedNodes,
      payload: {
        expanded: info.expanded,
        keys: [info.node.props.itemRef.compositeKey],
      },
    });
    onExpand && onExpand(info.node.props.itemRef, info.expanded);
  };

rc-tree component doesn't support itemRef like reference like in antd3.x

So, I tried many ways to overcome this problem. pls help me to find solution.

Start with type.ts and TreeView.ts, especially the TreeNode rendering and handleExpand callback. Read the rc-tree TreeNode API and compare it with the antd 3.x itemRef behavior shown in the issue. Done means each expanded node's current TreeNodeType data can be accessed by the expansion handler without unsupported props.