"""Market app fragment endpoints.

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

from __future__ import annotations

from quart import Blueprint, Response, g, render_template, request

from shared.infrastructure.fragments import FRAGMENT_HEADER
from shared.services.registry import services


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

    _handlers: dict[str, object] = {}

    @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")

    # --- container-nav fragment: market links --------------------------------

    async def _container_nav_handler():
        container_type = request.args.get("container_type", "page")
        container_id = int(request.args.get("container_id", 0))
        post_slug = request.args.get("post_slug", "")

        markets = await services.market.marketplaces_for_container(
            g.s, container_type, container_id,
        )
        if not markets:
            return ""
        return await render_template(
            "fragments/container_nav_markets.html",
            markets=markets, post_slug=post_slug,
        )

    _handlers["container-nav"] = _container_nav_handler

    # --- link-card fragment: product preview card --------------------------------

    async def _link_card_handler():
        from sqlalchemy import select
        from shared.models.market import Product
        from shared.infrastructure.urls import market_url

        slug = request.args.get("slug", "")
        keys_raw = request.args.get("keys", "")

        # Batch mode
        if keys_raw:
            slugs = [k.strip() for k in keys_raw.split(",") if k.strip()]
            parts = []
            for s in slugs:
                parts.append(f"<!-- fragment:{s} -->")
                product = (
                    await g.s.execute(select(Product).where(Product.slug == s))
                ).scalar_one_or_none()
                if product:
                    parts.append(await render_template(
                        "fragments/link_card.html",
                        title=product.title,
                        image=product.image,
                        description_short=product.description_short,
                        brand=product.brand,
                        regular_price=product.regular_price,
                        special_price=product.special_price,
                        link=market_url(f"/product/{product.slug}/"),
                    ))
            return "\n".join(parts)

        # Single mode
        if not slug:
            return ""
        product = (
            await g.s.execute(select(Product).where(Product.slug == slug))
        ).scalar_one_or_none()
        if not product:
            return ""
        return await render_template(
            "fragments/link_card.html",
            title=product.title,
            image=product.image,
            description_short=product.description_short,
            brand=product.brand,
            regular_price=product.regular_price,
            special_price=product.special_price,
            link=market_url(f"/product/{product.slug}/"),
        )

    _handlers["link-card"] = _link_card_handler

    bp._fragment_handlers = _handlers

    return bp