equicord-companion/src/sidebar/Nodes.ts

Equicord Companion is a vscode extension to test Equicord patches & webpack finds right from the comfort of your IDE

import { TNode } from "@type/sidebar";

import RuntimeCommand from "./RuntimeCommand";

import { TreeItem, TreeItemCollapsibleState } from "vscode";

export class Item extends TreeItem {
    parrent: TNode | null = null;
}


export class Button extends Item {
    constructor(
        label: string,
        private readonly action: (...args: any[]) => any,
    ) {
        super(label, TreeItemCollapsibleState.None);
    }

    makeCommand(id: string) {
        this.command = new RuntimeCommand(`${id}${this.label}`, this.action)
            .asCommand();
    }
}
export class Text extends Item {
    constructor(text: string) {
        super(text, TreeItemCollapsibleState.None);
    }
}

export class Section extends Item {
    public readonly children: TNode[];

    constructor(
        label: string,
        children: TNode[],
        collapseableState = TreeItemCollapsibleState.Expanded,
    ) {
        super(label, collapseableState);
        this.children = children.map((e) => {
            if (e instanceof Item)
                e.parrent = this;
            return e;
        });
    }
}