feat(TreeData): index

fjc0k · Jan 20, 2021 · 74cebdc · 74cebdc
1 parent 0467079
commit 74cebdc
Show file tree

Hide file tree

Showing 3 changed files with 581 additions and 0 deletions.
diff --git a/src/utils/TreeData.test.ts b/src/utils/TreeData.test.ts
@@ -94,6 +94,16 @@ describe('TreeData', () => {
     expect(new TreeData(data).pickNodeProps(['id']).export()).toMatchSnapshot()
   })
 
+  test('index', () => {
+    expect(
+      new TreeData(data)
+        .setNodeProps({
+          index: _ => _.index,
+        })
+        .export(),
+    ).toMatchSnapshot()
+  })
+
   test('filter', () => {
     expect(
       new TreeData(data).filter(_ => _.node.id.endsWith('.2')).export(),

diff --git a/src/utils/TreeData.ts b/src/utils/TreeData.ts
@@ -54,6 +54,11 @@ export interface TreeDataTraverseFnPayload<TNode extends TreeDataNode> {
    */
   node: TNode
 
+  /**
+   * 当前节点索引。
+   */
+  index: number
+
   /**
    * 当前深度。从 `0` 开始。
    */
@@ -161,6 +166,7 @@ export class TreeData<TNode extends TreeDataNode> {
 
       fn({
         node: node,
+        index: index,
         parentNode: parentNode,
         depth: depth,
         path: path,