import { test as baseTest, describe, expect, vi } from 'vitest'

import { LinkConnector } from '@/lib/litegraph/src/canvas/LinkConnector'
import type { MovingInputLink } from '@/lib/litegraph/src/canvas/MovingInputLink'
import { ToInputRenderLink } from '@/lib/litegraph/src/canvas/ToInputRenderLink'
import type { LinkNetwork } from '@/lib/litegraph/src/interfaces'
import type { ISlotType } from '@/lib/litegraph/src/interfaces'
import {
  LGraph,
  LGraphNode,
  LLink,
  Reroute,
  type RerouteId
} from '@/lib/litegraph/src/litegraph'
import { LinkDirection } from '@/lib/litegraph/src/types/globalEnums'

interface TestContext {
  network: LinkNetwork & { add(node: LGraphNode): void }
  connector: LinkConnector
  setConnectingLinks: ReturnType<typeof vi.fn>
  createTestNode: (id: number, slotType?: ISlotType) => LGraphNode
  createTestLink: (
    id: number,
    sourceId: number,
    targetId: number,
    slotType?: ISlotType
  ) => LLink
}

const test = baseTest.extend<TestContext>({
  // eslint-disable-next-line no-empty-pattern
  network: async ({}, use) => {
    const graph = new LGraph()
    const floatingLinks = new Map<number, LLink>()
    const reroutes = new Map<number, Reroute>()

    await use({
      links: new Map<number, LLink>(),
      reroutes,
      floatingLinks,
      getLink: graph.getLink.bind(graph),
      getNodeById: (id: number) => graph.getNodeById(id),
      addFloatingLink: (link: LLink) => {
        floatingLinks.set(link.id, link)
        return link
      },
      removeFloatingLink: (link: LLink) => floatingLinks.delete(link.id),
      getReroute: ((id: RerouteId | null | undefined) =>
        id == null ? undefined : reroutes.get(id)) as LinkNetwork['getReroute'],
      removeReroute: (id: number) => reroutes.delete(id),
      add: (node: LGraphNode) => graph.add(node)
    })
  },

  setConnectingLinks: async (
    // eslint-disable-next-line no-empty-pattern
    {},
    use: (mock: ReturnType<typeof vi.fn>) => Promise<void>
  ) => {
    const mock = vi.fn()
    await use(mock)
  },
  connector: async ({ setConnectingLinks }, use) => {
    const connector = new LinkConnector(setConnectingLinks)
    await use(connector)
  },

  createTestNode: async ({ network }, use) => {
    await use((id: number): LGraphNode => {
      const node = new LGraphNode('test')
      node.id = id
      network.add(node)
      return node
    })
  },
  createTestLink: async ({ network }, use) => {
    await use(
      (
        id: number,
        sourceId: number,
        targetId: number,
        slotType: ISlotType = 'number'
      ): LLink => {
        const link = new LLink(id, slotType, sourceId, 0, targetId, 0)
        network.links.set(link.id, link)
        return link
      }
    )
  }
})

