import type { ComfyNode } from './comfy/index.js';
import { app, api, ComfyWidgets, LiteGraph, TLGraphNode, ComfyApp } from './comfy/index.js';
import { commonPrefix, displayContext } from './common.js';

// "Show any" Node
app.registerExtension({
  name: 'Crystools.Debugger.ConsoleAny',
  beforeRegisterNodeDef(nodeType: ComfyNode, nodeData: TLGraphNode, appFromArg: ComfyApp) {
    if (nodeData.name === 'Show any [Crystools]') {
      // 3 is the index of the text field in the node
      displayContext(nodeType, appFromArg, 3);
    }
  },
});

app.registerExtension({
  name: 'Crystools.Debugger.Metadata',
  registerCustomNodes() {
    class MetadataNode extends TLGraphNode {
      constructor() {
        super(`${commonPrefix} Show Metadata `);
        this.serialize_widgets = false;
        this.isVirtualNode = true;

        const widget = ComfyWidgets.STRING(this, '', [
          '', {default: '', multiline: true},
        ], app).widget;
        widget.inputEl.readOnly = true;
        ComfyWidgets.BOOLEAN(this, 'Active', [
          '', {default: true},
        ]);
        ComfyWidgets.BOOLEAN(this, 'Parsed', [
          '', {default: true},
        ]);
        ComfyWidgets.COMBO(this, 'What', [
          ['Prompt', 'Workflow'], {default: 'Prompt'},
        ]);

        // It runs at finish on each prompt queue
        api.addEventListener('executed', this.fillMetadataWidget, false);
      }

      fillMetadataWidget = (): Promise<string> => {
        return app.graphToPrompt()
        .then((workflow: any): string => {
          let result = 'inactive';
          if (this.widgets?.length !== 4) {
            console.error('Something is wrong with the widgets, should be 4!');
            return 'error';
          }
          const output = this.widgets[0];
          const active = this.widgets[1]?.value;
          const parsed = this.widgets[2]?.value;
          let what = this.widgets[3]?.value.toLowerCase();

          if (active) {
            what = what === 'prompt' ? 'output' : what; // little fix for better understanding
            // @ts-ignore
            result = workflow[what];
            if (parsed) {
              result = JSON.stringify(result, null, 2);
            } else {
              result = JSON.stringify(result);
            }
          }
          if (output) {
            output.value = result;
          } else {
            console.error('Something is wrong with the widgets, output is undefined!');
            return 'error';
          }
          return result;
        });
      };
    }

    // I'm not sure for what they're using prototype and lots of black magic, don't change the order!
    LiteGraph.registerNodeType('Show Metadata [Crystools]', MetadataNode);
    MetadataNode.category = `crystools ${commonPrefix}/Debugger`;
    MetadataNode.shape = LiteGraph.BOX_SHAPE;
    MetadataNode.title = `${commonPrefix} Show Metadata`;
    // MetadataNode.collapsable = false;
    // MetadataNode.color = '#FF2222';
    // MetadataNode.bgcolor = '#000000';
  },
});