"""Account app fragment endpoints.

Exposes HTML fragments at ``/internal/fragments/<type>`` for consumption
by other coop apps via the fragment client.

Fragments:
  auth-menu   Desktop + mobile auth menu (sign-in or user link)
"""

from __future__ import annotations

from quart import Blueprint, Response, request, render_template

from shared.infrastructure.fragments import FRAGMENT_HEADER


def register():
    bp = Blueprint("fragments", __name__, url_prefix="/internal/fragments")

    # ---------------------------------------------------------------
    # Fragment handlers
    # ---------------------------------------------------------------

    async def _auth_menu():
        user_email = request.args.get("email", "")
        return await render_template(
            "fragments/auth_menu.html",
            user_email=user_email,
        )

    _handlers = {
        "auth-menu": _auth_menu,
    }

    # ---------------------------------------------------------------
    # Routing
    # ---------------------------------------------------------------

    @bp.before_request
    async def _require_fragment_header():
        if not request.headers.get(FRAGMENT_HEADER):
            return Response("", status=403)

    @bp.get("/<fragment_type>")
    async def get_fragment(fragment_type: str):
        handler = _handlers.get(fragment_type)
        if handler is None:
            return Response("", status=200, content_type="text/html")
        html = await handler()
        return Response(html, status=200, content_type="text/html")

    return bp