A Tree implementation in Node JS.

Tree.js

class Node {
  constructor(data, children = []) {
    this.data = data;
    this.children = children;
  }

  add(data) {
    let node = new Node(data);
    this.children.push(node);
  }

  add(data) {
    this.children.push(new Node(data));
  }

  remove(data) {
    this.children = this.children.filter((node) => {
      return node.data !== data;
    });
  }
}

class Tree {
  constructor() {
    this.root = null;
  }

  traverseBF(fn) {
    let buffer = [];
    if (this.root) {
      buffer.push(this.root);
      while (buffer.length) {
        let firstNode = buffer.shift();
        fn(firstNode);
        buffer = [...buffer, ...firstNode.children];
      }
    }
  }

  traverseDF(fn) {
    let buffer = [];
    if (this.root) {
      buffer.push(this.root);
      while (buffer.length) {
        let firstNode = buffer.shift();
        fn(firstNode);
        buffer = [...firstNode.children, ...buffer];
      }
    }
  }
}

module.exports = { Tree, Node };