// / <reference path="../node_modules/litegraph.js/src/litegraph.d.ts" />
// @ts-ignore
import {app} from "../../scripts/app.js";
import { BaseAnyInputConnectedNode } from "./base_any_input_connected_node.js";
import { RgthreeBaseNode } from "./base_node.js";
import type {LGraphNode as TLGraphNode, LiteGraph as TLiteGraph, IWidget} from './typings/litegraph.js';
import { PassThroughFollowing } from "./utils.js";
import { wait } from "./shared_utils.js";

declare const LiteGraph: typeof TLiteGraph;
declare const LGraphNode: typeof TLGraphNode;

export class BaseNodeModeChanger extends BaseAnyInputConnectedNode {

  override readonly inputsPassThroughFollowing: PassThroughFollowing = PassThroughFollowing.ALL;

  static collapsible = false;
  override isVirtualNode = true;

  // These Must be overriden
  readonly modeOn: number = -1;
  readonly modeOff: number = -1;

  constructor(title?: string) {
    super(title);

    wait(10).then(() => {
      if (this.modeOn < 0 || this.modeOff < 0) {
        throw new Error('modeOn and modeOff must be overridden.');
      }
    });
    this.addOutput("OPT_CONNECTION", "*");
  }

  override handleLinkedNodesStabilization(linkedNodes: TLGraphNode[]) {
    for (const [index, node] of linkedNodes.entries()) {
      let widget = this.widgets && this.widgets[index];
      if (!widget) {
        // When we add a widget, litegraph is going to mess up the size, so we
        // store it so we can retrieve it in computeSize. Hacky..
        (this as any)._tempWidth = this.size[0];
        widget = this.addWidget('toggle', '', false, '', {"on": 'yes', "off": 'no'});
      }
      node && this.setWidget(widget, node);
    }
    if (this.widgets && this.widgets.length > linkedNodes.length) {
      this.widgets.length = linkedNodes.length
    }
  }

  setWidget(widget: IWidget, linkedNode: TLGraphNode) {
    const off = linkedNode.mode === this.modeOff;
    widget.name = `Enable ${linkedNode.title}`;
    widget.options = {'on': 'yes', 'off': 'no'}
    widget.value = !off;
    (widget as any).doModeChange = (force?: boolean) => {
      let off = force == null ? linkedNode.mode === this.modeOff : force;
      linkedNode.mode = (off ? this.modeOn : this.modeOff) as 1 | 2 | 3 | 4;
      widget!.value = off;
    }
    widget.callback = () => {
      (widget as any).doModeChange();
    }
  }

  forceWidgetOff(widget: IWidget) {
    (widget as any).doModeChange(false);
  }
  forceWidgetOn(widget: IWidget) {
    (widget as any).doModeChange(true);
  }


  static override setUp<T extends RgthreeBaseNode>(clazz: new(title?: string) => T) {
    BaseAnyInputConnectedNode.setUp(clazz);
  }
}