feat: add updateCheckedInFlatData

Author: phphe

lib/HeTree.tsx

@@ -851,7 +851,7 @@ const flatDataDefaultOptions = {
   parentIdKey: 'parent_id',
 }
 export type WalkFlatDataYield<T> = [T, {
-  parent: T | null, parents: T[], index: number, treeIndex: number, id: Id, pid: Id, skipChildren: VoidFunction, exitWalk: VoidFunction
+  parent: T | null, parents: T[], index: number, treeIndex: number, id: Id, pid: Id | null, skipChildren: VoidFunction, exitWalk: VoidFunction
 }]
 export function* walkFlatDataGenerator<T extends Record<Id, any>>(flatData: T[], options0?: Partial<typeof flatDataDefaultOptions>): Generator<WalkFlatDataYield<T>> {
   const options = { ...flatDataDefaultOptions, ...options0 }
@@ -868,7 +868,7 @@ export function* walkFlatDataGenerator<T extends Record<Id, any>>(flatData: T[],
   let treeIndex = 0
   for (const node of flatData) {
     const id: Id = node[ID];
-    const pid: Id = node[PID];
+    const pid: Id = node[PID] ?? null;
     nodes[id] = node;
     const parent = nodes[pid] || null;
     childIdsById[id] = [];
@@ -1065,7 +1065,61 @@ export function openParentsInTreeData<T extends Record<Id, any>>(
   }
   return Array.from(openIdSet).sort()
 }
-
+// 'checked' utils methods =============
+export function updateCheckedInFlatData<T extends Record<Id, any>>(
+  flatData: T[],
+  checkedIds: Id[],
+  idOrIds: Id | Id[],
+  checked: boolean,
+  options?: Partial<typeof flatDataDefaultOptions>
+) {
+  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, id, pid }] of walkFlatDataGenerator(flatData, options)) {
+    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/test/flatData.test.ts

@@ -5,6 +5,7 @@ import {
   addToFlatData,
   removeByIdInFlatData,
   openParentsInFlatData,
+  updateCheckedInFlatData,
 } from "../../lib/HeTree";
 
 test("walkFlatDataGenerator: node, treeIndex", () => {
@@ -225,7 +226,7 @@ test("removeByIdInFlatData", () => {
   expect(cur.length).toBe(4);
   //
   cur = [...data];
-  console.log("if remove id 5", removeByIdInFlatData(cur, 5));
+  removeByIdInFlatData(cur, 5)
   expect(cur.length).toBe(7);
   //
   cur = createData("key");
@@ -245,6 +246,35 @@ test("openParentsInFlatData", () => {
   newOpenids = openParentsInFlatData(cur, [], 11);
   expect(newOpenids.toString()).toBe("");
 });
+test("updateCheckedInFlatData", () => {
+  let data = createData();
+  let cur = [...data];
+  let ids = [3];
+  let [newIds, semi,all] = updateCheckedInFlatData(cur,ids,[], true);
+  expect(newIds.toString()).toBe("3");
+  // 
+  [newIds, semi] = updateCheckedInFlatData(cur,ids,[3], true);
+  expect(newIds.toString()).toBe("3,6,7");
+  expect(semi.toString()).toBe("1");
+  // 
+  [newIds, semi] = updateCheckedInFlatData(cur,[],[8,10], true);
+  expect(newIds.toString()).toBe("10,2,4,5,8");
+  expect(semi.toString()).toBe("1");
+  // uncheck
+  [newIds, semi, all] = updateCheckedInFlatData(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] = updateCheckedInFlatData(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] = updateCheckedInFlatData(cur,[1,2,5,10,4,8,3,7,6],[6,7], false);
+  expect(semi.toString()).toBe('1');
+  
+});
 
 function createData(id = "id", parent_id = "parent_id") {
   // size 9