receipt.py

"""Receipt composers.

`build_receipt` is the shared receipt template — every receipt type
goes through it. Specific receipts (`render_receipt`, `render_list`) choose
their banner text, items, totals, footer notes and optional upside-down joke.
"""

import logging
import random
import re

from blocks import (
    Banner, Barcode, Block, Checkbox, Divider, Footer, Item, Spacer,
    Total, Underline, UpsideDown,
)
from printer import print_blocks
from providers import item_providers, joke

logger = logging.getLogger(__name__)

# ── Shared defaults ──────────────────────────────────────────────────────────

DEFAULT_THANKS  = "thank you for your vibes"
DEFAULT_FOOTER  = "COME BACK SOON"
DEFAULT_BARCODE = "creativereceipt"
TAX_PERCENT     = 13

RANDOM_THANKS = [
    "congrats, you made a list",
    "so many tasks",
    "thanks for writing this down",
    "tasks are fake, just vibes",
]

RANDOM_FOOTER = [
    "NOW IGNORE IT",
    "GOOD LUCK LOL",
    "MAYBE TOMORROW",
    "DOUBT EVERYTHING",
    "THAT'S ENOUGH FOR TODAY",
]

TITLE_VARIANTS = [
    "MAYBE DO",
    "TASKS? LOL",
    "PROCRASTINATION LIST",
    "WELL...",
    "IF I FEEL LIKE IT",
]

_NUM_RE = re.compile(r"[-+]?\d+(?:\.\d+)?")


def _extract_number(s: str) -> float | None:
    cleaned = s.replace("\xa0", "").replace(" ", "")
    m = _NUM_RE.search(cleaned)
    return float(m.group()) if m else None


def _fmt_money(n: float) -> str:
    return f"{n:,.2f}".replace(",", " ")


def _fetch_items(module) -> list[Item]:
    try:
        blocks = module.fetch() or []
    except Exception as e:
        logger.warning(f"Provider {module.__name__} failed: {e}")
        return []
    items = [b for b in blocks if isinstance(b, Item)]
    if len(items) != len(blocks):
        logger.warning(
            f"Provider {module.__name__} returned {len(blocks) - len(items)} "
            "non-Item block(s) — providers return Items only; ignored."
        )
    return items


def _joke() -> str | None:
    """Fetch one joke for the upside-down footer. None on failure."""
    try:
        return joke.fetch()
    except Exception as e:
        logger.warning(f"joke fetch failed: {e}")
        return None


# ── Shared template ──────────────────────────────────────────────────────────


def build_receipt(
    *,
    banner_text: str,
    items: list[Block],
    totals: list[Block] | None = None,
    column_header: tuple[str, str] | None = ("ITEM", "VALUE"),
    upside_down: str | None = None,
    thanks: str = DEFAULT_THANKS,
    footer: str = DEFAULT_FOOTER,
    barcode_text: str | None = DEFAULT_BARCODE,
) -> list[Block]:
    """Compose a receipt in the standard order.

    Use `column_header=None` for receipts without an item/value column header
    (e.g. plain checklists). Pass `barcode_text=None` to skip the footer barcode.
    """
    out: list[Block] = []

    out.append(Banner(banner_text))
    out.append(Spacer())

    if items:
        if column_header:
            out.append(Item(column_header[0], column_header[1], bold=True))
        out.append(Divider("dash"))
        out.extend(items)
        out.append(Divider("dash"))

    if totals:
        out.extend(totals)
        out.append(Spacer())

    out.append(Underline(thanks))
    out.append(Spacer())

    if upside_down:
        out.append(UpsideDown(upside_down))
        out.append(Spacer())

    out.append(Footer(footer))

    if barcode_text:
        out.append(Spacer())
        out.append(Barcode(barcode_text))

    out.append(Divider("dash"))

    return out


# ── Default receipt: auto-discovered item providers + joke ──────────────────


def render_receipt() -> list[Block]:
    all_items: list[Item] = []
    for module in item_providers():
        all_items += _fetch_items(module)

    totals: list[Block] = []
    if all_items:
        subtotal = sum(
            n for n in (_extract_number(it.value) for it in all_items) if n is not None
        )
        tax = subtotal * TAX_PERCENT / 100
        grand = subtotal + tax
        totals = [
            Total("SUBTOTAL", _fmt_money(subtotal)),
            Total(f"HST ({TAX_PERCENT}%)", _fmt_money(tax)),
            Total("TOTAL", f"{_fmt_money(grand)} ₽", big=True),
        ]

    return build_receipt(
        banner_text="TODAY ONLY: 0% OFF EVERYTHING",
        items=all_items,
        totals=totals,
        upside_down=_joke(),
    )


def print_receipt() -> None:
    print_blocks(render_receipt())


# ── Shopping / to-do list receipt: checkboxes only ──────────────────────────


def render_list(items: list[str]) -> list[Block]:
    """A checklist receipt: '[ ] item' rows with a count total."""
    if not items:
        return []
    return build_receipt(
        banner_text=random.choice(TITLE_VARIANTS),
        items=[Checkbox(t) for t in items],
        totals=[Total("ITEMS", str(len(items)), big=True)],
        column_header=None,
        upside_down=_joke(),
        thanks=random.choice(RANDOM_THANKS),
        footer=random.choice(RANDOM_FOOTER),
    )


def print_list(items: list[str]) -> None:
    blocks = render_list(items)
    if blocks:
        print_blocks(blocks)