openpencil/tests/e2e/color-picker/basic.spec.ts

import { expect, test, type Page } from '@playwright/test'

import type { OkHCLPayload } from '@open-pencil/core/color'

import { CanvasHelper } from '#tests/helpers/canvas'

let page: Page
let canvas: CanvasHelper

test.describe.configure({ mode: 'serial' })

test.beforeAll(async ({ browser }) => {
  page = await browser.newPage()
  await page.goto('/')
  canvas = new CanvasHelper(page)
  await canvas.waitForInit()
})

test.afterAll(async () => {
  await page.close()
})

async function getSelectedFill() {
  return page.evaluate(() => {
    const store = window.openPencil?.getStore?.()
    if (!store) throw new Error('OpenPencil store not initialized')
    const id = [...store.state.selectedIds][0]
    if (!id) return null
    const node = store.graph.getNode(id)
    return node?.fills?.[0] ?? null
  })
}

async function getSelectedFillOkHCL() {
  return page.evaluate(() => {
    const store = window.openPencil?.getStore?.()
    if (!store) throw new Error('OpenPencil store not initialized')
    const id = [...store.state.selectedIds][0]
    if (!id) return null
    const node = store.graph.getNode(id)
    const entry = node?.pluginData.find(
      (value) => value.pluginId === 'open-pencil' && value.key === 'okhcl'
    )
    if (!entry) return null
    const payload = JSON.parse(entry.value) as Partial<OkHCLPayload>
    return payload.kind === 'fill' && payload.index === 0 ? (payload.color ?? null) : null
  })
}

async function openFillPicker() {
  const solidTab = page.getByTestId('fill-picker-tab-solid')
  if (await solidTab.isVisible().catch(() => false)) return
  const swatch = page.getByTestId('fill-picker-swatch').first()
  await swatch.click()
  await expect(solidTab).toBeVisible()
}

async function chooseFormat(label: 'RGB' | 'HSL' | 'HSB' | 'OkHCL') {
  await page.getByTestId('color-format-select').click()
  await page.getByRole('option', { name: label, exact: true }).click()
}

async function dragSlider(testId: string, ratio: number) {
  const slider = page.getByTestId(testId).locator(':scope > [data-orientation="horizontal"]')
  const box = await slider.boundingBox()
  if (!box) throw new Error(`Missing slider: ${testId}`)
  await slider.click({
    position: {
      x: Math.max(2, Math.min(box.width - 2, box.width * ratio)),
      y: box.height / 2
    }
  })
  await canvas.waitForRender()
}

test('rgb hue slider updates selected fill color', async () => {
  await canvas.clearCanvas()
  await canvas.drawRect(100, 100, 160, 120)
  await canvas.waitForRender()

  await openFillPicker()
  const before = await getSelectedFill()
  await dragSlider('color-slider-hue', 0.65)
  const after = await getSelectedFill()

  expect(after).not.toBeNull()
  expect(
    before?.color.r !== after?.color.r ||
      before?.color.g !== after?.color.g ||
      before?.color.b !== after?.color.b
  ).toBe(true)
})

test('rgb alpha slider updates fill opacity and alpha', async () => {
  await openFillPicker()
  await dragSlider('color-slider-alpha', 0.3)
  const after = await getSelectedFill()

  expect(after).not.toBeNull()
  expect(after?.opacity).toBeLessThan(1)
  expect(after?.color.a).toBeCloseTo(after?.opacity ?? 0, 3)
})

test('hsl saturation slider changes saturation', async () => {
  await openFillPicker()
  await chooseFormat('HSL')
  const before = await getSelectedFill()
  await dragSlider('color-slider-hsl-s', 0.2)
  const after = await getSelectedFill()

  expect(after).not.toBeNull()
  expect(
    before?.color.r !== after?.color.r ||
      before?.color.g !== after?.color.g ||
      before?.color.b !== after?.color.b
  ).toBe(true)
})

test('hsl lightness slider changes color independently', async () => {
  await openFillPicker()
  await chooseFormat('HSL')
  const before = await getSelectedFill()
  await dragSlider('color-slider-hsl-l', 0.8)
  const after = await getSelectedFill()

  expect(after).not.toBeNull()
  expect(
    before?.color.r !== after?.color.r ||
      before?.color.g !== after?.color.g ||
      before?.color.b !== after?.color.b
  ).toBe(true)
})

test('hsb saturation and brightness sliders both affect fill color', async () => {
  await openFillPicker()
  await chooseFormat('HSB')

  const beforeS = await getSelectedFill()
  await dragSlider('color-slider-hsb-s', 0.75)
  const afterS = await getSelectedFill()
  expect(afterS).not.toBeNull()
  expect(
    beforeS?.color.r !== afterS?.color.r ||
      beforeS?.color.g !== afterS?.color.g ||
      beforeS?.color.b !== afterS?.color.b
  ).toBe(true)

  const beforeB = await getSelectedFill()
  await dragSlider('color-slider-hsb-b', 0.9)
  const afterB = await getSelectedFill()
  expect(afterB).not.toBeNull()
  expect(
    beforeB?.color.r !== afterB?.color.r ||
      beforeB?.color.g !== afterB?.color.g ||
      beforeB?.color.b !== afterB?.color.b
  ).toBe(true)
})

test('gradient stops support keyboard nudging and removal', async () => {
  await openFillPicker()
  const gradientTab = page.getByTestId('fill-picker-tab-gradient')
  await gradientTab.click()
  await expect(gradientTab).toHaveAttribute('data-active', 'true')
  await page.getByTestId('fill-picker-add-stop').click()

  const stops = page.getByTestId('fill-picker-gradient-bar').getByRole('slider')
  await expect(stops).toHaveCount(3)
  const first = stops.first()
  await first.focus()
  const before = Number(await first.getAttribute('aria-valuenow'))
  await first.press('ArrowRight')
  await expect(first).toHaveAttribute('aria-valuenow', String(Math.min(100, before + 1)))
  await first.press('Delete')
  await expect(stops).toHaveCount(2)
})

test('okhcl channels preserve intent metadata while updating the fill', async () => {
  await openFillPicker()
  await page.getByTestId('fill-picker-tab-solid').click()
  await chooseFormat('OkHCL')

  await dragSlider('color-slider-okhcl-c', 0.6)
  const afterChroma = await getSelectedFill()
  const chromaIntent = await getSelectedFillOkHCL()
  expect(afterChroma).not.toBeNull()
  expect(chromaIntent?.c).toBeGreaterThan(0)

  await dragSlider('color-slider-okhcl-l', 0.75)
  const lightnessIntent = await getSelectedFillOkHCL()
  expect(lightnessIntent?.l).toBeCloseTo(0.75, 1)
  expect(lightnessIntent?.c).toBeCloseTo(chromaIntent?.c ?? 0, 3)
})