import { app } from "../../scripts/app.js";
import { chainCallback, setupInputWidgetWithAutocomplete } from "./utils.js";

app.registerExtension({
  name: "LoraManager.Prompt",

  async beforeRegisterNodeDef(nodeType) {
    if (nodeType.comfyClass === "Prompt (LoraManager)") {
      chainCallback(nodeType.prototype, "onNodeCreated", function () {
        this.serialize_widgets = true;

        const textWidget = this.widgets?.[0];
        if (!textWidget) {
          return;
        }

        const originalCallback =
          typeof textWidget.callback === "function" ? textWidget.callback : null;

        textWidget.callback = setupInputWidgetWithAutocomplete(
          this,
          textWidget,
          originalCallback,
          "embeddings"
        );
      });
    }
  },
  async loadedGraphNode(node) {
    if (node.comfyClass == "Prompt (LoraManager)") {
      const textWidget = node.widgets?.[0];
      if (textWidget && !node.autocomplete) {
        const { setupInputWidgetWithAutocomplete } = await import("./utils.js");
        const modelType = "embeddings";
        const autocompleteOptions = {
          maxItems: 20,
          minChars: 1,
          debounceDelay: 200,
        };
        textWidget.callback = setupInputWidgetWithAutocomplete(node, textWidget, textWidget.callback, modelType, autocompleteOptions);
      }
    }
  },
});