@1771technologies/lytenyte-pro 1.0.0-beta.11 → 1.0.0-beta.12

package/dist/row-data-source-client/use-client-tree-data-source.js

@@ -2,528 +2,539 @@ import {} from "../+types.js";
 import { useRef } from "react";
 import { atom, createStore } from "@1771technologies/atom";
 import { traverse } from "./tree/traverse.js";
-import { dateComparator, makeGridAtom, numberComparator, stringComparator, } from "@1771technologies/lytenyte-shared";
+import {
+  dateComparator,
+  makeGridAtom,
+  numberComparator,
+  stringComparator,
+} from "@1771technologies/lytenyte-shared";
 import { equal } from "@1771technologies/lytenyte-js-utils";
 import { makeClientTree } from "./tree/client-tree.js";
 import { computeFilteredRows } from "./filter/compute-filtered-rows.js";
 import { builtIns } from "./built-ins/built-ins.js";
 export function makeClientTreeDataSource(p) {
-    const rdsStore = createStore();
-    const data = atom(p.data);
-    const topData = atom(p.topData ?? []);
-    const bottomData = atom(p.bottomData ?? []);
-    const dataToSrc$ = atom((g) => {
-        return new Map(g(data).map((c, i) => [c, i]));
-    });
-    const cache = new Map();
-    const centerNodes = atom((g) => {
-        const nodes = [];
-        const d = g(data);
-        for (let i = 0; i < d.length; i++) {
-            if (!cache.has(i)) {
-                cache.set(i, {
-                    id: "",
-                    kind: "leaf",
-                    data: d[i],
-                });
-            }
-            const node = cache.get(i);
-            node.data = d[i];
-            nodes.push(node);
-        }
-        return nodes;
-    });
-    const topNodes = atom((g) => {
-        return g(topData).map((c, i) => ({ data: c, id: `top-${i}`, kind: "leaf" }));
-    });
-    const botNodes = atom((g) => {
-        return g(bottomData).map((c, i) => ({ data: c, id: `bottom-${i}`, kind: "leaf" }));
-    });
-    const pinnedIdMap = atom((g) => {
-        const combined = new Map([...g(topNodes), ...g(botNodes)].map((c) => [c.id, c]));
-        return combined;
-    });
-    const models = atom({
-        sort: [],
-        filter: {},
-        filterIn: {},
-        quickSearch: null,
-        agg: {},
-        groupExpansions: {},
-        columnPivotGroupExpansions: {},
-    });
-    const sortModel = atom((g) => g(models).sort);
-    const filterModel = atom((g) => g(models).filter);
-    const filterInModel = atom((g) => g(models).filterIn);
-    const groupExpansions = atom((g) => g(models).groupExpansions);
-    const aggModel = atom((g) => g(models).agg);
-    const quickSearch = atom((g) => g(models).quickSearch);
-    const grid$ = atom(null);
-    const snapshot = atom(0);
-    const tree = atom((g) => {
-        g(snapshot);
-        const grid = g(grid$);
-        const rows = g(centerNodes);
-        const filtered = computeFilteredRows(rows, grid, g(filterModel), g(filterInModel), g(quickSearch), grid?.state.quickSearchSensitivity.get() ?? "case-sensitive", false);
-        const rowAggModel = Object.entries(g(aggModel)).map(([name, agg]) => {
-            if (typeof agg.fn === "function") {
-                const fn = agg.fn;
-                return {
-                    name,
-                    fn: (rows) => fn(rows.map((c) => c.data), grid),
-                };
-            }
-            const key = agg.fn;
-            const fn = (data) => {
-                const fieldData = data.map((r) => grid?.api.columnField(name, { kind: "leaf", data: r.data }));
-                return builtIns[key](fieldData);
-            };
-            return { name, fn };
+  const rdsStore = createStore();
+  const data = atom(p.data);
+  const topData = atom(p.topData ?? []);
+  const bottomData = atom(p.bottomData ?? []);
+  const dataToSrc$ = atom((g) => {
+    return new Map(g(data).map((c, i) => [c, i]));
+  });
+  const cache = new Map();
+  const centerNodes = atom((g) => {
+    const nodes = [];
+    const d = g(data);
+    for (let i = 0; i < d.length; i++) {
+      if (!cache.has(i)) {
+        cache.set(i, {
+          id: "",
+          kind: "leaf",
+          data: d[i],
         });
-        return makeClientTree({
-            rowData: filtered,
-            rowAggModel: grid ? rowAggModel : [],
-            rowBranchModel: p.getPathFromData,
-            rowIdGroup: p.rowIdBranch,
-            rowIdLeaf: p.rowIdLeaf,
-        });
-    });
-    const sortComparator = atom((g) => {
-        const model = g(sortModel);
-        const grid = g(grid$);
-        if (!model.length || !grid)
-            return () => 0;
-        const comparator = (l, r) => {
-            let res = 0;
-            for (const sortSpec of model) {
-                const sort = sortSpec.sort;
-                const columnId = sortSpec.columnId;
-                const ld = l.kind === 2
-                    ? { kind: "branch", data: l.data, key: l.key }
-                    : { kind: "leaf", data: l.data.data };
-                const rd = r.kind === 2
-                    ? { kind: "branch", data: r.data, key: r.key }
-                    : { kind: "leaf", data: r.data.data };
-                if (sort.kind === "custom") {
-                    res = sort.comparator(ld, rd, sort.options ?? {});
-                }
-                else if (sort.kind === "number") {
-                    const left = grid.api.columnField(columnId, ld);
-                    const right = grid.api.columnField(columnId, rd);
-                    res = numberComparator(left, right, sort.options ?? {});
-                }
-                else if (sort.kind === "date") {
-                    const left = grid.api.columnField(columnId, ld);
-                    const right = grid.api.columnField(columnId, rd);
-                    res = dateComparator(left, right, sort.options ?? {});
-                }
-                else if (sort.kind === "string") {
-                    const left = grid.api.columnField(columnId, ld);
-                    const right = grid.api.columnField(columnId, rd);
-                    res = stringComparator(left, right, sort.options ?? {});
-                }
-                else {
-                    res = 0;
-                }
-                res = sortSpec.isDescending ? -res : res;
-                if (res !== 0)
-                    break;
-            }
-            return res;
+      }
+      const node = cache.get(i);
+      node.data = d[i];
+      nodes.push(node);
+    }
+    return nodes;
+  });
+  const topNodes = atom((g) => {
+    return g(topData).map((c, i) => ({ data: c, id: `top-${i}`, kind: "leaf" }));
+  });
+  const botNodes = atom((g) => {
+    return g(bottomData).map((c, i) => ({ data: c, id: `bottom-${i}`, kind: "leaf" }));
+  });
+  const pinnedIdMap = atom((g) => {
+    const combined = new Map([...g(topNodes), ...g(botNodes)].map((c) => [c.id, c]));
+    return combined;
+  });
+  const models = atom({
+    sort: [],
+    filter: {},
+    filterIn: {},
+    quickSearch: null,
+    agg: {},
+    groupExpansions: {},
+    columnPivotGroupExpansions: {},
+  });
+  const sortModel = atom((g) => g(models).sort);
+  const filterModel = atom((g) => g(models).filter);
+  const filterInModel = atom((g) => g(models).filterIn);
+  const groupExpansions = atom((g) => g(models).groupExpansions);
+  const aggModel = atom((g) => g(models).agg);
+  const quickSearch = atom((g) => g(models).quickSearch);
+  const grid$ = atom(null);
+  const snapshot = atom(0);
+  const tree = atom((g) => {
+    g(snapshot);
+    const grid = g(grid$);
+    const rows = g(centerNodes);
+    const filtered = computeFilteredRows(
+      rows,
+      grid,
+      g(filterModel),
+      g(filterInModel),
+      g(quickSearch),
+      grid?.state.quickSearchSensitivity.get() ?? "case-sensitive",
+      false,
+    );
+    const rowAggModel = Object.entries(g(aggModel)).map(([name, agg]) => {
+      if (typeof agg.fn === "function") {
+        const fn = agg.fn;
+        return {
+          name,
+          fn: (rows) =>
+            fn(
+              rows.map((c) => c.data),
+              grid,
+            ),
         };
-        return comparator;
+      }
+      const key = agg.fn;
+      const fn = (data) => {
+        const fieldData = data.map((r) =>
+          grid?.api.columnField(name, { kind: "leaf", data: r.data }),
+        );
+        return builtIns[key](fieldData);
+      };
+      return { name, fn };
     });
-    const idToNode = atom((g) => {
-        const map = new Map();
-        traverse(g(tree).root, (node) => {
-            map.set(node.id, node);
-        });
-        return map;
+    return makeClientTree({
+      rowData: filtered,
+      rowAggModel: grid ? rowAggModel : [],
+      rowBranchModel: p.getPathFromData,
+      rowIdGroup: p.rowIdBranch,
+      rowIdLeaf: p.rowIdLeaf,
     });
-    const initialized = atom(false);
-    const flat = atom((g) => {
-        if (!g(initialized))
-            return { flat: [], idMap: new Map(), idToIndexMap: new Map() };
-        const idMap = new Map();
-        const idToIndexMap = new Map();
-        const flattened = [];
-        const comparator = g(sortComparator);
-        const expansions = g(groupExpansions);
-        const defaultExpansion = g(grid$)?.state.rowGroupDefaultExpansion.get() ?? false;
-        let index = g(topNodes).length;
-        traverse(g(tree).root, (node) => {
-            if (node.kind === 1) {
-                node.data.id = node.id;
-                flattened.push(node.data);
-            }
-            else {
-                flattened.push({
-                    kind: "branch",
-                    data: node.data,
-                    id: node.id,
-                    key: node.key,
-                    depth: node.depth,
-                });
-            }
-            idMap.set(node.id, flattened.at(-1));
-            idToIndexMap.set(node.id, index);
-            index++;
-            if (node.kind === 2) {
-                const expanded = expansions[node.id] ??
-                    (typeof defaultExpansion === "number"
-                        ? node.depth <= defaultExpansion
-                        : defaultExpansion);
-                return expanded;
-            }
-        }, comparator);
-        return { flat: flattened, idMap, idToIndexMap };
+  });
+  const sortComparator = atom((g) => {
+    const model = g(sortModel);
+    const grid = g(grid$);
+    if (!model.length || !grid) return () => 0;
+    const comparator = (l, r) => {
+      let res = 0;
+      for (const sortSpec of model) {
+        const sort = sortSpec.sort;
+        const columnId = sortSpec.columnId;
+        const ld =
+          l.kind === 2
+            ? { kind: "branch", data: l.data, key: l.key }
+            : { kind: "leaf", data: l.data.data };
+        const rd =
+          r.kind === 2
+            ? { kind: "branch", data: r.data, key: r.key }
+            : { kind: "leaf", data: r.data.data };
+        if (sort.kind === "custom") {
+          res = sort.comparator(ld, rd, sort.options ?? {});
+        } else if (sort.kind === "number") {
+          const left = grid.api.columnField(columnId, ld);
+          const right = grid.api.columnField(columnId, rd);
+          res = numberComparator(left, right, sort.options ?? {});
+        } else if (sort.kind === "date") {
+          const left = grid.api.columnField(columnId, ld);
+          const right = grid.api.columnField(columnId, rd);
+          res = dateComparator(left, right, sort.options ?? {});
+        } else if (sort.kind === "string") {
+          const left = grid.api.columnField(columnId, ld);
+          const right = grid.api.columnField(columnId, rd);
+          res = stringComparator(left, right, sort.options ?? {});
+        } else {
+          res = 0;
+        }
+        res = sortSpec.isDescending ? -res : res;
+        if (res !== 0) break;
+      }
+      return res;
+    };
+    return comparator;
+  });
+  const idToNode = atom((g) => {
+    const map = new Map();
+    traverse(g(tree).root, (node) => {
+      map.set(node.id, node);
     });
-    const flatLength = atom((g) => g(flat).flat.length);
-    const cleanup = [];
-    const init = (grid) => {
-        while (cleanup.length)
-            cleanup.pop()?.();
-        rdsStore.set(grid$, grid);
-        const store = grid.state.rowDataStore;
-        // Monitor row count changes
+    return map;
+  });
+  const initialized = atom(false);
+  const flat = atom((g) => {
+    if (!g(initialized)) return { flat: [], idMap: new Map(), idToIndexMap: new Map() };
+    const idMap = new Map();
+    const idToIndexMap = new Map();
+    const flattened = [];
+    const comparator = g(sortComparator);
+    const expansions = g(groupExpansions);
+    const defaultExpansion = g(grid$)?.state.rowGroupDefaultExpansion.get() ?? false;
+    let index = g(topNodes).length;
+    traverse(
+      g(tree).root,
+      (node) => {
+        if (node.kind === 1) {
+          node.data.id = node.id;
+          flattened.push(node.data);
+        } else {
+          flattened.push({
+            kind: "branch",
+            data: node.data,
+            id: node.id,
+            key: node.key,
+            depth: node.depth,
+          });
+        }
+        idMap.set(node.id, flattened.at(-1));
+        idToIndexMap.set(node.id, index);
+        index++;
+        if (node.kind === 2) {
+          const expanded =
+            expansions[node.id] ??
+            (typeof defaultExpansion === "number"
+              ? node.depth <= defaultExpansion
+              : defaultExpansion);
+          return expanded;
+        }
+      },
+      comparator,
+    );
+    return { flat: flattened, idMap, idToIndexMap };
+  });
+  const flatLength = atom((g) => g(flat).flat.length);
+  const cleanup = [];
+  const init = (grid) => {
+    while (cleanup.length) cleanup.pop()?.();
+    rdsStore.set(grid$, grid);
+    const store = grid.state.rowDataStore;
+    // Monitor row count changes
+    const centerCount = rdsStore.get(flatLength);
+    store.rowCenterCount.set(centerCount);
+    cleanup.push(
+      rdsStore.sub(flatLength, () => {
         const centerCount = rdsStore.get(flatLength);
         store.rowCenterCount.set(centerCount);
-        cleanup.push(rdsStore.sub(flatLength, () => {
-            const centerCount = rdsStore.get(flatLength);
-            store.rowCenterCount.set(centerCount);
-            grid.state.rowDataStore.rowClearCache();
-        }));
+        grid.state.rowDataStore.rowClearCache();
+      }),
+    );
+    const top = rdsStore.get(topData).length;
+    store.rowTopCount.set(top);
+    cleanup.push(
+      rdsStore.sub(topData, () => {
         const top = rdsStore.get(topData).length;
         store.rowTopCount.set(top);
-        cleanup.push(rdsStore.sub(topData, () => {
-            const top = rdsStore.get(topData).length;
-            store.rowTopCount.set(top);
-            grid.state.rowDataStore.rowClearCache();
-        }));
-        const bottom = rdsStore.get(bottomData).length;
-        store.rowBottomCount.set(bottom);
-        cleanup.push(rdsStore.sub(bottomData, () => {
-            const bot = rdsStore.get(bottomData).length;
-            store.rowBottomCount.set(bot);
-            grid.state.rowDataStore.rowClearCache();
+        grid.state.rowDataStore.rowClearCache();
+      }),
+    );
+    const bottom = rdsStore.get(bottomData).length;
+    store.rowBottomCount.set(bottom);
+    cleanup.push(
+      rdsStore.sub(bottomData, () => {
+        const bot = rdsStore.get(bottomData).length;
+        store.rowBottomCount.set(bot);
+        grid.state.rowDataStore.rowClearCache();
+      }),
+    );
+    const sort = grid.state.sortModel.get();
+    const filter = grid.state.filterModel.get();
+    const filterIn = grid.state.filterInModel.get();
+    const groupExpansions = grid.state.rowGroupExpansions.get();
+    const agg = grid.state.aggModel.get();
+    const quickSearch = grid.state.quickSearch.get();
+    const columnPivotGroupExpansions = grid.state.columnPivotRowGroupExpansions.get();
+    rdsStore.set(models, {
+      agg,
+      filter,
+      filterIn,
+      quickSearch,
+      groupExpansions,
+      sort,
+      columnPivotGroupExpansions,
+    });
+    rdsStore.set(initialized, true);
+    // Sort model monitoring
+    cleanup.push(
+      grid.state.sortModel.watch(() => {
+        rdsStore.set(models, (prev) => ({ ...prev, sort: grid.state.sortModel.get() }));
+        grid.state.rowDataStore.rowClearCache();
+      }),
+    );
+    // Filter model monitoring
+    cleanup.push(
+      grid.state.filterModel.watch(() => {
+        rdsStore.set(models, (prev) => ({ ...prev, filter: grid.state.filterModel.get() }));
+        grid.state.rowDataStore.rowClearCache();
+      }),
+    );
+    cleanup.push(
+      grid.state.quickSearch.watch(() => {
+        rdsStore.set(models, (prev) => ({ ...prev, quickSearch: grid.state.quickSearch.get() }));
+        grid.state.rowDataStore.rowClearCache();
+      }),
+    );
+    cleanup.push(
+      grid.state.filterInModel.watch(() => {
+        rdsStore.set(models, (prev) => ({ ...prev, filterIn: grid.state.filterInModel.get() }));
+        grid.state.rowDataStore.rowClearCache();
+      }),
+    );
+    // Row group model monitoring
+    cleanup.push(
+      grid.state.rowGroupModel.watch(() => {
+        rdsStore.set(models, (prev) => ({ ...prev, group: grid.state.rowGroupModel.get() }));
+        grid.state.rowDataStore.rowClearCache();
+      }),
+    );
+    // Row group expansions monitoring
+    cleanup.push(
+      grid.state.rowGroupExpansions.watch(() => {
+        rdsStore.set(models, (prev) => ({
+          ...prev,
+          groupExpansions: grid.state.rowGroupExpansions.get(),
         }));
-        const sort = grid.state.sortModel.get();
-        const filter = grid.state.filterModel.get();
-        const filterIn = grid.state.filterInModel.get();
-        const groupExpansions = grid.state.rowGroupExpansions.get();
-        const agg = grid.state.aggModel.get();
-        const quickSearch = grid.state.quickSearch.get();
-        const columnPivotGroupExpansions = grid.state.columnPivotRowGroupExpansions.get();
-        rdsStore.set(models, {
-            agg,
-            filter,
-            filterIn,
-            quickSearch,
-            groupExpansions,
-            sort,
-            columnPivotGroupExpansions,
+        grid.state.rowDataStore.rowClearCache();
+      }),
+    );
+    // Agg model monitoring
+    cleanup.push(
+      grid.state.aggModel.watch(() => {
+        rdsStore.set(models, (prev) => ({ ...prev, agg: grid.state.aggModel.get() }));
+        grid.state.rowDataStore.rowClearCache();
+      }),
+    );
+  };
+  const rowById = (id) => {
+    const pinned = rdsStore.get(pinnedIdMap);
+    if (pinned.has(id)) return pinned.get(id);
+    const t = rdsStore.get(flat);
+    return t.idMap.get(id) ?? null;
+  };
+  const rowByIndex = (index) => {
+    const top = rdsStore.get(topNodes);
+    const bot = rdsStore.get(botNodes);
+    const center = rdsStore.get(flat).flat;
+    const topOffset = top.length;
+    const centerOffset = topOffset + center.length;
+    const botOffset = centerOffset + bot.length;
+    if (index < topOffset) return top[index];
+    if (index < centerOffset) return center[index - topOffset];
+    if (index < botOffset) return bot[index - centerOffset];
+    return null;
+  };
+  const rowUpdate = (updates) => {
+    const grid = rdsStore.get(grid$);
+    const d = rdsStore.get(data);
+    const idMap = rdsStore.get(idToNode);
+    const dataToSrc = rdsStore.get(dataToSrc$);
+    for (const [key, next] of updates.entries()) {
+      const row = typeof key === "string" ? rowById(key) : rowByIndex(key);
+      const treeNode = typeof key === "string" ? idMap.get(key) : null;
+      if ((!row && !treeNode) || !grid) {
+        console.error(`Failed to find the row with identifier ${key} which is being updated.`);
+        continue;
+      }
+      if (row?.kind === "branch") {
+        row.data = next;
+      } else {
+        const data = row?.kind === "leaf" ? row.data : treeNode?.data.data;
+        const source = dataToSrc.get(data);
+        if (source == null) {
+          console.error(`Failed to find the row with identifier ${key} which is being updated.`);
+          continue;
+        }
+        d[source] = next;
+      }
+    }
+    rdsStore.set(data, [...d]);
+    rdsStore.set(snapshot, (prev) => prev + 1);
+    grid.state.rowDataStore.rowClearCache();
+  };
+  const rowToIndex = (rowId) => {
+    const f = rdsStore.get(flat);
+    const top = rdsStore.get(topNodes);
+    const bot = rdsStore.get(botNodes);
+    const topCount = top.length;
+    const center = f.flat.length;
+    const rowIndex = f.idToIndexMap.get(rowId);
+    if (rowIndex != null) return rowIndex;
+    if (rowIndex == null) {
+      const foundTop = top.findIndex((row) => row.id === rowId);
+      if (foundTop !== -1) return foundTop;
+      const foundBot = bot.findIndex((row) => row.id === rowId);
+      if (foundBot !== -1) return foundBot + topCount + center;
+    }
+    return null;
+  };
+  const rowAllChildIds = (rowId) => {
+    const t = rdsStore.get(tree);
+    const ids = [];
+    const node = t.idToNode.get(rowId);
+    if (node?.kind === 2) {
+      traverse(node, (n) => {
+        ids.push(n.id);
+      });
+    }
+    return ids;
+  };
+  return [
+    {
+      init,
+      rowById,
+      rowByIndex,
+      rowAllChildIds,
+      rowUpdate,
+      inFilterItems: (c) => {
+        const grid = rdsStore.get(grid$);
+        if (!grid) return [];
+        const data = rdsStore.get(centerNodes);
+        const values = new Set(
+          data.map((row) => {
+            const field = grid.api.columnField(c, row);
+            return field;
+          }),
+        );
+        if (p.transformInFilterItem) {
+          return p.transformInFilterItem({ column: c, values: [...values] });
+        }
+        return [...values].map((x) => ({ id: `${x}`, label: `${x}`, value: x }));
+      },
+      rowAdd: (newRows, place = "end") => {
+        rdsStore.set(data, (prev) => {
+          if (!newRows.length) return prev;
+          let next;
+          if (place === "beginning") next = [...newRows, ...prev];
+          else if (place === "end") next = [...prev, ...newRows];
+          else {
+            next = [...prev];
+            next.splice(place, 0, ...newRows);
+          }
+          return next;
         });
-        rdsStore.set(initialized, true);
-        // Sort model monitoring
-        cleanup.push(grid.state.sortModel.watch(() => {
-            rdsStore.set(models, (prev) => ({ ...prev, sort: grid.state.sortModel.get() }));
-            grid.state.rowDataStore.rowClearCache();
-        }));
-        // Filter model monitoring
-        cleanup.push(grid.state.filterModel.watch(() => {
-            rdsStore.set(models, (prev) => ({ ...prev, filter: grid.state.filterModel.get() }));
-            grid.state.rowDataStore.rowClearCache();
-        }));
-        cleanup.push(grid.state.quickSearch.watch(() => {
-            rdsStore.set(models, (prev) => ({ ...prev, quickSearch: grid.state.quickSearch.get() }));
-            grid.state.rowDataStore.rowClearCache();
-        }));
-        cleanup.push(grid.state.filterInModel.watch(() => {
-            rdsStore.set(models, (prev) => ({ ...prev, filterIn: grid.state.filterInModel.get() }));
-            grid.state.rowDataStore.rowClearCache();
-        }));
-        // Row group model monitoring
-        cleanup.push(grid.state.rowGroupModel.watch(() => {
-            rdsStore.set(models, (prev) => ({ ...prev, group: grid.state.rowGroupModel.get() }));
-            grid.state.rowDataStore.rowClearCache();
-        }));
-        // Row group expansions monitoring
-        cleanup.push(grid.state.rowGroupExpansions.watch(() => {
-            rdsStore.set(models, (prev) => ({
-                ...prev,
-                groupExpansions: grid.state.rowGroupExpansions.get(),
-            }));
-            grid.state.rowDataStore.rowClearCache();
-        }));
-        // Agg model monitoring
-        cleanup.push(grid.state.aggModel.watch(() => {
-            rdsStore.set(models, (prev) => ({ ...prev, agg: grid.state.aggModel.get() }));
-            grid.state.rowDataStore.rowClearCache();
-        }));
-    };
-    const rowById = (id) => {
-        const pinned = rdsStore.get(pinnedIdMap);
-        if (pinned.has(id))
-            return pinned.get(id);
-        const t = rdsStore.get(flat);
-        return t.idMap.get(id) ?? null;
-    };
-    const rowByIndex = (index) => {
-        const top = rdsStore.get(topNodes);
-        const bot = rdsStore.get(botNodes);
-        const center = rdsStore.get(flat).flat;
-        const topOffset = top.length;
-        const centerOffset = topOffset + center.length;
-        const botOffset = centerOffset + bot.length;
-        if (index < topOffset)
-            return top[index];
-        if (index < centerOffset)
-            return center[index - topOffset];
-        if (index < botOffset)
-            return bot[index - centerOffset];
-        return null;
-    };
-    const rowUpdate = (updates) => {
         const grid = rdsStore.get(grid$);
-        const d = rdsStore.get(data);
-        const idMap = rdsStore.get(idToNode);
-        const dataToSrc = rdsStore.get(dataToSrc$);
-        for (const [key, next] of updates.entries()) {
-            const row = typeof key === "string" ? rowById(key) : rowByIndex(key);
-            const treeNode = typeof key === "string" ? idMap.get(key) : null;
-            if ((!row && !treeNode) || !grid) {
-                console.error(`Failed to find the row with identifier ${key} which is being updated.`);
-                continue;
-            }
-            if (row?.kind === "branch") {
-                row.data = next;
-            }
-            else {
-                const data = row?.kind === "leaf" ? row.data : treeNode?.data.data;
-                const source = dataToSrc.get(data);
-                if (source == null) {
-                    console.error(`Failed to find the row with identifier ${key} which is being updated.`);
-                    continue;
-                }
-                d[source] = next;
+        grid?.state.rowDataStore.rowClearCache();
+      },
+      rowDelete: (rows) => {
+        const rowData = new Set(
+          rows
+            .map((c) => {
+              if (typeof c === "number") return rowByIndex(c)?.data;
+              else return rowById(c)?.data;
+            })
+            .filter((c) => !!c),
+        );
+        rdsStore.set(data, (prev) => {
+          if (!rowData.size) return prev;
+          return prev.filter((d) => !rowData.has(d));
+        });
+        const grid = rdsStore.get(grid$);
+        grid?.state.rowDataStore.rowClearCache();
+      },
+      rowSetBotData: (data) => {
+        rdsStore.set(bottomData, data);
+        const grid = rdsStore.get(grid$);
+        grid?.state.rowDataStore.rowClearCache();
+      },
+      rowSetCenterData: (d) => {
+        rdsStore.set(data, d);
+        const grid = rdsStore.get(grid$);
+        grid?.state.rowDataStore.rowClearCache();
+      },
+      rowSetTopData: (data) => {
+        rdsStore.set(topData, data);
+        const grid = rdsStore.get(grid$);
+        grid?.state.rowDataStore.rowClearCache();
+      },
+      rowExpand: (expansions) => {
+        const grid = rdsStore.get(grid$);
+        if (!grid) return;
+        const mode = grid.state.columnPivotMode.get();
+        if (mode)
+          grid.state.columnPivotRowGroupExpansions.set((prev) => ({ ...prev, ...expansions }));
+        else grid.state.rowGroupExpansions.set((prev) => ({ ...prev, ...expansions }));
+      },
+      rowToIndex,
+      rowSelect: (params) => {
+        const grid = rdsStore.get(grid$);
+        if (!grid) return;
+        if (params.mode === "none") return;
+        if (params.mode === "single") {
+          if (params.deselect) {
+            grid.state.rowSelectedIds.set(new Set());
+          } else {
+            grid.state.rowSelectedIds.set(new Set([params.startId]));
+          }
+          return;
+        }
+        const ids = new Set();
+        if (params.startId === params.endId) {
+          ids.add(params.startId);
+          if (params.selectChildren) {
+            rowAllChildIds(params.startId).forEach((c) => ids.add(c));
+          }
+        } else {
+          const first = rowToIndex(params.startId);
+          const last = rowToIndex(params.endId);
+          if (first == null || last == null) return;
+          const start = Math.min(first, last);
+          const end = Math.max(first, last);
+          for (let i = start; i <= end; i++) {
+            const row = rowByIndex(i);
+            if (!row) continue;
+            if (params.selectChildren) {
+              rowAllChildIds(row.id).forEach((c) => ids.add(c));
             }
+            if (row?.id) ids.add(row.id);
+          }
         }
-        rdsStore.set(data, [...d]);
-        rdsStore.set(snapshot, (prev) => prev + 1);
-        grid.state.rowDataStore.rowClearCache();
-    };
-    const rowToIndex = (rowId) => {
-        const f = rdsStore.get(flat);
-        const top = rdsStore.get(topNodes);
-        const bot = rdsStore.get(botNodes);
-        const topCount = top.length;
-        const center = f.flat.length;
-        const rowIndex = f.idToIndexMap.get(rowId);
-        if (rowIndex != null)
-            return rowIndex;
-        if (rowIndex == null) {
-            const foundTop = top.findIndex((row) => row.id === rowId);
-            if (foundTop !== -1)
-                return foundTop;
-            const foundBot = bot.findIndex((row) => row.id === rowId);
-            if (foundBot !== -1)
-                return foundBot + topCount + center;
+        if (params.deselect) {
+          const current = grid.state.rowSelectedIds.get();
+          const next = current.difference(ids);
+          grid.state.rowSelectedIds.set(next);
+        } else {
+          const current = grid.state.rowSelectedIds.get();
+          const next = current.union(ids);
+          grid.state.rowSelectedIds.set(next);
+        }
+      },
+      rowSelectAll: (params) => {
+        const grid = rdsStore.get(grid$);
+        if (!grid) return;
+        if (params.deselect) {
+          grid.state.rowSelectedIds.set(new Set());
+          return;
         }
-        return null;
-    };
-    const rowAllChildIds = (rowId) => {
         const t = rdsStore.get(tree);
-        const ids = [];
-        const node = t.idToNode.get(rowId);
-        if (node?.kind === 2) {
-            traverse(node, (n) => {
-                ids.push(n.id);
-            });
+        grid.state.rowSelectedIds.set(new Set(t.idsAll));
+      },
+      rowAreAllSelected: (rowId) => {
+        const g = rdsStore.get(grid$);
+        if (!g) return false;
+        const selected = g.state.rowSelectedIds.get();
+        if (rowId) {
+          const row = rowById(rowId);
+          if (!row) return false;
+          const childIds = new Set(rowAllChildIds(rowId));
+          return childIds.isSubsetOf(selected);
         }
-        return ids;
-    };
-    return [
-        {
-            init,
-            rowById,
-            rowByIndex,
-            rowAllChildIds,
-            rowUpdate,
-            inFilterItems: (c) => {
-                const grid = rdsStore.get(grid$);
-                if (!grid)
-                    return [];
-                const data = rdsStore.get(centerNodes);
-                const values = new Set(data.map((row) => {
-                    const field = grid.api.columnField(c, row);
-                    return field;
-                }));
-                if (p.transformInFilterItem) {
-                    return p.transformInFilterItem({ column: c, values: [...values] });
-                }
-                return [...values].map((x) => ({ id: `${x}`, label: `${x}`, value: x }));
-            },
-            rowAdd: (newRows, place = "end") => {
-                rdsStore.set(data, (prev) => {
-                    if (!newRows.length)
-                        return prev;
-                    let next;
-                    if (place === "beginning")
-                        next = [...newRows, ...prev];
-                    else if (place === "end")
-                        next = [...prev, ...newRows];
-                    else {
-                        next = [...prev];
-                        next.splice(place, 0, ...newRows);
-                    }
-                    return next;
-                });
-                const grid = rdsStore.get(grid$);
-                grid?.state.rowDataStore.rowClearCache();
-            },
-            rowDelete: (rows) => {
-                const rowData = new Set(rows
-                    .map((c) => {
-                    if (typeof c === "number")
-                        return rowByIndex(c)?.data;
-                    else
-                        return rowById(c)?.data;
-                })
-                    .filter((c) => !!c));
-                rdsStore.set(data, (prev) => {
-                    if (!rowData.size)
-                        return prev;
-                    return prev.filter((d) => !rowData.has(d));
-                });
-                const grid = rdsStore.get(grid$);
-                grid?.state.rowDataStore.rowClearCache();
-            },
-            rowSetBotData: (data) => {
-                rdsStore.set(bottomData, data);
-                const grid = rdsStore.get(grid$);
-                grid?.state.rowDataStore.rowClearCache();
-            },
-            rowSetCenterData: (d) => {
-                rdsStore.set(data, d);
-                const grid = rdsStore.get(grid$);
-                grid?.state.rowDataStore.rowClearCache();
-            },
-            rowSetTopData: (data) => {
-                rdsStore.set(topData, data);
-                const grid = rdsStore.get(grid$);
-                grid?.state.rowDataStore.rowClearCache();
-            },
-            rowExpand: (expansions) => {
-                const grid = rdsStore.get(grid$);
-                if (!grid)
-                    return;
-                const mode = grid.state.columnPivotMode.get();
-                if (mode)
-                    grid.state.columnPivotRowGroupExpansions.set((prev) => ({ ...prev, ...expansions }));
-                else
-                    grid.state.rowGroupExpansions.set((prev) => ({ ...prev, ...expansions }));
-            },
-            rowToIndex,
-            rowSelect: (params) => {
-                const grid = rdsStore.get(grid$);
-                if (!grid)
-                    return;
-                if (params.mode === "none")
-                    return;
-                if (params.mode === "single") {
-                    if (params.deselect) {
-                        grid.state.rowSelectedIds.set(new Set());
-                    }
-                    else {
-                        grid.state.rowSelectedIds.set(new Set([params.startId]));
-                    }
-                    return;
-                }
-                const ids = new Set();
-                if (params.startId === params.endId) {
-                    ids.add(params.startId);
-                    if (params.selectChildren) {
-                        rowAllChildIds(params.startId).forEach((c) => ids.add(c));
-                    }
-                }
-                else {
-                    const first = rowToIndex(params.startId);
-                    const last = rowToIndex(params.endId);
-                    if (first == null || last == null)
-                        return;
-                    const start = Math.min(first, last);
-                    const end = Math.max(first, last);
-                    for (let i = start; i <= end; i++) {
-                        const row = rowByIndex(i);
-                        if (!row)
-                            continue;
-                        if (params.selectChildren) {
-                            rowAllChildIds(row.id).forEach((c) => ids.add(c));
-                        }
-                        if (row?.id)
-                            ids.add(row.id);
-                    }
-                }
-                if (params.deselect) {
-                    const current = grid.state.rowSelectedIds.get();
-                    const next = current.difference(ids);
-                    grid.state.rowSelectedIds.set(next);
-                }
-                else {
-                    const current = grid.state.rowSelectedIds.get();
-                    const next = current.union(ids);
-                    grid.state.rowSelectedIds.set(next);
-                }
-            },
-            rowSelectAll: (params) => {
-                const grid = rdsStore.get(grid$);
-                if (!grid)
-                    return;
-                if (params.deselect) {
-                    grid.state.rowSelectedIds.set(new Set());
-                    return;
-                }
-                const t = rdsStore.get(tree);
-                grid.state.rowSelectedIds.set(new Set(t.idsAll));
-            },
-            rowAreAllSelected: (rowId) => {
-                const g = rdsStore.get(grid$);
-                if (!g)
-                    return false;
-                const selected = g.state.rowSelectedIds.get();
-                if (rowId) {
-                    const row = rowById(rowId);
-                    if (!row)
-                        return false;
-                    const childIds = new Set(rowAllChildIds(rowId));
-                    return childIds.isSubsetOf(selected);
-                }
-                const f = rdsStore.get(tree);
-                return f.idsAll.isSubsetOf(selected);
-            },
-        },
-        {
-            top: makeGridAtom(topData, rdsStore),
-            center: makeGridAtom(data, rdsStore),
-            bottom: makeGridAtom(bottomData, rdsStore),
-        },
-    ];
+        const f = rdsStore.get(tree);
+        return f.idsAll.isSubsetOf(selected);
+      },
+    },
+    {
+      top: makeGridAtom(topData, rdsStore),
+      center: makeGridAtom(data, rdsStore),
+      bottom: makeGridAtom(bottomData, rdsStore),
+    },
+  ];
 }
 export function useClientTreeDataSource(p) {
-    const ds = useRef(null);
-    const dataAtomRef = useRef(null);
-    if (!ds.current)
-        [ds.current, dataAtomRef.current] = makeClientTreeDataSource(p);
-    const da = dataAtomRef.current;
-    if (p.reflectData) {
-        // Need to queue the microtask since it we cannot update state during render.
-        if (p.data !== da.center.get())
-            queueMicrotask(() => ds.current.rowSetCenterData(p.data));
-        if (!equal(p.topData ?? [], da.top.get())) {
-            queueMicrotask(() => ds.current.rowSetTopData(p.topData ?? []));
-        }
-        if (!equal(p.bottomData ?? [], da.bottom.get()))
-            queueMicrotask(() => ds.current.rowSetBotData(p.bottomData ?? []));
+  const ds = useRef(null);
+  const dataAtomRef = useRef(null);
+  if (!ds.current) [ds.current, dataAtomRef.current] = makeClientTreeDataSource(p);
+  const da = dataAtomRef.current;
+  if (p.reflectData) {
+    // Need to queue the microtask since it we cannot update state during render.
+    if (p.data !== da.center.get()) queueMicrotask(() => ds.current.rowSetCenterData(p.data));
+    if (!equal(p.topData ?? [], da.top.get())) {
+      queueMicrotask(() => ds.current.rowSetTopData(p.topData ?? []));
     }
-    return ds.current;
+    if (!equal(p.bottomData ?? [], da.bottom.get()))
+      queueMicrotask(() => ds.current.rowSetBotData(p.bottomData ?? []));
+  }
+  return ds.current;
 }