Function createTree

src/aria/private/tree/tree.spec.ts:64–114

(treeData: TestTreeItem<V>[], treeInputs: TestTreeInputs<V>)

describe('Tree Pattern', () => {
  function createTree<V>(treeData: TestTreeItem<V>[], treeInputs: TestTreeInputs<V>) {
    const items = signal<TreeItemPattern<V>[]>([]);
    const itemPatternInputsMap = new Map<string, TestTreeItemInputs<V>>();
    const tree = new TreePattern<V>({
      ...treeInputs,
      items,
    });

    let nextId = 0;

    function buildItems(
      treeData: TestTreeItem<V>[],
      parent: TreeItemPattern<V> | TreePattern<V>,
    ): TreeItemPattern<V>[] {
      const levelItems: TreeItemPattern<V>[] = [];

      for (const node of treeData) {
        const itemId = `treeitem-${nextId++}`;
        const element = createTreeItemElement(itemId);
        const itemPatternInputs: TestTreeItemInputs<V> = {
          id: signal(itemId),
          value: signal(node.value),
          element: signal(element),
          disabled: signal(node.disabled),
          selectable: signal(node.selectable),
          expanded: signal(node.expanded),
          searchTerm: signal(String(node.value)),
          parent: signal(parent),
          hasChildren: signal((node.children ?? []).length > 0),
          children: signal<TreeItemPattern<V>[]>([]),
          tree: signal(tree),
        };

        const item = new TreeItemPattern(itemPatternInputs);
        itemPatternInputsMap.set(itemId, itemPatternInputs);

        items.set([...items(), item]);
        levelItems.push(item);

        const childItems = buildItems(node.children ?? [], item);
        itemPatternInputs.children.set(childItems);
      }
      return levelItems;
    }

    // Build tree items recursively.
    buildItems(treeData, tree as TreePattern<V>);
    tree.activeItem.set(items()[0]);

    return {tree, items, itemPatternInputsMap};
  }

  function getItemByValue<V>(items: TreeItemPattern<V>[], value: V) {
    return items.find(i => i.value() === value)!;