feat: add updateCheckedInTreeData

Author: phphe

lib/HeTree.tsx

@@ -1120,6 +1120,64 @@ export function updateCheckedInFlatData<T extends Record<Id, any>>(
   const allChecked = [...newCheckedIds, ...semiCheckedIds];
   return [newCheckedIds.sort(), semiCheckedIds.sort(), allChecked.sort()]
 }
+export function updateCheckedInTreeData<T extends Record<Id, any>>(
+  flatData: T[],
+  checkedIds: Id[],
+  idOrIds: Id | Id[],
+  checked: boolean,
+  options0?: Partial<typeof treeDataDefaultOptions>
+) {
+  const options = { ...treeDataDefaultOptions, ...options0 }
+  const { idKey: ID, childrenKey: CHILDREN } = options
+  const checkedIdSet = new Set(checkedIds)
+  const idsToUpdate = Array.isArray(idOrIds) ? idOrIds : [idOrIds];
+  const all = new Map<Id, Checked>()
+  const changedPids = new Set<Id>(idsToUpdate)
+  const pidById: Record<Id, Id | null> = {}
+  const childIdsById = new Map<Id | null, Id[]>()
+  childIdsById.set(null, [])
+  for (const [node, { parents, parent }] of walkTreeDataGenerator(flatData, CHILDREN)) {
+    const id = node[ID];
+    const pid = parent?.[ID] ?? null
+    pidById[id] = pid
+    childIdsById.get(pid)!.push(id)
+    childIdsById.set(id, [])
+    all.set(id, checkedIdSet.has(id))
+    if (changedPids.has(id) || (pid && changedPids.has(pid))) {
+      // update self and children
+      all.set(id, checked)
+      changedPids.add(id)
+    }
+  }
+  // update parents
+  for (const id of idsToUpdate) {
+    let cur = pidById[id];
+    while (cur != null) {
+      let allChecked = true
+      let hasChecked = false
+      for (const childId of childIdsById.get(cur)!) {
+        if (all.get(childId) === false) {
+          allChecked = false
+        } else {
+          hasChecked = true
+        }
+      }
+      all.set(cur, allChecked ? true : (hasChecked ? null : false))
+      cur = pidById[cur]
+    }
+  }
+  const newCheckedIds: Id[] = [];
+  const semiCheckedIds: Id[] = [];
+  all.forEach((v, k) => {
+    if (v === true) {
+      newCheckedIds.push(k)
+    } else if (v === null) {
+      semiCheckedIds.push(k)
+    }
+  })
+  const allChecked = [...newCheckedIds, ...semiCheckedIds];
+  return [newCheckedIds.sort(), semiCheckedIds.sort(), allChecked.sort()]
+}
 // private methods
 function calculateDistance(x1: number, y1: number, x2: number, y2: number) {
   return Math.sqrt(Math.pow((x2 - x1), 2) + Math.pow((y2 - y1), 2));

src/App.tsx

@@ -2,6 +2,7 @@ import example_data from "./examples/example_data.json";
 import { sortFlatData, useHeTree } from "../lib/index";
 import { useState } from "react";
 import { useImmer } from "use-immer";
+
 function App() {
   // const [flatData, setflatData] = useState(() => {
   //   const list: (typeof example_data)[] = [];

src/test/treeData.test.ts

@@ -4,6 +4,7 @@ import {
   findTreeData,
   filterTreeData,
   openParentsInTreeData,
+  updateCheckedInTreeData,
 } from "../../lib/HeTree";
 
 test("walkTreeDataGenerator", () => {
@@ -155,6 +156,34 @@ test("openParentsInTreeData", () => {
   newOpenids = openParentsInTreeData(cur, [], "===================");
   expect(newOpenids.toString()).toBe("");
 });
+test("updateCheckedInTreeData", () => {
+  let data = createData2();
+  let cur = [...data];
+  let ids = [3];
+  let [newIds, semi,all] = updateCheckedInTreeData(cur,ids,[], true);
+  expect(newIds.toString()).toBe("3");
+  // 
+  [newIds, semi] = updateCheckedInTreeData(cur,ids,[3], true);
+  expect(newIds.toString()).toBe("3,6,7");
+  expect(semi.toString()).toBe("1");
+  // 
+  [newIds, semi] = updateCheckedInTreeData(cur,[],[8,10], true);
+  expect(newIds.toString()).toBe("10,2,4,5,8");
+  expect(semi.toString()).toBe("1");
+  // uncheck
+  [newIds, semi, all] = updateCheckedInTreeData(cur,[10,2,4,8,5],[8], false);
+  expect(newIds.toString()).toBe("10,5");
+  expect(semi.toString()).toBe("1,2");
+  expect(all.toString()).toBe("1,10,2,5");
+  // 
+  [newIds, semi, all] = updateCheckedInTreeData(cur,[1,2,5,10,4,8,3,7,6],[6,7], !false);
+  expect(newIds.length).toBe(9);
+  expect(semi.length).toBe(0);
+  expect(all.length).toBe(9);
+  // 
+  [newIds, semi, all] = updateCheckedInTreeData(cur,[1,2,5,10,4,8,3,7,6],[6,7], false);
+  expect(semi.toString()).toBe('1');
+});
 
 function createData() {
   // return example tree data
@@ -284,3 +313,71 @@ function createData() {
     },
   ];
 }
+
+function createData2() {
+  // same to flatData.test
+  // size 9
+  /* structure
+  1
+    2
+      5
+        10
+      4
+        8
+    3
+      7
+      6
+  */
+  return [
+    {
+        "id": 1,
+        "name": "Root Category",
+        "children": [
+            {
+                "id": 2,
+                "name": "Technology",
+                "children": [
+                    {
+                        "id": 5,
+                        "name": "Hardware",
+                        "children": [
+                            {
+                                "id": 10,
+                                "name": "Computer Components",
+                                "children": []
+                            }
+                        ]
+                    },
+                    {
+                        "id": 4,
+                        "name": "Programming",
+                        "children": [
+                            {
+                                "id": 8,
+                                "name": "Python",
+                                "children": []
+                            }
+                        ]
+                    }
+                ]
+            },
+            {
+                "id": 3,
+                "name": "Science",
+                "children": [
+                    {
+                        "id": 7,
+                        "name": "Biology",
+                        "children": []
+                    },
+                    {
+                        "id": 6,
+                        "name": "Physics",
+                        "children": []
+                    }
+                ]
+            }
+        ]
+    }
+]
+}