Function toggleAllChildren

app/hooks/useVirtualTree.ts:154–203

(
  state: TreeState<T>,
  id: string
)

  };
}
function toggleAllChildren<T extends { id: string; children?: T[] }>(
  state: TreeState<T>,
  id: string
): TreeState<T> {
  const item = state.items.find(({ id: nodeId }) => nodeId === id);

  if (!item) {
    return state;
  }

  if (!item.node.children || item.node.children.length === 0) {
    return state;
  }

  const allCollapsed = item.node.children.every(
    (child) => state.collapsedState[child.id]
  );

  if (allCollapsed) {
    const collapsedState = item.node.children.reduce(
      (acc, child) => ({
        ...acc,
        [child.id]: false,
      }),
      state.collapsedState
    );

    return {
      ...state,
      collapsedState,
      items: createNodeItems(state.nodes, 0, collapsedState),
      focusedNodeId: id,
    };
  }

  const collapsedState = item.node.children.reduce(
    (acc, child) => ({
      ...acc,
      [child.id]: true,
    }),
    state.collapsedState
  );

  return {
    ...state,
    collapsedState,
    items: createNodeItems(state.nodes, 0, collapsedState),
    focusedNodeId: id,
  };
}

export function useVirtualTree<T extends { id: string; children?: T[] }, R>(
  options: UseVirtualTreeOptions<T, R>