Add DeepMap.getCountOfLeafNodesStartingWith method

Author: radubrehar

examples/src/pages/tests/table/treegrid/expand-collapse.page.tsx

@@ -107,7 +107,7 @@ export default function DataTestPage() {
         treeExpandState={treeExpandState}
         onTreeExpandStateChange={(expandState) => {
           console.log('expandState', expandState);
-          debugger;
+          // console.log(expandState.collapsedPaths)
           setTreeExpandState(expandState);
         }}
       >

examples/src/pages/tests/table/utils/DeepMap.spec.ts

@@ -645,6 +645,11 @@ export default test.describe('DeepMap', () => {
       ['3', '31'],
       ['1', '10'],
     ]);
+    expect(
+      map.getCountOfLeafNodesStartingWith([], {
+        excludeSelf: true,
+      }),
+    ).toEqual(2);
 
     map = new DeepMap<string | number, boolean>();
     map.set(['3', '31'], true);
@@ -660,6 +665,11 @@ export default test.describe('DeepMap', () => {
       ['3', '31'],
       ['1', '10'],
     ]);
+    expect(
+      map.getCountOfLeafNodesStartingWith([], {
+        excludeSelf: true,
+      }),
+    ).toEqual(2);
 
     map = new DeepMap<string | number, boolean>();
     map.set(['1', '10'], true);
@@ -673,11 +683,15 @@ export default test.describe('DeepMap', () => {
       }),
     ).toEqual([]);
     expect(
-      map.getKeysForLeafNodesStartingWith(['1', '10'], {
+      map.getCountOfLeafNodesStartingWith(['1', '10'], {
+        excludeSelf: true,
+      }),
+    ).toEqual(0);
+    expect(
+      map.getCountOfLeafNodesStartingWith(['1', '10'], {
         excludeSelf: false,
-        respectOrder: true,
       }),
-    ).toEqual([['1', '10']]);
+    ).toEqual(1);
   });
 
   test('getKeysForLeafNodesStartingWith should work correctly - with depthLimit', () => {
@@ -694,13 +708,25 @@ export default test.describe('DeepMap', () => {
         depthLimit: 1,
       }),
     ).toEqual([['4']]);
+    expect(
+      map.getCountOfLeafNodesStartingWith([], {
+        excludeSelf: true,
+        depthLimit: 1,
+      }),
+    ).toEqual(1);
     expect(
       map.getKeysForLeafNodesStartingWith([], {
         excludeSelf: true,
         depthLimit: 2,
         respectOrder: true,
       }),
     ).toEqual([['3', '31'], ['1', '10'], ['4']]);
+    expect(
+      map.getCountOfLeafNodesStartingWith([], {
+        excludeSelf: true,
+        depthLimit: 2,
+      }),
+    ).toEqual(3);
 
     map = new DeepMap<string | number, boolean>();
     expect(
@@ -712,9 +738,16 @@ export default test.describe('DeepMap', () => {
     expect(
       map.getKeysForLeafNodesStartingWith([], { excludeSelf: false }),
     ).toEqual([[]]);
+    expect(
+      map.getCountOfLeafNodesStartingWith([], { excludeSelf: false }),
+    ).toEqual(1);
+
     expect(
       map.getKeysForLeafNodesStartingWith([], { excludeSelf: true }),
     ).toEqual([]);
+    expect(
+      map.getCountOfLeafNodesStartingWith([], { excludeSelf: true }),
+    ).toEqual(0);
   });
 
   test('getLeafNodesStartingWith should work correctly', () => {
@@ -728,6 +761,7 @@ export default test.describe('DeepMap', () => {
 
     const result = map.getLeafNodesStartingWith([], (pair) => pair.keys);
     expect(result).toEqual([['1', '10'], ['1', '20'], ['3', '31'], ['4']]);
+    expect(map.getCountOfLeafNodesStartingWith([])).toEqual(4);
   });
 
   test('getLeafNodesStartingWith should work correctly - second scenario', () => {
@@ -757,6 +791,7 @@ export default test.describe('DeepMap', () => {
       ['3', '31', '311', '3110'],
       ['3', '31', '311', '3111'],
     ]);
+    expect(map.getCountOfLeafNodesStartingWith([])).toEqual(8);
   });
 
   test('visit depth first, with index', () => {

source/src/components/DataSource/TreeSelectionState.ts

@@ -229,10 +229,12 @@ export class TreeSelectionState<T = any> {
   setConfig(getConfig: GetTreeSelectionStateConfig<T>) {
     const config = typeof getConfig === 'function' ? getConfig() : getConfig;
 
+    const treePaths = Array.isArray(config.treePaths)
+      ? new DeepMap(config.treePaths.map((path) => [path, true as const]))
+      : config.treePaths;
+
     this.config = {
-      treePaths: Array.isArray(config.treePaths)
-        ? new DeepMap(config.treePaths.map((path) => [path, true]))
-        : config.treePaths,
+      treePaths,
       strictCheckPaths: config.strictCheckPaths ?? true,
     };
     this.xcache();

source/src/utils/DeepMap/index.ts

@@ -125,6 +125,24 @@ export class DeepMap<KeyType, ValueType> {
     return result;
   }
 
+  getCountOfLeafNodesStartingWith(
+    keys: KeyType[],
+    options?: StartingWithMethodOptions,
+  ) {
+    let count = 0;
+    this.getStartingWith(
+      keys,
+      (_keys, _value, pair) => {
+        if (!pair.map && pair.hasOwnProperty('value')) {
+          count++;
+        }
+      },
+      options,
+    );
+
+    return count;
+  }
+
   getKeysForLeafNodesStartingWith(
     keys: KeyType[],
     options?: StartingWithMethodOptionsWithOrder,