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
	}

}