"""CDEK order registration payload mappers."""

from typing import Any

from app.schemas.payment import (
    InitPaymentRequest,
    PaymentPackage,
    PaymentParty,
)


class CDEKOrderMappingError(ValueError):
    """Raised when CDEK order payload cannot be mapped."""


def map_cdek_order_request(request: InitPaymentRequest) -> dict[str, Any]:
    payload: dict[str, Any] = {
        "type": request.type,
        "tariff_code": request.tariff_code,
        "sender": _map_order_party(request.sender),
        "recipient": _map_order_party(request.recipient),
        "from_location": request.from_location.model_dump(mode="python"),
        "to_location": request.to_location.model_dump(mode="python"),
        "packages": [_map_order_package(package) for package in request.packages],
    }
    if request.comment is not None:
        payload["comment"] = request.comment
    if request.services is not None:
        payload["services"] = [
            service.model_dump(mode="python") for service in request.services
        ]
    return payload


def map_cdek_order_response(payload: dict[str, Any]) -> str:
    entity = payload.get("entity")
    if not isinstance(entity, dict):
        raise CDEKOrderMappingError("CDEK order response must include entity object.")

    order_uuid = entity.get("uuid")
    if not isinstance(order_uuid, str) or not order_uuid:
        raise CDEKOrderMappingError("CDEK order response must include entity.uuid.")

    return order_uuid


def _map_order_party(party: PaymentParty) -> dict[str, Any]:
    return {
        "name": party.name,
        "email": party.email,
        "phones": [party.phone.model_dump(mode="python")],
    }


def _map_order_package(package: PaymentPackage) -> dict[str, Any]:
    payload = package.model_dump(mode="python", exclude_none=True)
    payload["weight"] = package.weight * 1000
    return payload