import { LiteGraph } from '@comfyorg/litegraph'
import type { LGraphNode } from '@comfyorg/litegraph'
import { useEventListener } from '@vueuse/core'

import { app } from '@/scripts/app'
import { useCanvasStore } from '@/stores/graphStore'
import { useWorkspaceStore } from '@/stores/workspaceStore'
import { ComfyWorkflowJSON } from '@/types/comfyWorkflow'
import { isImageNode } from '@/utils/litegraphUtil'

/**
 * Adds a handler on paste that extracts and loads images or workflows from pasted JSON data
 */
export const usePaste = () => {
  const workspaceStore = useWorkspaceStore()
  const canvasStore = useCanvasStore()

  useEventListener(document, 'paste', async (e: ClipboardEvent) => {
    // ctrl+shift+v is used to paste nodes with connections
    // this is handled by litegraph
    if (workspaceStore.shiftDown) return

    const canvas = canvasStore.canvas
    if (!canvas) return

    const graph = canvas.graph
    // @ts-expect-error: Property 'clipboardData' does not exist on type 'Window & typeof globalThis'.
    // Did you mean 'Clipboard'?ts(2551)
    // TODO: Not sure what the code wants to do.
    let data = e.clipboardData || window.clipboardData
    const items = data.items

    // Look for image paste data
    for (const item of items) {
      if (item.type.startsWith('image/')) {
        let imageNode: LGraphNode | null = null

        // If an image node is selected, paste into it
        const currentNode = canvas.current_node as LGraphNode
        if (
          currentNode &&
          currentNode.is_selected &&
          isImageNode(currentNode)
        ) {
          imageNode = currentNode
        }

        // No image node selected: add a new one
        if (!imageNode) {
          const newNode = LiteGraph.createNode('LoadImage')
          // @ts-expect-error array to Float32Array
          newNode.pos = [...canvas.graph_mouse]
          imageNode = graph.add(newNode) ?? null
          graph.change()
        }
        const blob = item.getAsFile()
        imageNode?.pasteFile?.(blob)
        return
      }
    }

    // No image found. Look for node data
    data = data.getData('text/plain')
    let workflow: ComfyWorkflowJSON | null = null
    try {
      data = data.slice(data.indexOf('{'))
      workflow = JSON.parse(data)
    } catch (err) {
      try {
        data = data.slice(data.indexOf('workflow\n'))
        data = data.slice(data.indexOf('{'))
        workflow = JSON.parse(data)
      } catch (error) {
        workflow = null
      }
    }

    if (workflow && workflow.version && workflow.nodes && workflow.extra) {
      await app.loadGraphData(workflow)
    } else {
      if (
        (e.target instanceof HTMLTextAreaElement &&
          e.target.type === 'textarea') ||
        (e.target instanceof HTMLInputElement && e.target.type === 'text')
      ) {
        return
      }

      // Litegraph default paste
      canvas.pasteFromClipboard()
    }
  })
}