Persistent immutable tree update function

tree.ts

interface TreeNode {
  id: number;
  children: TreeNode[];
}

/* основная идея в том, чтобы не менять узел, если не менялись дочерние */

function updateTree(tree: TreeNode, id: number, node: TreeNode) {
  if (tree.id === id) { // мы нашли нужную ноду
    return {
      ...tree,
      children: [...tree.children, node]
    };
  } else { // ищем в дочерних, и вероятно обновляем
    let childrenChanged: boolean = false; // надеемся, что не будет обновления

    let newChildren = tree.children.map(subtree => {
      let newSubtree = updateTree(subtree, id, node);

      if (subtree !== newSubtree) { // отмечаем, что обновление все таки было
        childrenChanged = true;
      }

      return newSubtree;
    });

    if (childrenChanged) { // обновилось? -> строим новый узел
      return {
        ...tree,
        children: newChildren
      };
    } else { // иначе, возвращаем старый
      return tree;
    }
  }
}

let tree: TreeNode = {
  id: 1,
  children: []
};

let newTree = updateTree(tree, 1, {id: 2, children: []});

console.log(newTree);