from __future__ import annotations

from decimal import Decimal

from shared.infrastructure.data_client import fetch_data
from shared.contracts.dtos import CalendarEntryDTO, TicketDTO, dto_from_dict
from .identity import current_cart_identity


async def get_calendar_cart_entries(session):
    """
    Return all *pending* calendar entries (as CalendarEntryDTOs) for the
    current cart identity (user or anonymous session).
    """
    ident = current_cart_identity()
    params = {}
    if ident["user_id"] is not None:
        params["user_id"] = ident["user_id"]
    if ident["session_id"] is not None:
        params["session_id"] = ident["session_id"]
    raw = await fetch_data("events", "pending-entries", params=params, required=False) or []
    return [dto_from_dict(CalendarEntryDTO, e) for e in raw]


def calendar_total(entries) -> Decimal:
    """
    Total cost of pending calendar entries.
    """
    return sum(
        (Decimal(str(e.cost)) if e.cost else Decimal(0))
        for e in entries
        if e.cost is not None
    )


async def get_ticket_cart_entries(session):
    """Return all reserved tickets (as TicketDTOs) for the current identity."""
    ident = current_cart_identity()
    params = {}
    if ident["user_id"] is not None:
        params["user_id"] = ident["user_id"]
    if ident["session_id"] is not None:
        params["session_id"] = ident["session_id"]
    raw = await fetch_data("events", "pending-tickets", params=params, required=False) or []
    return [dto_from_dict(TicketDTO, t) for t in raw]


def ticket_total(tickets) -> Decimal:
    """Total cost of reserved tickets."""
    return sum((Decimal(str(t.price)) if t.price else Decimal(0) for t in tickets), Decimal(0))