@react-fabric/data 1.0.0-beta.6 → 1.0.0-beta.60

package/src/tree/useTree.ts

@@ -22,10 +22,20 @@
  */
 
 import { useDebounce } from "@react-fabric/core";
-import { EMPTY_ARRAY } from "@react-fabric/utilities";
+import { EMPTY_ARRAY, isEmpty } from "@react-fabric/utilities";
 import memoizeOne from "memoize-one";
-import { useCallback, useEffect, useReducer, useRef } from "react";
-import { type InternalNode, type TreeNodeType } from "./types";
+import {
+  useCallback,
+  useEffect,
+  useReducer,
+  useRef,
+  useEffectEvent,
+} from "react";
+import {
+  type InternalNode,
+  type TreeNodeType,
+  type TreePanelProps,
+} from "./types";
 import {
   filterTree,
   flattenTree,
@@ -52,6 +62,10 @@ type TreeActions =
    * update item check state
    */
   | { type: "checked"; id: string[] }
+  /**
+   * toggle item select, used as action handler
+   */
+  | { type: "toggleSelect"; id: string; shiftKey?: boolean }
   /**
    * toggle item check, used as action handler
    */
@@ -71,13 +85,18 @@ type TreeActions =
   /**
    * collapse all nodes
    */
-  | { type: "collapseAll" };
+  | { type: "collapseAll" }
+  /**
+   * loaded dynamic nodes
+   */
+  | { type: "loaded" };
 
 interface TreeState {
   items: InternalNode[];
   itemMap: Map<string, InternalNode>;
   tree: InternalNode[];
-  selected?: string;
+  query?: string;
+  selected?: AnyObject;
   checked?: string[];
 }
 
@@ -87,47 +106,139 @@ export const useTree = <T extends KeyValue>({
   items = EMPTY_ARRAY,
   checked = EMPTY_ARRAY,
   selected,
+  multiple,
+  sorter,
   matcher,
+  onLoad,
+  onQuery,
   onSelect,
   onChecked,
-}: {
-  items?: Array<TreeNodeType<T>>;
-  selected?: string;
-  checked?: string[];
-  onSelect?: (id: string) => void;
-  onChecked?: (leafs: string[], nodes: string[]) => void;
-  matcher?: (data: T, query: string) => boolean;
+  defaultExpanded,
+  onExpandToggle,
+}: Partial<TreePanelProps<T>> & {
+  selected?: AnyObject;
+  multiple?: boolean;
 }) => {
-  const componentEvents = useRef({ onSelect, onChecked });
-
+  const eventSelect = useEffectEvent(onSelect ?? (() => undefined));
+  const eventChecked = useEffectEvent(onChecked ?? (() => undefined));
+  const intialExpand = useRef([...(defaultExpanded ?? EMPTY_ARRAY)]);
   const itemList = createItemList(items);
 
+  const fireExpandToggle = useCallback(
+    (itemMap: InternalNode[]) => {
+      const expandedList = itemMap
+        .filter((node) => node.open)
+        .map((node) => node.id);
+      setTimeout(() => onExpandToggle?.(expandedList), 100);
+    },
+    [onExpandToggle],
+  );
+
+  const loadChildren = useCallback(
+    (node: InternalNode) => {
+      if (node.open && !node.leaf && !node.children?.length) {
+        node.loading = true;
+        // delay onload callback to avoid state issues
+        setTimeout(() => {
+          void Promise.resolve(onLoad?.(node.id))
+            .then((resp = EMPTY_ARRAY) => {
+              node.children = refactorTree(
+                resp,
+                { sorter, defaultExpanded },
+                { level: node.level + 1, parent: node.id },
+              );
+            })
+            .finally(() => {
+              node.loading = false;
+              dispatch({ type: "loaded" });
+              intialExpand.current?.length > 0 &&
+                dispatch({
+                  type: "expand",
+                  id: intialExpand.current.shift() as unknown as string,
+                });
+            });
+        }, 100);
+      } else {
+        dispatch({ type: "loaded" });
+        intialExpand.current?.length > 0 &&
+          dispatch({
+            type: "expand",
+            id: intialExpand.current.shift() as unknown as string,
+          });
+      }
+    },
+    [onLoad],
+  );
+
+  const fireSelected = useDebounce((nodeId: any, data: T) => {
+    eventSelect(nodeId, data);
+  }, []);
+
+  const fireChecked = useDebounce(
+    (leafs: string[], nodes: string[], partials: string[]) => {
+      eventChecked(leafs, nodes, partials);
+    },
+    [],
+  );
+
+  const fireQuery = useDebounce(
+    (query: string) => {
+      onQuery?.(query);
+    },
+    [onQuery],
+  );
+
   const [state, dispatch] = useReducer(
     (state: TreeState, action: TreeActions) => {
       if (action.type === "load") {
-        state.items = refactorTree(action.items);
+        state.items = refactorTree(action.items, { sorter, defaultExpanded });
         state.itemMap = makeTreeMap(state.items);
         state.tree = flattenTree(state.items);
+        state.selected = undefined;
+        state.checked = undefined;
+        state.query = "";
         return { ...state };
       }
       if (action.type === "toggleExpand") {
         const node = state.itemMap.get(action.id);
-        node && (node.open = !node.open);
+        if (node && !node.leaf) {
+          node && (node.open = !node.open);
+          loadChildren(node);
+          return { ...state };
+        }
+      }
+      if (action.type === "loaded") {
+        state.itemMap = makeTreeMap(state.items);
+        fireExpandToggle(Array.from(state.itemMap.values()));
         state.tree = flattenTree(state.items);
+        state.selected = selected;
+        updateSelection({
+          nodes: state.itemMap,
+          multiple,
+          selected,
+        });
+        state.checked?.forEach((id) => updateChecked(state.itemMap, id, 1));
         return { ...state };
       }
       if (action.type === "expand") {
-        // TODO: check if children available, if not call onLoad to fetch node children
         const node = state.itemMap.get(action.id);
-        node && (node.open = true);
-        state.tree = flattenTree(state.items);
-        return { ...state };
+        if (node && !node.leaf) {
+          node.open = true;
+          let parentId = node.parent;
+          while (parentId) {
+            const node = state.itemMap.get(parentId);
+            node && (node.open = true);
+            parentId = node?.parent;
+          }
+          loadChildren(node);
+          return { ...state };
+        }
       }
       if (action.type === "expandAll") {
-        // TODO: prevent opening dynamic nodes that dont have children
         state.itemMap.forEach((node) => {
-          !node.leaf && (node.open = true);
+          !node.leaf && node.children?.length && (node.open = true);
         });
+        fireExpandToggle(Array.from(state.itemMap.values()));
         state.tree = flattenTree(state.items);
         return { ...state };
       }
@@ -135,34 +246,77 @@ export const useTree = <T extends KeyValue>({
         state.itemMap.forEach((node) => {
           !node.leaf && (node.open = false);
         });
+        fireExpandToggle([]);
         state.tree = flattenTree(state.items);
         return { ...state };
       }
       if (action.type === "filter") {
-        filterTree(state.itemMap, action.query, matcher);
+        if (onQuery) {
+          // TODO: onQuery should provide new tree listing
+          // save current tree state as preQueryState
+          // on clearing filter restore preQueryState
+          if (!isEmpty(action.query)) {
+            // state.prevItems = state.items;
+            fireQuery(action.query);
+          } else {
+            // state.items = state.prevItems;
+            // build tree map
+          }
+        } else {
+          filterTree(state.itemMap, action.query, matcher);
+          state.tree = flattenTree(state.items);
+        }
+        return { ...state, query: action.query };
+      }
+      if (action.type === "toggleSelect") {
+        if (multiple) {
+          const isSelected = state.selected?.includes(action.id);
+          if (action.shiftKey && isSelected)
+            state.selected =
+              state.selected?.filter((id: string) => id !== action.id) ?? [];
+          else if (action.shiftKey && !isSelected)
+            state.selected = [...(state.selected ?? []), action.id];
+          else state.selected = [action.id];
+        } else {
+          state.selected = action.id;
+        }
+        updateSelection({
+          nodes: state.itemMap,
+          selected: state.selected,
+          multiple,
+        });
         state.tree = flattenTree(state.items);
+        fireSelected(state.selected, state.itemMap.get(action.id)?.data);
         return { ...state };
       }
       if (action.type === "select") {
-        if (state.selected) {
-          updateSelection(state.itemMap, state.selected);
-        }
-        updateSelection(state.itemMap, action.id, true);
+        state.selected = action.id;
+        updateSelection({
+          nodes: state.itemMap,
+          multiple,
+          selected: action.id,
+        });
         state.tree = flattenTree(state.items);
-        componentEvents.current.onSelect?.(action.id);
-        return { ...state, selected: action.id };
+        return { ...state };
       }
       if (action.type === "toggleCheck") {
         updateChecked(state.itemMap, action.id);
-        const checkList: string[][] = [[], []];
+        const checkList: string[][] = [[], [], []];
         Array.from(state.itemMap.values())
-          .filter((node) => node.checked === 1)
-          .forEach((node) => checkList[node.leaf ? 0 : 1].push(node.id));
-        componentEvents.current.onChecked?.(checkList[0], checkList[1]);
-        return { ...state };
+          .filter((node) => node.checked !== 0)
+          .forEach((node) =>
+            checkList[node.leaf ? 0 : node.checked].push(node.id),
+          );
+        fireChecked(
+          checkList[0], // leafs
+          checkList[1], // nodes
+          checkList[2], // partials
+        );
+        return { ...state, checked: checkList[0] };
       }
       if (action.type === "checked") {
-        action.id.forEach((id) => updateChecked(state.itemMap, id));
+        state.itemMap.forEach((item) => (item.checked = 0));
+        action.id.forEach((id) => updateChecked(state.itemMap, id, 1));
         return { ...state, checked: action.id };
       }
       return state;
@@ -172,15 +326,20 @@ export const useTree = <T extends KeyValue>({
       items: [],
       tree: [],
       checked: [],
+      query: "",
     },
   );
 
-  useEffect(() => {
-    componentEvents.current = { onSelect, onChecked };
-  }, [onSelect, onChecked]);
-
   useEffect(() => {
     dispatch({ type: "load", items: itemList });
+    itemList.length &&
+      intialExpand.current?.length > 0 &&
+      dispatch({
+        type: "expand",
+        id: intialExpand.current.shift() as unknown as string,
+      });
+    selected && dispatch({ type: "select", id: selected });
+    checked && dispatch({ type: "checked", id: checked });
   }, [itemList]);
 
   useEffect(() => {
@@ -199,14 +358,24 @@ export const useTree = <T extends KeyValue>({
     dispatch({ type: "toggleCheck", id });
   }, []);
 
-  const select = useCallback((id: string) => {
-    dispatch({ type: "select", id });
+  const select = useCallback((id: string, shiftKey = false) => {
+    dispatch({ type: "toggleSelect", id, shiftKey });
   }, []);
 
   const expand = useCallback((id: string) => {
     dispatch({ type: "expand", id });
   }, []);
 
+  const expandAndLoad = useCallback((expandList: string[]) => {
+    if (expandList.length > 0) {
+      intialExpand.current = [...expandList];
+      dispatch({
+        type: "expand",
+        id: intialExpand.current.shift() as unknown as string,
+      });
+    }
+  }, []);
+
   const expandAll = useCallback(() => {
     dispatch({ type: "expandAll" });
   }, []);
@@ -215,16 +384,23 @@ export const useTree = <T extends KeyValue>({
     dispatch({ type: "collapseAll" });
   }, []);
 
-  const onFilter = useDebounce((query: string) => {
-    dispatch({ type: "filter", query });
-  }, []);
+  const onFilter = useDebounce(
+    (query: string) => {
+      dispatch({ type: "filter", query });
+    },
+    [],
+    300,
+  );
 
   return {
     tree: state.tree,
+    query: state.query,
+    selected: state.selected,
     toggleExpand,
     expandAll,
     collapseAll,
     toggleCheck,
+    expandAndLoad,
     select,
     expand,
     onFilter,

package/src/tree/utils.ts

@@ -21,10 +21,15 @@
  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  */
 
-import { compareValues, isString, matchString } from "@react-fabric/utilities";
+import {
+  compareValues,
+  isArray,
+  isString,
+  matchString,
+} from "@react-fabric/utilities";
 import { type InternalNode, type TreeNodeType } from "./types";
 
-const sorter = (a: TreeNodeType, b: TreeNodeType) => {
+const defaultSorter = (a: TreeNodeType, b: TreeNodeType) => {
   if (!!a.leaf !== !!b.leaf) return a.leaf ? 1 : -1;
   return compareValues()(a.label, b.label);
 };
@@ -48,13 +53,17 @@ const getIntermediate = (list: AnyObject[] = []) => {
 // convert node item to internal node used by the tree
 export const refactorTree = (
   nodes: TreeNodeType[],
+  {
+    sorter = defaultSorter,
+    defaultExpanded = [],
+  }: { sorter?: any; defaultExpanded?: string[] },
   options: { level: number; parent?: string } = { level: 0, parent: undefined },
 ) => {
   const list: InternalNode[] = [];
-
-  nodes.sort(sorter).forEach((node) => {
+  if (sorter !== false) nodes = nodes.sort(sorter);
+  nodes.forEach((node) => {
     list.push({
-      open: false,
+      open: defaultExpanded.includes(node.id) && !!node.children?.length,
       level: options.level,
       parent: options.parent,
       checked: 0,
@@ -66,10 +75,14 @@ export const refactorTree = (
       leaf: !!node.leaf,
       children:
         node.children &&
-        refactorTree(node.children, {
-          level: options.level + 1,
-          parent: node.id,
-        }),
+        refactorTree(
+          node.children,
+          { sorter, defaultExpanded },
+          {
+            level: options.level + 1,
+            parent: node.id,
+          },
+        ),
     });
   });
 
@@ -106,34 +119,59 @@ export const flattenTree = (
           [...lines, idx === filtered.length - 1 ? 0 : 1],
         ),
       );
+    node.open &&
+      node.children?.length === 0 &&
+      list.push({
+        empty: true,
+        leaf: true,
+        lines: [...lines, idx === filtered.length - 1 ? 0 : 1, 2],
+      } as any);
   });
   return list;
 };
 
-export const updateSelection = (
-  nodes: Map<string, InternalNode>,
-  id: string,
-  selected?: true,
-) => {
-  const node = nodes.get(id);
+const selectNode = (nodes: Map<string, InternalNode>, id?: string) => {
+  const node = nodes.get(id ?? "");
   if (node) {
     // change selection
-    node.selected = selected;
+    node.selected = true;
     // update all parent childSelected flag
     let parent = node.parent;
     while (parent) {
       const parentNode = nodes.get(parent);
-      parentNode && (parentNode.childSelected = selected);
-      selected && parentNode && (parentNode.open = true);
+      parentNode && (parentNode.childSelected = true);
+      // selected && parentNode && (parentNode.open = true);
       parent = parentNode?.parent;
     }
   }
 };
 
-export const updateChecked = (list: Map<string, AnyObject>, id: string) => {
+export const updateSelection = ({
+  nodes,
+  multiple,
+  selected,
+}: {
+  nodes: Map<string, InternalNode>;
+  multiple?: boolean;
+  selected?: string | string[];
+}) => {
+  nodes.forEach((node) => {
+    node.selected = undefined;
+    node.childSelected = undefined;
+  });
+  multiple && isArray(selected)
+    ? selected?.forEach((id) => selectNode(nodes, id))
+    : selectNode(nodes, selected as string);
+};
+
+export const updateChecked = (
+  list: Map<string, AnyObject>,
+  id: string,
+  checked?: 0 | 1 | 2,
+) => {
   const node = list.get(id);
   if (node) {
-    node.checked = node.checked === 1 ? 0 : 1;
+    node.checked = checked ?? node.checked === 0 ? 1 : 0;
     // check all children
     node.children &&
       getChildren(node.children).forEach(