import test, { type Locator, type Page } from 'playwright/test'
import * as actions from './actions'
import { expect } from './customExpect'
import { CONTROL_KEY } from './keyboard'
import * as locate from './locate'
import { edgesFromNodeWithBinding, edgesToNodeWithBinding } from './locate'

/**
 * Every edge consists of multiple parts.
 * See edgeRendering.spec.ts for explanation.
 */
const EDGE_PARTS = 2

test.beforeEach(async ({ page }) => {
  await page.addInitScript(() => {
    class MockClipboard {
      private contents: ClipboardItem[] = []
      async read(): Promise<ClipboardItem[]> {
        return [...this.contents]
      }
      async write(contents: ClipboardItem[]) {
        this.contents = [...contents]
      }
    }
    Object.assign(window.navigator, {
      mockClipboard: new MockClipboard(),
    })
  })
})

test('Copy component with context menu', async ({ page }) => {
  await actions.goToGraph(page)
  const originalNodes = await locate.graphNode(page).count()
  const nodeToCopy = locate.graphNodeByBinding(page, 'final')
  await nodeToCopy.click({ button: 'right' })
  await expect(nodeToCopy).toBeSelected()
  await page
    .locator('.ComponentContextMenu')
    .getByRole('button', { name: 'Copy Component' })
    .click()
  await page.keyboard.press(`${CONTROL_KEY}+V`)
  await expect(nodeToCopy).not.toBeSelected()
  await expect(locate.selectedNodes(page)).toHaveCount(1)
  await expect(locate.graphNode(page)).toHaveCount(originalNodes + 1)
})

test('Copy component with comment', async ({ page }) => {
  await actions.goToGraph(page)

  // Check state before operation.
  const originalNodes = await locate.graphNode(page).count()
  await expect(locate.nodeCommentContent(page)).toExist()
  const originalNodeComments = await locate.nodeCommentContent(page).count()

  // Select a node.
  const nodeToCopy = locate.graphNodeByBinding(page, 'final')
  await nodeToCopy.click()
  await expect(nodeToCopy).toBeSelected()
  // Copy and paste it.
  await page.keyboard.press(`${CONTROL_KEY}+C`)
  await page.keyboard.press(`${CONTROL_KEY}+V`)
  await expect(nodeToCopy).not.toBeSelected()
  await expect(locate.selectedNodes(page)).toHaveCount(1)

  // Node and comment have been copied.
  await expect(locate.graphNode(page)).toHaveCount(originalNodes + 1)
  await expect(locate.nodeCommentContent(page)).toHaveCount(originalNodeComments + 1)
})

async function testCopyMultiple(
  page: Page,
  copyNodes: (node1: Locator, node2: Locator) => Promise<void>,
) {
  await actions.goToGraph(page)

  // Check state before operation.
  const originalNodes = await locate.graphNode(page).count()
  await expect(locate.nodeCommentContent(page)).toExist()
  const originalNodeComments = await locate.nodeCommentContent(page).count()

  // Select some nodes.
  const node1 = locate.graphNodeByBinding(page, 'final')
  const node2 = locate.graphNodeByBinding(page, 'prod')

  // Copy and paste.
  await copyNodes(node1, node2)
  await page.keyboard.press(`${CONTROL_KEY}+V`)
  await expect(node1).not.toBeSelected()
  await expect(node2).not.toBeSelected()
  await expect(locate.selectedNodes(page)).toHaveCount(2)

  // Nodes and comment have been copied.
  await expect(locate.graphNode(page)).toHaveCount(originalNodes + 2)
  // `final` node has a comment.
  await expect(locate.nodeCommentContent(page)).toHaveCount(originalNodeComments + 1)
  // Check that two copied nodes are isolated, i.e. connected to each other, not original nodes.
  await expect(locate.graphNodeByBinding(page, 'prod1')).toBeVisible()
  await expect(locate.graphNodeByBinding(page, 'final1')).toBeVisible()
  await expect(await edgesFromNodeWithBinding(page, 'sum')).toHaveCount(2 * EDGE_PARTS)
  await expect(await edgesFromNodeWithBinding(page, 'prod')).toHaveCount(1 * EDGE_PARTS)

  await expect(await edgesToNodeWithBinding(page, 'prod')).toHaveCount(1 * EDGE_PARTS)
  await expect(await edgesToNodeWithBinding(page, 'final')).toHaveCount(1 * EDGE_PARTS)
  await expect(await edgesToNodeWithBinding(page, 'prod1')).toHaveCount(1 * EDGE_PARTS)
  await expect(await edgesToNodeWithBinding(page, 'final1')).toHaveCount(1 * EDGE_PARTS)
}

test('Copy multiple components with keyboard shortcut', async ({ page }) => {
  await testCopyMultiple(page, async (node1, node2) => {
    await node1.click()
    await node2.click({ modifiers: ['Shift'] })
    await expect(node1).toBeSelected()
    await expect(node2).toBeSelected()
    await page.keyboard.press(`${CONTROL_KEY}+C`)
  })
})

test('Copy multiple components with context menu', async ({ page }) => {
  await testCopyMultiple(page, async (node1, node2) => {
    await node1.click()
    await node2.click({ modifiers: ['Shift'] })
    await expect(node1).toBeSelected()
    await expect(node2).toBeSelected()
    await node1.click({ button: 'right' })
    await page
      .locator('.ComponentContextMenu')
      .getByRole('button', { name: 'Copy Selected Components' })
      .click()
  })
})