108 lines
2.6 KiB
TypeScript
108 lines
2.6 KiB
TypeScript
import { toast } from 'react-toastify';
|
|
|
|
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);
|
|
if (this.children.length === 5000) {
|
|
toast.warn('An element with 5,000+ children was found. This can slow down rendering and degrade performance.', {
|
|
autoClose: false,
|
|
});
|
|
}
|
|
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(msg: { id: number, parentID: number, time: number }): boolean {
|
|
const { id, parentID } = msg;
|
|
if (!this.nodes[parentID]) {
|
|
// TODO: iframe case
|
|
// console.error(`Wrong! Node with id ${ parentID } (parentId) not found.`);
|
|
return false;
|
|
}
|
|
if (this.nodes[id]) {
|
|
// console.error(`Wrong! Node with id ${ id } already exists.`);
|
|
return false;
|
|
}
|
|
this.nodes[id] = this.nodes[parentID].newChild();
|
|
return true;
|
|
}
|
|
|
|
removeNode({ id }: { id: number }) {
|
|
if (!this.nodes[id]) {
|
|
// Might be text node
|
|
// console.error(`Wrong! Node with id ${ id } not found.`);
|
|
return false;
|
|
}
|
|
this.nodes[id].removeNode();
|
|
return true;
|
|
}
|
|
|
|
moveNode(msg: { id: number, parentID: number, time: number }) {
|
|
const { id, parentID, time } = msg;
|
|
if (!this.nodes[id]) {
|
|
console.warn(`Node Counter: Node with id ${id} (parent: ${parentID}) not found. time: ${time}`);
|
|
return false;
|
|
}
|
|
if (!this.nodes[parentID]) {
|
|
console.warn(
|
|
`Node Counter: Node with id ${parentID} (parentId) not found. time: ${time}`,
|
|
);
|
|
return false;
|
|
}
|
|
this.nodes[id].moveNode(this.nodes[parentID]);
|
|
return true;
|
|
}
|
|
|
|
get count() {
|
|
return this.root.count;
|
|
}
|
|
}
|