describe('LinkConnector', () => {
  test('should initialize with default state', ({ connector }) => {
    expect(connector.state).toEqual({
      connectingTo: undefined,
      multi: false,
      draggingExistingLinks: false
    })
    expect(connector.renderLinks).toEqual([])
    expect(connector.inputLinks).toEqual([])
    expect(connector.outputLinks).toEqual([])
    expect(connector.hiddenReroutes.size).toBe(0)
  })

  describe('Moving Input Links', () => {
    test('should handle moving input links', ({
      network,
      connector,
      createTestNode
    }) => {
      const sourceNode = createTestNode(1)
      const targetNode = createTestNode(2)

      const slotType: ISlotType = 'number'
      sourceNode.addOutput('out', slotType)
      targetNode.addInput('in', slotType)

      const link = new LLink(1, slotType, 1, 0, 2, 0)
      network.links.set(link.id, link)
      targetNode.inputs[0].link = link.id

      connector.moveInputLink(network, targetNode.inputs[0])

      expect(connector.state.connectingTo).toBe('input')
      expect(connector.state.draggingExistingLinks).toBe(true)
      expect(connector.inputLinks).toContain(link)
      expect(link._dragging).toBe(true)
    })

    test('should not move input link if already connecting', ({
      connector,
      network
    }) => {
      connector.state.connectingTo = 'input'

      expect(() => {
        connector.moveInputLink(network, { link: 1 } as any)
      }).toThrow('Already dragging links.')
    })
  })

  describe('Moving Output Links', () => {
    test('should handle moving output links', ({
      network,
      connector,
      createTestNode
    }) => {
      const sourceNode = createTestNode(1)
      const targetNode = createTestNode(2)

      const slotType: ISlotType = 'number'
      sourceNode.addOutput('out', slotType)
      targetNode.addInput('in', slotType)

      const link = new LLink(1, slotType, 1, 0, 2, 0)
      network.links.set(link.id, link)
      sourceNode.outputs[0].links = [link.id]

      connector.moveOutputLink(network, sourceNode.outputs[0])

      expect(connector.state.connectingTo).toBe('output')
      expect(connector.state.draggingExistingLinks).toBe(true)
      expect(connector.state.multi).toBe(true)
      expect(connector.outputLinks).toContain(link)
      expect(link._dragging).toBe(true)
    })

    test('should not move output link if already connecting', ({
      connector,
      network
    }) => {
      connector.state.connectingTo = 'output'

      expect(() => {
        connector.moveOutputLink(network, { links: [1] } as any)
      }).toThrow('Already dragging links.')
    })
  })

  describe('Dragging New Links', () => {
    test('should handle dragging new link from output', ({
      network,
      connector,
      createTestNode
    }) => {
      const sourceNode = createTestNode(1)
      const slotType: ISlotType = 'number'
      sourceNode.addOutput('out', slotType)

      connector.dragNewFromOutput(network, sourceNode, sourceNode.outputs[0])

      expect(connector.state.connectingTo).toBe('input')
      expect(connector.renderLinks.length).toBe(1)
      expect(connector.state.draggingExistingLinks).toBe(false)
    })

    test('should handle dragging new link from input', ({
      network,
      connector,
      createTestNode
    }) => {
      const targetNode = createTestNode(1)
      const slotType: ISlotType = 'number'
      targetNode.addInput('in', slotType)

      connector.dragNewFromInput(network, targetNode, targetNode.inputs[0])

      expect(connector.state.connectingTo).toBe('output')
      expect(connector.renderLinks.length).toBe(1)
      expect(connector.state.draggingExistingLinks).toBe(false)
    })
  })

  describe('Dragging from reroutes', () => {
    test('should handle dragging from reroutes', ({
      network,
      connector,
      createTestNode,
      createTestLink
    }) => {
      const originNode = createTestNode(1)
      const targetNode = createTestNode(2)

      const output = originNode.addOutput('out', 'number')
      targetNode.addInput('in', 'number')

      const link = createTestLink(1, 1, 2)
      const reroute = new Reroute(1, network, [0, 0], undefined, [link.id])
      network.reroutes.set(reroute.id, reroute)
      link.parentId = reroute.id

      connector.dragFromReroute(network, reroute)

      expect(connector.state.connectingTo).toBe('input')
      expect(connector.state.draggingExistingLinks).toBe(false)
      expect(connector.renderLinks.length).toBe(1)

      const renderLink = connector.renderLinks[0]
      expect(renderLink instanceof ToInputRenderLink).toBe(true)
      expect(renderLink.toType).toEqual('input')
      expect(renderLink.node).toEqual(originNode)
      expect(renderLink.fromSlot).toEqual(output)
      expect(renderLink.fromReroute).toEqual(reroute)
      expect(renderLink.fromDirection).toEqual(LinkDirection.NONE)
      expect(renderLink.network).toEqual(network)
    })
  })

  describe('Reset', () => {
    test('should reset state and clear links', ({ network, connector }) => {
      connector.state.connectingTo = 'input'
      connector.state.multi = true
      connector.state.draggingExistingLinks = true

      const link = new LLink(1, 'number', 1, 0, 2, 0)
      link._dragging = true
      connector.inputLinks.push(link)

      const reroute = new Reroute(1, network)
      reroute.pos = [0, 0]
      reroute._dragging = true
      connector.hiddenReroutes.add(reroute)

      connector.reset()

      expect(connector.state).toEqual({
        connectingTo: undefined,
        multi: false,
        draggingExistingLinks: false
      })
      expect(connector.renderLinks).toEqual([])
      expect(connector.inputLinks).toEqual([])
      expect(connector.outputLinks).toEqual([])
      expect(connector.hiddenReroutes.size).toBe(0)
      expect(link._dragging).toBeUndefined()
      expect(reroute._dragging).toBeUndefined()
    })
  })

  describe('Event Handling', () => {
    test('should handle event listeners until reset', ({
      connector,
      createTestNode
    }) => {
      const listener = vi.fn()
      connector.listenUntilReset('input-moved', listener)

      const sourceNode = createTestNode(1)

      const mockRenderLink = {
        node: sourceNode,
        fromSlot: { name: 'out', type: 'number' },
        fromPos: [0, 0],
        fromDirection: LinkDirection.RIGHT,
        toType: 'input',
        link: new LLink(1, 'number', 1, 0, 2, 0)
      } as MovingInputLink

      connector.events.dispatch('input-moved', mockRenderLink)
      expect(listener).toHaveBeenCalled()

      connector.reset()
      connector.events.dispatch('input-moved', mockRenderLink)
      expect(listener).toHaveBeenCalledTimes(1)
    })
  })

  describe('Export', () => {
    test('should export current state', ({ network, connector }) => {
      connector.state.connectingTo = 'input'
      connector.state.multi = true

      const link = new LLink(1, 'number', 1, 0, 2, 0)
      connector.inputLinks.push(link)

      const exported = connector.export(network)

      expect(exported.state).toEqual(connector.state)
      expect(exported.inputLinks).toEqual(connector.inputLinks)
      expect(exported.outputLinks).toEqual(connector.outputLinks)
      expect(exported.renderLinks).toEqual(connector.renderLinks)
      expect(exported.network).toBe(network)
    })
  })
})