"""Expert trajectories: the ground truth replayed as the actions that build it.

Every Crello template decomposes into one `add_*` call per layer, back to
front, with exact arguments. Replaying that sequence reconstructs the ground
truth, and scores 1.0, which makes it both the SFT corpus and the strongest
test the environment has: if the expert does not score 1.0, the reward or the
tools are wrong.

The same trajectory is emitted for the HTML surface as a single page.
"""

from __future__ import annotations

from .doc import DesignDoc, Element
from .html_profile import doc_to_html


def _round(v: float) -> float:
    r = round(float(v), 2)
    return int(r) if r == int(r) else r


def element_call(el: Element) -> tuple[str, dict]:
    box = {"x": _round(el.x), "y": _round(el.y)}
    extra = {}
    if el.angle:
        extra["angle"] = _round(el.angle)
    if el.opacity < 1:
        extra["opacity"] = _round(el.opacity)
    if el.kind == "image":
        return "add_image", {
            "asset_id": el.asset_id, **box, "width": _round(el.width),
            "height": _round(el.height), **extra,
        }
    if el.kind == "shape":
        args = {"shape": el.shape, **box, "width": _round(el.width), "height": _round(el.height),
                "fill": el.fill or "none", **extra}
        if el.stroke and el.stroke_width:
            args.update(stroke=el.stroke, stroke_width=_round(el.stroke_width))
        if el.radius:
            args["radius"] = _round(el.radius)
        return "add_shape", args
    args = {"text": el.text, **box}
    if not el.auto_width:
        args["width"] = _round(el.width)
    args.update(font_family=el.font_family, font_size=_round(el.font_size),
                color=el.color or "#000000")
    if el.align != "left":
        args["align"] = el.align
    for key in ("bold", "italic", "underline", "uppercase"):
        if getattr(el, key):
            args[key] = True
    if el.line_height != 1.0:
        args["line_height"] = _round(el.line_height)
    if el.letter_spacing:
        args["letter_spacing"] = _round(el.letter_spacing)
    if el.runs:
        args["runs"] = [r.model_dump(exclude_none=True) for r in el.runs]
    args.update(extra)
    return "add_text", args


def tool_calls(doc: DesignDoc) -> list[dict]:
    calls = []
    if doc.background.lower() != "#ffffff":
        calls.append({"name": "set_background", "arguments": {"color": doc.background}})
    for el in doc.elements:
        if not el.visible:
            continue
        name, args = element_call(el)
        calls.append({"name": name, "arguments": args})
    calls.append({"name": "submit", "arguments": {}})
    return calls


def html_page(doc: DesignDoc) -> str:
    return doc_to_html(doc)