import { app } from "/scripts/app.js";
import * as shared from '/extensions/mtb/comfy_shared.js'
import { log } from '/extensions/mtb/comfy_shared.js'
import { MtbWidgets } from '/extensions/mtb/mtb_widgets.js'

// TODO: respect inputs order...



app.registerExtension({
    name: "mtb.Debug",
    async beforeRegisterNodeDef(nodeType, nodeData, app) {
        if (nodeData.name === "Debug (mtb)") {
            const onConnectionsChange = nodeType.prototype.onConnectionsChange;
            nodeType.prototype.onConnectionsChange = function (type, index, connected, link_info) {
                const r = onConnectionsChange ? onConnectionsChange.apply(this, arguments) : undefined;
                // TODO: remove all widgets on disconnect once computed
                shared.dynamic_connection(this, index, connected, "anything_", "*")

                //- infer type
                if (link_info) {
                    const fromNode = this.graph._nodes.find((otherNode) => otherNode.id == link_info.origin_id);
                    const type = fromNode.outputs[link_info.origin_slot].type;
                    this.inputs[index].type = type;
                    // this.inputs[index].label = type.toLowerCase()
                }
                //- restore dynamic input
                if (!connected) {
                    this.inputs[index].type = "*";
                    this.inputs[index].label = `anything_${index + 1}`
                }
            }

            const onExecuted = nodeType.prototype.onExecuted;
            nodeType.prototype.onExecuted = function (message) {
                log(message)
                onExecuted?.apply(this, arguments);
                log(message)
                if (this.widgets) {
                    // const pos = this.widgets.findIndex((w) => w.name === "anything_1");
                    // if (pos !== -1) {
                    for (let i = 0; i < this.widgets.length; i++) {
                        this.widgets[i].onRemove?.();
                    }
                    this.widgets.length = 0;

                }
                let widgetI = 1
                if (message.text) {
                    for (const txt of message.text) {
                        const w = this.addCustomWidget(MtbWidgets.DEBUG_STRING(txt, widgetI))
                        w.parent = this;
                        widgetI++;
                    }
                }
                if (message.b64_images) {
                    for (const img of message.b64_images) {
                        const w = this.addCustomWidget(MtbWidgets.DEBUG_IMG(img, widgetI))
                        w.parent = this;
                        widgetI++;
                    }
                    // this.onResize?.(this.size);
                    // this.resize?.(this.size)
                    this.setSize(this.computeSize())
                };

                this.onRemoved = function () {
                    // When removing this node we need to remove the input from the DOM
                    for (let y in this.widgets) {
                        if (this.widgets[y].canvas) {
                            this.widgets[y].canvas.remove();
                        }
                        this.widgets[y].onRemove?.();
                    }
                }
            }
        }
    }
}
);