import { app } from '../../scripts/app'
import { LGraphCanvas, LiteGraph } from '@comfyorg/litegraph'

let touchZooming
let touchCount = 0

app.registerExtension({
  name: 'Comfy.SimpleTouchSupport',
  setup() {
    let zoomPos
    let touchTime
    let lastTouch

    function getMultiTouchPos(e) {
      return Math.hypot(
        e.touches[0].clientX - e.touches[1].clientX,
        e.touches[0].clientY - e.touches[1].clientY
      )
    }

    app.canvasEl.addEventListener(
      'touchstart',
      (e) => {
        touchCount++
        lastTouch = null
        if (e.touches?.length === 1) {
          // Store start time for press+hold for context menu
          touchTime = new Date()
          lastTouch = e.touches[0]
        } else {
          touchTime = null
          if (e.touches?.length === 2) {
            // Store center pos for zoom
            zoomPos = getMultiTouchPos(e)
            app.canvas.pointer_is_down = false
          }
        }
      },
      true
    )

    app.canvasEl.addEventListener('touchend', (e: TouchEvent) => {
      touchZooming = false
      touchCount = e.touches?.length ?? touchCount - 1
      if (touchTime && !e.touches?.length) {
        if (new Date().getTime() - touchTime > 600) {
          try {
            // hack to get litegraph to use this event
            e.constructor = CustomEvent
          } catch (error) {}
          // @ts-expect-error
          e.clientX = lastTouch.clientX
          // @ts-expect-error
          e.clientY = lastTouch.clientY

          app.canvas.pointer_is_down = true
          // @ts-expect-error
          app.canvas._mousedown_callback(e)
        }
        touchTime = null
      }
    })

    app.canvasEl.addEventListener(
      'touchmove',
      (e) => {
        touchTime = null
        if (e.touches?.length === 2) {
          app.canvas.pointer_is_down = false
          touchZooming = true
          // @ts-expect-error
          LiteGraph.closeAllContextMenus()
          // @ts-expect-error
          app.canvas.search_box?.close()
          const newZoomPos = getMultiTouchPos(e)

          const midX = (e.touches[0].clientX + e.touches[1].clientX) / 2
          const midY = (e.touches[0].clientY + e.touches[1].clientY) / 2

          let scale = app.canvas.ds.scale
          const diff = zoomPos - newZoomPos
          if (diff > 0.5) {
            scale *= 1 / 1.07
          } else if (diff < -0.5) {
            scale *= 1.07
          }
          app.canvas.ds.changeScale(scale, [midX, midY])
          app.canvas.setDirty(true, true)
          zoomPos = newZoomPos
        }
      },
      true
    )
  }
})

const processMouseDown = LGraphCanvas.prototype.processMouseDown
LGraphCanvas.prototype.processMouseDown = function (e) {
  if (touchZooming || touchCount) {
    return
  }
  return processMouseDown.apply(this, arguments)
}

const processMouseMove = LGraphCanvas.prototype.processMouseMove
LGraphCanvas.prototype.processMouseMove = function (e) {
  if (touchZooming || touchCount > 1) {
    return
  }
  return processMouseMove.apply(this, arguments)
}