#!/usr/bin/env python3
"""Capture workflow walkthrough media from a live TranscriptX UI."""

from __future__ import annotations

import re
import subprocess
from pathlib import Path

from playwright.sync_api import sync_playwright

from tests.e2e_gui.helpers import (
    DEFAULT_VIEWPORT,
    click_section_tab,
    goto_app,
    jump_to_transcript_if_present,
    nav,
    select_transcript,
    wait,
)

BASE = "http://127.0.0.1:8501"
OUT = Path("/workspace/.local/workflow_media/raw")
FINAL = Path("/workspace/docs/_static/workflows")
VIEWPORT = DEFAULT_VIEWPORT


def shot(page, name: str) -> Path:
    OUT.mkdir(parents=True, exist_ok=True)
    if not name.endswith(".png"):
        name = f"{name}.png"
    path = OUT / name
    page.screenshot(path=str(path), full_page=False)
    print("wrote", path, path.stat().st_size)
    return path


def make_gif(pattern: str, out_name: str) -> None:
    frames = sorted(OUT.glob(pattern))
    if len(frames) < 2:
        print("skip gif", out_name, "frames", len(frames))
        return
    tmp = OUT / out_name
    subprocess.run(
        ["convert", "-delay", "50", "-loop", "0", *[str(f) for f in frames], str(tmp)],
        check=False,
    )
    if not tmp.exists():
        return
    dest = FINAL / out_name
    subprocess.run(
        ["gifsicle", "-O3", "--colors", "128", "-o", str(dest), str(tmp)],
        check=False,
    )
    if not dest.exists():
        dest.write_bytes(tmp.read_bytes())
    print("gif", dest, dest.stat().st_size)


def promote_png(name: str) -> None:
    src = OUT / name
    if not src.exists():
        print("missing", name)
        return
    dest = FINAL / name
    # compress copy
    subprocess.run(
        ["pngquant", "--quality=65-85", "--force", "--output", str(dest), str(src)],
        check=False,
    )
    if not dest.exists():
        dest.write_bytes(src.read_bytes())
    subprocess.run(["optipng", "-o2", str(dest)], check=False)
    print("png", dest, dest.stat().st_size)


def main() -> None:
    OUT.mkdir(parents=True, exist_ok=True)
    FINAL.mkdir(parents=True, exist_ok=True)
    (OUT / "gif_frames_run").mkdir(exist_ok=True)
    (OUT / "gif_frames_speaker").mkdir(exist_ok=True)
    (OUT / "gif_frames_jump").mkdir(exist_ok=True)
    (OUT / "gif_frames_export").mkdir(exist_ok=True)

    with sync_playwright() as p:
        browser = p.chromium.launch(headless=True)
        page = browser.new_page(viewport=VIEWPORT, device_scale_factor=1)
        goto_app(page, BASE)

        # Import page
        nav(page, "Import Transcript")
        shot(page, "first-analysis-import.png")

        # Select transcript for subsequent pages
        select_transcript(page)

        # Run Analysis
        nav(page, "Run Analysis")
        shot(page, "first-analysis-run-analysis.png")
        shot(page, "local-ai-llm-setup.png")
        for i in range(5):
            page.screenshot(path=str(OUT / f"gif_frames_run/frame_{i:02d}.png"))
            page.wait_for_timeout(350)

        # Overview
        nav(page, "Overview")
        wait(page, 2000)
        shot(page, "first-analysis-overview.png")
        shot(page, "investigate-overview.png")
        shot(page, "local-ai-overview-summary.png")

        # Speaker Identification (prefer CCv2 Next control)
        nav(page, "Speaker Identification")
        shot(page, "speaker-identification-page.png")
        for i in range(5):
            page.screenshot(path=str(OUT / f"gif_frames_speaker/frame_{i:02d}.png"))
            nxt = page.locator(".tx-sid-next")
            if nxt.count() == 0:
                nxt = page.get_by_role("button", name="Next", exact=True)
            if nxt.count() == 0:
                nxt = page.get_by_role("button", name=re.compile(r"(^|\s)Next$"))
            # Icon-only CCv2 control keeps accessible name "Next" via aria-label.
            if nxt.count():
                try:
                    nxt.first.click(force=True)
                    wait(page, 900)
                except Exception:
                    page.wait_for_timeout(400)
            else:
                page.wait_for_timeout(400)

        # Transcript
        nav(page, "Transcript")
        shot(page, "speaker-identification-transcript.png")

        # Insights
        nav(page, "Insights")
        for label in ("Highlights", "Actions", "Summary"):
            click_section_tab(page, label)
            if label == "Highlights":
                shot(page, "investigate-highlights.png")
                for i in range(4):
                    page.screenshot(path=str(OUT / f"gif_frames_jump/frame_{i:02d}.png"))
                    if jump_to_transcript_if_present(page):
                        page.screenshot(
                            path=str(OUT / f"gif_frames_jump/frame_{i:02d}_after.png")
                        )
                        break
                    page.wait_for_timeout(400)
            if label == "Actions":
                shot(page, "local-ai-meeting-extracts.png")

        # Artifacts
        nav(page, "Artifacts")
        for label in ("Browse", "Export"):
            click_section_tab(page, label)
            if label == "Browse":
                shot(page, "export-artifacts-browse.png")
            if label == "Export":
                shot(page, "export-panel.png")
                for i in range(4):
                    page.screenshot(path=str(OUT / f"gif_frames_export/frame_{i:02d}.png"))
                    create = page.get_by_role("button", name="Create Export")
                    if create.count():
                        try:
                            create.first.click(force=True)
                            wait(page, 2500)
                        except Exception:
                            pass
                    page.wait_for_timeout(400)

        browser.close()

    make_gif("gif_frames_run/frame_*.png", "first-analysis-run-complete.gif")
    make_gif("gif_frames_speaker/frame_*.png", "speaker-identification-naming.gif")
    make_gif("gif_frames_jump/frame_*.png", "investigate-evidence-jump.gif")
    make_gif("gif_frames_export/frame_*.png", "export-download.gif")

    for name in [
        "first-analysis-import.png",
        "first-analysis-run-analysis.png",
        "first-analysis-overview.png",
        "speaker-identification-page.png",
        "speaker-identification-transcript.png",
        "investigate-overview.png",
        "investigate-highlights.png",
        "local-ai-llm-setup.png",
        "local-ai-overview-summary.png",
        "local-ai-meeting-extracts.png",
        "export-artifacts-browse.png",
        "export-panel.png",
    ]:
        promote_png(name)

    print("done")


if __name__ == "__main__":
    main()