"""Toolbar controls follow a predictable writing workflow."""

import json
import subprocess
from pathlib import Path


ROOT = Path(__file__).resolve().parents[1]
DOC_JS = (ROOT / "static/js/document.js").read_text(encoding="utf-8")


def test_toolbar_groups_define_writing_actions_before_view_controls():
    ordering = DOC_JS.split("const _DOCUMENT_TOOLBAR_GROUPS", 1)[1].split(
        "function _orderDocumentToolbar", 1
    )[0]

    assert ordering.index("name: 'type'") < ordering.index("name: 'inline-basic'")
    assert ordering.index("name: 'inline-basic'") < ordering.index("name: 'inline-color'")
    assert ordering.index("name: 'inline-color'") < ordering.index("name: 'alignment'")
    assert ordering.index("name: 'alignment'") < ordering.index("name: 'spacing'")
    assert ordering.index("name: 'spacing'") < ordering.index("name: 'link'")
    assert ordering.index("name: 'link'") < ordering.index("name: 'paragraph'")
    assert ordering.index("name: 'paragraph'") < ordering.index("name: 'insert'")
    assert ordering.index("name: 'insert'") < ordering.index("name: 'document'")
    assert ordering.index("name: 'document'") < ordering.index("name: 'view'")
    assert ordering.index("'[data-dd=\"heading\"]'") < ordering.index("'[data-md=\"bold\"]'")
    assert ordering.index("'#doc-fontsize-btn'") < ordering.index("'[data-dd=\"heading\"]'")
    assert ordering.index("'#doc-fontsize-btn'") < ordering.index("'#doc-find-toolbar-btn'")
    assert "_orderDocumentToolbar(itemsWrap);" in DOC_JS
    assert 'title="Editor display size" aria-label="Editor display size"' in DOC_JS


def test_rich_toolbar_rendered_order_is_stable_on_desktop_and_mobile():
    script = r"""
      import { chromium } from 'playwright';
      const browser = await chromium.launch({ headless: true });

      async function inspect(viewport, suffix) {
        const page = await browser.newPage({ viewport });
        await page.goto('http://127.0.0.1:7011/static/js/documentStats.js');
        await page.setContent('<link rel="stylesheet" href="/static/style.css?v=20260831richtexttools91"><div id="toast"></div><div id="chat-container"></div><div id="sidebar"></div>');
        await page.evaluate(async suffix => {
          const mod = await import(`/static/js/document.js?v=20260831richtexttools91&toolbar-order=${suffix}`);
          mod.init('/api');
          mod.injectFreshDoc({
            id: `toolbar-order-${suffix}`,
            title: 'Toolbar order',
            language: 'richtext',
            current_content: '<p>Writing tools</p>',
            version_count: 1,
          });
          await new Promise(resolve => setTimeout(resolve, 450));
        }, suffix);

        const state = await page.evaluate(() => {
          const toolbar = document.querySelector('#md-toolbar-items');
          const controls = Array.from(toolbar.querySelectorAll(':scope > [data-toolbar-group]'));
          const key = item => item.dataset.dd || item.dataset.md || item.id;
          const visible = controls.filter(item => {
            const style = getComputedStyle(item);
            return style.display !== 'none' && style.visibility !== 'hidden';
          });
          return {
            all: controls.map(item => [item.dataset.toolbarGroup, key(item)]),
            visible: visible.map(key),
            separators: Array.from(toolbar.querySelectorAll(':scope > [data-toolbar-separator]'))
              .map(item => item.dataset.toolbarSeparator),
            toolbarOverflow: toolbar.scrollWidth > toolbar.clientWidth,
            pageOverflow: document.documentElement.scrollWidth - document.documentElement.clientWidth,
          };
        });
        await page.close();
        return state;
      }

      const desktop = await inspect({ width: 900, height: 700 }, 'desktop');
      const mobile = await inspect({ width: 390, height: 844 }, 'mobile');
      console.log(JSON.stringify({ desktop, mobile }));
      await browser.close();
    """
    result = subprocess.run(
        ["node", "--input-type=module", "-e", script],
        cwd=ROOT,
        check=False,
        capture_output=True,
        text=True,
    )
    assert result.returncode == 0, result.stderr
    data = json.loads(result.stdout)

    expected_groups = [
        "display-size", "type", "inline-basic", "inline-color", "alignment", "spacing",
        "link", "paragraph", "insert", "inline-rich", "document", "view",
    ]
    expected_separators = [
        "display-size-type",
        "type-inline-basic",
        "inline-basic-inline-color",
        "inline-color-alignment",
        "alignment-spacing",
        "spacing-link",
        "link-paragraph",
        "paragraph-insert",
        "insert-inline-rich",
        "inline-rich-document",
        "document-view",
    ]
    for state in data.values():
        groups = [group for group, _ in state["all"]]
        assert list(dict.fromkeys(groups)) == expected_groups
        assert state["separators"] == expected_separators
        assert state["visible"][:6] == [
            "doc-fontsize-btn",
            "heading",
            "font",
            "textsize",
            "bold",
            "italic",
        ]
        assert state["visible"].index("link") < state["visible"].index("list")
        assert state["visible"].index("list") < state["visible"].index("md-toolbar-attach-btn")
        assert state["visible"].index("md-toolbar-attach-btn") < state["visible"].index("doc-find-toolbar-btn")
        assert state["visible"].index("subscript") > state["visible"].index("md-toolbar-attach-btn")
        if "doc-outline-toolbar-btn" in state["visible"]:
            assert state["visible"].index("doc-fontsize-btn") < state["visible"].index("doc-outline-toolbar-btn")
        assert state["pageOverflow"] == 0

    assert data["mobile"]["toolbarOverflow"] is True