/*
Preview Any - original implement from
https://github.com/rgthree/rgthree-comfy/blob/main/py/display_any.py
upstream requested in https://github.com/Kosinkadink/rfcs/blob/main/rfcs/0000-corenodes.md#preview-nodes
 */
import { app } from '@/scripts/app'
import { type DOMWidget } from '@/scripts/domWidget'
import { ComfyWidgets } from '@/scripts/widgets'
import { useExtensionService } from '@/services/extensionService'

useExtensionService().registerExtension({
  name: 'Comfy.PreviewAny',
  async beforeRegisterNodeDef(nodeType, nodeData) {
    if (nodeData.name === 'PreviewAny') {
      const onNodeCreated = nodeType.prototype.onNodeCreated

      nodeType.prototype.onNodeCreated = function () {
        onNodeCreated ? onNodeCreated.apply(this, []) : undefined

        const showValueWidget = ComfyWidgets['MARKDOWN'](
          this,
          'preview',
          ['MARKDOWN', {}],
          app
        ).widget as DOMWidget<HTMLTextAreaElement, string>

        const showValueWidgetPlain = ComfyWidgets['STRING'](
          this,
          'preview',
          ['STRING', { multiline: true }],
          app
        ).widget as DOMWidget<HTMLTextAreaElement, string>

        const showAsPlaintextWidget = ComfyWidgets['BOOLEAN'](
          this,
          'previewMode',
          [
            'BOOLEAN',
            { label_on: 'Markdown', label_off: 'Plaintext', default: false }
          ],
          app
        )

        showAsPlaintextWidget.widget.callback = (value: boolean) => {
          showValueWidget.hidden = !value
          showValueWidget.options.hidden = !value
          showValueWidgetPlain.hidden = value
          showValueWidgetPlain.options.hidden = value
        }

        showValueWidget.hidden = true
        showValueWidget.options.hidden = true
        showValueWidget.options.read_only = true
        showValueWidget.element.readOnly = true
        showValueWidget.element.disabled = true
        showValueWidget.serialize = false

        showValueWidgetPlain.hidden = false
        showValueWidgetPlain.options.hidden = false
        showValueWidgetPlain.options.read_only = true
        showValueWidgetPlain.element.readOnly = true
        showValueWidgetPlain.element.disabled = true
        showValueWidgetPlain.serialize = false
      }

      const onExecuted = nodeType.prototype.onExecuted

      nodeType.prototype.onExecuted = function (message) {
        onExecuted === null || onExecuted === void 0
          ? void 0
          : onExecuted.apply(this, [message])

        const previewWidgets =
          this.widgets?.filter((w) => w.name === 'preview') ?? []

        for (const previewWidget of previewWidgets) {
          const text = message.text ?? ''
          previewWidget.value = Array.isArray(text) ? (text[0] ?? '') : text
        }
      }
    }
  }
})