add metadata to INode and shallow copy when flattening tree

dgreene1 · Jun 12, 2023 · 7d5b74a · 7d5b74a
1 parent b305a8d
commit 7d5b74a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/TreeView/types.ts b/src/TreeView/types.ts
@@ -28,6 +28,8 @@ export type EventCallback = <T, E>(
     parent: NodeId | null;
     /** Used to indicated whether a node is branch to be able load async data onExpand*/
     isBranch?: boolean;
+    /** User-defined metadata */
+    metadata?: any;
   }
 
   export interface INodeRendererProps {

diff --git a/src/TreeView/utils.ts b/src/TreeView/utils.ts
@@ -288,6 +288,7 @@ interface ITreeNode {
   id?: NodeId;
   name: string;
   children?: ITreeNode[];
+  metadata?: any;
 }
 
 export const flattenTree = function(tree: ITreeNode): INode[] {
@@ -300,6 +301,7 @@ export const flattenTree = function(tree: ITreeNode): INode[] {
       name: tree.name,
       children: [],
       parent,
+      ...tree.metadata && { metadata: {...tree.metadata }}
     };
 
     if (flattenedTree.find((x) => x.id === node.id)) {