elexis/lib/$NodeManager.ts

import { $Container } from "./node/$Container";
import { $Node } from "./node/$Node";
import { $Text } from "./node/$Text";

export class $NodeManager {
    $container: $Container;
    $elementList = new Set<$Node>
    constructor(container: $Container) {
        this.$container = container;
    }

    add(element: $Node | string) {
        if (typeof element === 'string') {
            const text = new $Text(element);
            this.$elementList.add(text);
            (text as Mutable<$Node>).parent = this.$container;
        } else {
            this.$elementList.add(element);
            (element as Mutable<$Node>).parent = this.$container;
        }
    }

    remove(element: $Node) {
        if (!this.$elementList.has(element)) return this;
        this.$elementList.delete(element);
        (element as Mutable<$Node>).parent = undefined;
        return this;
    }

    removeAll(render = true) {
        this.$elementList.forEach(ele => this.remove(ele));
        if (render) this.render();
    }

    replace(target: $Node, replace: $Node) {
        const array = this.array
        array.splice(array.indexOf(target), 1, replace);
        target.remove();
        this.$elementList.clear();
        array.forEach(node => this.$elementList.add(node));
        return this;
    }

    render() {
        const [domList, nodeList] = [this.array.map(node => node.dom), Array.from(this.dom.childNodes)];
        const appendedNodeList: Node[] = []; // appended node list
        // Rearrange
        while (nodeList.length || domList.length) { // while nodeList or domList has item
            const [node, dom] = [nodeList.at(0), domList.at(0)];
            if (!dom) { if (node && !appendedNodeList.includes(node)) node.remove(); nodeList.shift()} 
            else if (!node) { if (!dom.$.__hidden) this.dom.append(dom); domList.shift();}
            else if (dom !== node) { 
                if (!dom.$.__hidden) { this.dom.insertBefore(dom, node); appendedNodeList.push(dom) }
                domList.shift();
            }
            else {
                if (dom.$.__hidden) this.dom.removeChild(dom);
                domList.shift(); nodeList.shift();
            }
        }
    }

    get array() {return [...this.$elementList.values()]};

    get dom() {return this.$container.dom}
}