"""Test dashboard routes."""
from __future__ import annotations

import asyncio

from quart import Blueprint, Response, make_response, request


def register(url_prefix: str = "/") -> Blueprint:
    bp = Blueprint("dashboard", __name__, url_prefix=url_prefix)

    @bp.get("/")
    async def index():
        """Full page dashboard with last results."""
        from shared.browser.app.csrf import generate_csrf_token
        from sxc.pages.renders import render_dashboard_page_sx, render_results_partial_sx
        import runner

        result = runner.get_results()
        running = runner.is_running()
        csrf = generate_csrf_token()
        active_filter = request.args.get("filter")
        active_service = request.args.get("service")

        is_sx = bool(request.headers.get("SX-Request") or request.headers.get("HX-Request"))

        if is_sx:
            from shared.sx.helpers import sx_response
            inner = await render_results_partial_sx(
                result, running, csrf,
                active_filter=active_filter,
                active_service=active_service,
            )
            # Wrap in #main-panel so sx-select="#main-panel" works
            sx = (f'(section :id "main-panel" :class "flex-1 md:h-full md:min-h-0'
                  f' overflow-y-auto overscroll-contain js-grid-viewport" {inner})')
            return sx_response(sx)

        from shared.sx.page import get_template_context
        ctx = await get_template_context()
        html = await render_dashboard_page_sx(
            ctx, result, running, csrf,
            active_filter=active_filter,
            active_service=active_service,
        )
        return await make_response(html, 200)

    @bp.post("/run")
    async def run():
        """Trigger a test run, redirect to /."""
        import runner

        if not runner.is_running():
            asyncio.create_task(runner.run_tests())

        # HX-Redirect for HTMX, regular redirect for non-HTMX
        if request.headers.get("SX-Request") or request.headers.get("HX-Request"):
            resp = Response("", status=200)
            resp.headers["HX-Redirect"] = "/"
            return resp

        from quart import redirect as qredirect
        return qredirect("/")

    @bp.get("/test/<path:nodeid>")
    async def test_detail(nodeid: str):
        """Test detail view — full page or sx wire format."""
        import runner

        test = runner.get_test(nodeid)
        if not test:
            from quart import abort
            abort(404)

        is_htmx = bool(request.headers.get("SX-Request") or request.headers.get("HX-Request"))

        if is_htmx:
            # S-expression wire format — sx.js renders client-side
            from shared.sx.helpers import sx_response
            from sxc.pages.renders import test_detail_sx
            return sx_response(await test_detail_sx(test))

        # Full page render (direct navigation / refresh)
        from shared.sx.page import get_template_context
        from sxc.pages.renders import render_test_detail_page_sx

        ctx = await get_template_context()
        html = await render_test_detail_page_sx(ctx, test)
        return await make_response(html, 200)

    @bp.get("/results")
    async def results():
        """HTMX partial — poll target for results table."""
        from shared.browser.app.csrf import generate_csrf_token
        from shared.sx.helpers import sx_response
        from sxc.pages.renders import render_results_partial_sx
        import runner

        result = runner.get_results()
        running = runner.is_running()
        csrf = generate_csrf_token()
        active_filter = request.args.get("filter")
        active_service = request.args.get("service")

        html = await render_results_partial_sx(
            result, running, csrf,
            active_filter=active_filter,
            active_service=active_service,
        )

        headers = {}
        if running:
            headers["HX-Trigger-After-Swap"] = "test-still-running"
        return sx_response(html, headers=headers)

    return bp