import { test, expect } from 'playwright-test-coverage';
import { Page } from '@playwright/test';

import { visitStudy, checkForScreenshot, screenShotPaths } from './utils';
import { press } from './utils/keyboardUtils';

test.beforeEach(async ({ page }) => {
  const studyInstanceUID = '1.3.12.2.1107.5.2.32.35162.30000015050317233592200000046';
  const mode = 'segmentation';
  await visitStudy(page, studyInstanceUID, mode, 2000);

  await page.getByTestId('panelSegmentationWithToolsLabelMap-btn').click();
  await page.getByTestId('addSegmentation').click();

  await page.waitForTimeout(500);
});

async function performDrawingToolInteraction(page: Page, toolName: string) {
  const brushRadiusInput = page.getByTestId(`${toolName}-radius`).locator('input');
  const viewportGrid = page.getByTestId('viewport-grid');
  const circle = viewportGrid.locator('svg.svg-layer circle').first();

  await expect(brushRadiusInput).toHaveValue('25');
  await viewportGrid.click({ position: { x: 275, y: 300 } });
  let radius = parseFloat(await circle.getAttribute('r'));
  expect(radius).toBeGreaterThanOrEqual(65);
  expect(radius).toBeLessThanOrEqual(68);

  await page.waitForTimeout(500);
  await press({ page, key: '[', nTimes: 2 });
  await expect(brushRadiusInput).toHaveValue('19');
  await viewportGrid.click({ position: { x: 500, y: 300 } });
  radius = parseFloat(await circle.getAttribute('r'));
  expect(radius).toBeGreaterThanOrEqual(49);
  expect(radius).toBeLessThanOrEqual(52);

  await page.waitForTimeout(500);

  await press({ page, key: ']', nTimes: 5 });
  await expect(brushRadiusInput).toHaveValue('34');
  await viewportGrid.click({ position: { x: 275, y: 500 } });
  radius = parseFloat(await circle.getAttribute('r'));
  expect(radius).toBeGreaterThanOrEqual(87);
  expect(radius).toBeLessThanOrEqual(90);

  await page.waitForTimeout(500);

  await brushRadiusInput.fill('42');
  await viewportGrid.click({ position: { x: 500, y: 500 } });
  radius = parseFloat(await circle.getAttribute('r'));
  expect(radius).toBeGreaterThanOrEqual(108);
  expect(radius).toBeLessThanOrEqual(111);

  await page.waitForTimeout(500);
}

test('(shouldUpdateThis) should resize segmentation brush tool', async ({ page }) => {
  await page.getByTestId('Brush-btn').click();

  await performDrawingToolInteraction(page, 'brush');

  await checkForScreenshot(
    page,
    page.getByTestId('viewport-grid'),
    screenShotPaths.segDrawingToolsResizing.brushTool
  );
});

test('(shouldUpdateThis) should resize segmentation eraser tool', async ({ page }) => {
  await page.getByTestId('Brush-btn').click();

  await page.getByTestId('brush-radius').locator('input').fill('99.5');
  await page.getByTestId('viewport-grid').click({ position: { x: 400, y: 400 } });

  await page.waitForTimeout(500);

  await page.getByTestId('Eraser-btn').click();
  await page.getByTestId('eraser-radius').locator('input').fill('25');

  await page.waitForTimeout(500);

  await performDrawingToolInteraction(page, 'eraser');

  await checkForScreenshot(
    page,
    page.getByTestId('viewport-grid'),
    screenShotPaths.segDrawingToolsResizing.eraserTool
  );
});

test('(shouldUpdateThis) should resize segmentation threshold tool', async ({ page }) => {
  await page.getByTestId('Threshold-btn').click();

  await performDrawingToolInteraction(page, 'threshold');
});