D3 Сворачиваемое дерево - сворачивайте по одному уровню за раз

У меня есть разборное дерево d3. У меня есть внешняя кнопка, которая управляет расширением и сворачиванием дерева. Я могу все разложить, все разложить. Кроме того, я могу расширяться на один уровень за раз. У меня проблемы с одновременным свертыванием одного уровня.

Для тех, кто не понимает d3. Если узел расширен, то d.children не пуст и d.children пуст, и если узел свернут d.children не пуст, а d.children пуст.

Вот моя логика расширения на один уровень:

private expandNodeOneLevel(d) {
    var children = (d.children) ? d.children : d._children;
    if (!d.children) {
       if (children) {
           d.children = d._children;
           d._children = null;
       }
       return;
     }
    if (children) {
       children.forEach((node) => this.expandNodeOneLevel(node));
     }
}



 public expandOneLevelNodes() {
    this.expandNodeOneLevel(this.root);
    this.update(this.root);
  }

Вот моя логика - сворачивать по одному уровню за раз, что неверно, так как сворачиваются все узлы дерева. У меня проблемы с нарушением рекурсии.

private collpaseOneLevelNodes = (d) => {
    var children = (d.children) ? d.children : d._children;
    if (!children) {
      return;
    }
    children.forEach((node) => {
      this.collpaseOneLevelNodes(node);
    });
    if (d.children) {
        d._children = d.children;
        d.children = null;
    }
    return;
  }



 public collapseOneLevel() {
    this.collpaseOneLevelNodes(this.root);
    this.update(this.root);
  }

Я был бы очень признателен, если бы кто-нибудь указал, в чем моя ошибка, или какой-либо другой лучший подход.