import { toolbarButtons } from '@affine-test/kit/bs/linked-toolbar';
import { test } from '@affine-test/kit/playwright';
import {
  pressArrowUp,
  pressEnter,
  pressTab,
} from '@affine-test/kit/utils/keyboard';
import { openHomePage } from '@affine-test/kit/utils/load-page';
import {
  clickNewPageButton,
  createLinkedPage,
  type,
  waitForEditorLoad,
} from '@affine-test/kit/utils/page-logic';
import { expect } from '@playwright/test';

test.beforeEach(async ({ page }) => {
  await openHomePage(page);
  await clickNewPageButton(page, 'List Test');
  await waitForEditorLoad(page);
});

test.describe('split list', () => {
  test('split at the start of the list', async ({ page }) => {
    await pressEnter(page);
    await type(page, '1. aaa\nbbb\nccc\nddd');
    await pressArrowUp(page, 2);
    await pressTab(page);
    await pressArrowUp(page, 2);

    const listLocator = page.locator('affine-list');

    /**
     * 1. |aaa
     *   a. bbb
     * 2. ccc
     * 3. ddd
     */
    await expect(listLocator.nth(0).locator('rich-text')).toHaveText([
      'aaa',
      'bbb',
    ]);
    await expect(
      listLocator.nth(0).locator('.affine-list-block__numbered')
    ).toHaveText(['1.', 'a.']);
    await expect(listLocator.nth(2).locator('rich-text')).toHaveText('ccc');
    await expect(
      listLocator.nth(2).locator('.affine-list-block__numbered')
    ).toHaveText('2.');
    await expect(listLocator.nth(3).locator('rich-text')).toHaveText('ddd');
    await expect(
      listLocator.nth(3).locator('.affine-list-block__numbered')
    ).toHaveText('3.');

    await pressEnter(page);
    /**
     * 1.
     * 2. aaa
     *   a. bbb
     * 3. ccc
     * 4. ddd
     */
    await expect(
      listLocator.nth(0).locator('.affine-list-block__numbered')
    ).toHaveText('1.');
    await expect(listLocator.nth(1).locator('rich-text')).toHaveText([
      'aaa',
      'bbb',
    ]);
    await expect(
      listLocator.nth(1).locator('.affine-list-block__numbered')
    ).toHaveText(['2.', 'a.']);
    await expect(listLocator.nth(3).locator('rich-text')).toHaveText('ccc');
    await expect(
      listLocator.nth(3).locator('.affine-list-block__numbered')
    ).toHaveText('3.');
    await expect(listLocator.nth(4).locator('rich-text')).toHaveText('ddd');
    await expect(
      listLocator.nth(4).locator('.affine-list-block__numbered')
    ).toHaveText('4.');
  });

  test('tab in list should not propagate out of editor', async ({ page }) => {
    await pressEnter(page);
    await type(page, '1. aaa');
    await page.keyboard.press('Enter');
    await page.keyboard.press('Tab');
    await createLinkedPage(page, 'Test Page');
    const inlineLink = page.locator('affine-reference');
    const { switchViewBtn, cardViewBtn } = toolbarButtons(page);
    const list = page.locator('affine-list');
    const card = page.locator('affine-embed-linked-doc-block');

    const pressWithCount = async (key: string, count: number) => {
      for (let i = 0; i < count; i++) {
        await new Promise(resolve => setTimeout(resolve, 5));
        await page.keyboard.press(key);
      }
    };

    await inlineLink.hover();
    await switchViewBtn.click();
    await cardViewBtn.click();

    await expect(list.filter({ has: card })).toHaveCount(1);

    await page.keyboard.press('Shift+Tab');

    await expect(list.filter({ hasNot: card })).toHaveCount(1);

    await pressWithCount('Tab', 5);

    await expect(list.filter({ has: card })).toHaveCount(1);

    await pressWithCount('Shift+Tab', 5);

    await expect(list.filter({ hasNot: card })).toHaveCount(1);
  });
});