"""The heat map and the weld map of one drawing page, as a JPEG at the sheet's own size (the old renderMapCanvas)."""

from __future__ import annotations

from collections.abc import Mapping
from pathlib import Path
from typing import Any

from ..domain import assets
from . import raster
from .style import fmt_d, site_tz


def asset_file(conn: Any, settings: Any, asset_id: Any) -> tuple[dict[str, Any], Path]:
    """The assets row and its stored file. Raises when the row or the file is missing."""
    row = assets.get(conn, asset_id) if asset_id else None
    if row is None:
        raise FileNotFoundError("The file is not in the asset store.")
    return row, assets.open_blob(settings, row)


def page_file(conn: Any, settings: Any, page: Mapping[str, Any]) -> tuple[Path, int]:
    """The stored PDF of a packet page and the page number inside it."""
    _row, path = asset_file(conn, settings, page.get("asset_id"))
    return path, int(page.get("pn") or 1)


def void_text(page: Mapping[str, Any], tz: str | None = None) -> str:
    """VOID - superseded by CO-2 (dwg Rev B) 10/16/2026. Empty for a live page (the old voidText)."""
    v = page.get("void")
    if not v:
        return ""
    co = f"CO-{v['co_no']}" if v.get("co_no") is not None else "a change order"
    new = v.get("new")
    sheet = f" ({new.get('dwg') or 'new sheet'} Rev {new.get('rev') or '?'})" if new else ""
    day = fmt_d(v.get("at"), tz, empty="")
    return f"VOID - superseded by {co}{sheet}" + (f" {day}" if day else "")


def heat_placement(packet: Mapping[str, Any], page_n: int) -> Mapping[str, Any] | None:
    """The heat table placed on a page, or None."""
    for h in packet.get("heat_table") or []:
        if h.get("page") == page_n and h.get("x") is not None:
            return h
    return None


def legacy_tags(packet: Mapping[str, Any], page_n: int) -> list[Mapping[str, Any]]:
    """The older heat tags of a page. They show only while no table is placed on it."""
    if heat_placement(packet, page_n) is not None:
        return []
    return [t for t in packet.get("heat_tags") or [] if t.get("page") == page_n]


def page_welds(packet: Mapping[str, Any], page_n: int) -> list[Mapping[str, Any]]:
    return [w for w in packet.get("welds") or [] if w.get("page") == page_n]


def _render(conn: Any, settings: Any, page: Mapping[str, Any]) -> tuple[Any, float, float]:
    path, pn = page_file(conn, settings, page)
    return raster.render_page(path, pn)


def heat_map_page(
    conn: Any, settings: Any, packet: Mapping[str, Any], page: Mapping[str, Any]
) -> tuple[bytes, float, float]:
    """The sheet with its heat table (or its older heat tags when no table is placed), VOID when the page is void.
    Returns (jpeg bytes, width pt, height pt)."""
    img, w, h = _render(conn, settings, page)
    n = int(page["n"])
    bom = packet.get("bom") or []
    placement = heat_placement(packet, n)
    if placement is not None:
        raster.draw_heat_table(img, placement, raster.heat_rows(bom))
    else:
        raster.draw_heat_tags(img, packet.get("heat_tags") or [], bom, n)
    if page.get("void"):
        img = raster.stamp_void(img, void_text(page, site_tz(settings)))
    return raster.to_jpeg(img), w, h


def weld_map_page(
    conn: Any, settings: Any, packet: Mapping[str, Any], page: Mapping[str, Any]
) -> tuple[bytes, float, float]:
    """The sheet with its weld bubbles, VOID when the page is void. Returns (jpeg bytes, width pt, height pt)."""
    img, w, h = _render(conn, settings, page)
    raster.draw_weld_marks(img, packet.get("welds") or [], int(page["n"]))
    if page.get("void"):
        img = raster.stamp_void(img, void_text(page, site_tz(settings)))
    return raster.to_jpeg(img), w, h


def clean_page(conn: Any, settings: Any, page: Mapping[str, Any]) -> tuple[bytes, float, float]:
    """The sheet with no marks (VOID when void), as a raster. Prints copy the vector page instead; this is for a
    page whose PDF cannot be copied."""
    img, w, h = _render(conn, settings, page)
    if page.get("void"):
        img = raster.stamp_void(img, void_text(page, site_tz(settings)))
    return raster.to_jpeg(img), w, h