Tree utils for searching, tracking and backtracking

tree-utils.ts

export function treeForwardTracking(
  treeNodes: TreeNode[],
  action: (TreeNode: TreeNode) => void
): void {
  treeNodes.forEach((node) => {
    action(node);
    if (node.children) {
      treeForwardTracking(node.children, action);
    }
  });
}

export function treeBackwardTracking(
  treeNodes: TreeNode[],
  action: (TreeNode: TreeNode) => void
): void {
  treeNodes.forEach((node) => {
    if (node.children) {
      treeBackwardTracking(node.children, action);
    }
    action(node);
  });
}

export function findTreeNodesByIds(
  tree: TreeNode[],
  ids: (string | undefined | null)[]
): TreeNode[] {
  ids = ids.filter((id) => id);
  if (ids.length === 0) {
    return [];
  }
  return tree.reduce((matchingNodes: TreeNode[], node: TreeNode) => {
    if (ids.includes(node.id)) {
      matchingNodes.push(node);
    }
    if (node.children) {
      matchingNodes.push(...findTreeNodesByIds(node.children, ids));
    }
    return matchingNodes;
  }, []);
}

export interface TreeNode {
  name: string;
  key: string;
  id: string;
  parentId: string | undefined;
  focused?: boolean;
  selected?: boolean;
  disabled?: boolean;
  expanded?: boolean;
  partiallySelected?: boolean;
  children?: Array<TreeNode>;
  itemCount?: number;
  additionalText?: string;
  badgeDisabled?: boolean;
  showParentSelection?: boolean;
  hidden?: boolean;
}