openreplay/frontend/app/player/web/managers/WindowNodeCounter.ts

class NodeCounter {
  private parent: NodeCounter | null = null;

  private _count: number = 0;

  private children: Array<NodeCounter> = [];

  bubbleCount(count: number) {
    this._count += count;
    if (this.parent != null) {
      this.parent.bubbleCount(count);
    }
  }

  newChild(): NodeCounter {
    const child = new NodeCounter();
    this.children.push(child);
    child.parent = this;
    this.bubbleCount(1);
    return child;
  }

  removeChild(child: NodeCounter) {
    this.children = this.children.filter((c) => c != child);
    this.bubbleCount(-(child._count + 1));
  }

  removeNode() {
    if (this.parent) {
      this.parent.removeChild(this);
    }
    this.parent = null;
  }

  moveNode(newParent: NodeCounter) {
    this.removeNode();
    newParent.children.push(this);
    this.parent = newParent;
    newParent.bubbleCount(this._count + 1);
  }

  get count() {
    return this._count;
  }
}

export default class WindowNodeCounter {
  private root: NodeCounter = new NodeCounter();

  private nodes: Array<NodeCounter> = [this.root];

  reset() {
    this.root = new NodeCounter();
    this.nodes = [this.root];
  }

  addNode(id: number, parentID: number) {
    if (!this.nodes[parentID]) {
      // TODO: iframe case
      // console.error(`Wrong! Node with id ${ parentID } (parentId) not found.`);
      return;
    }
    if (this.nodes[id]) {
      // console.error(`Wrong! Node with id ${ id } already exists.`);
      return;
    }
    this.nodes[id] = this.nodes[parentID].newChild();
  }

  removeNode(id: number) {
    if (!this.nodes[id]) {
      // Might be text node
      // console.error(`Wrong! Node with id ${ id } not found.`);
      return;
    }
    this.nodes[id].removeNode();
  }

  moveNode(id: number, parentId: number) {
    if (!this.nodes[id]) {
      console.warn(`Node Counter: Node with id ${id} not found.`);
      return;
    }
    if (!this.nodes[parentId]) {
      console.warn(
        `Node Counter: Node with id ${parentId} (parentId) not found.`,
      );
      return;
    }
    this.nodes[id].moveNode(this.nodes[parentId]);
  }

  get count() {
    return this.root.count;
  }
}