"""Order lookup tools -- read-only operations."""

from __future__ import annotations

from types import MappingProxyType

from langchain_core.tools import tool

MOCK_ORDERS: MappingProxyType[str, dict] = MappingProxyType(
    {
        "1042": {
            "order_id": "1042",
            "status": "shipped",
            "items": ["Wireless Headphones", "USB-C Cable"],
            "total": 89.99,
            "placed_at": "2026-03-25",
        },
        "1043": {
            "order_id": "1043",
            "status": "processing",
            "items": ["Laptop Stand"],
            "total": 49.99,
            "placed_at": "2026-03-28",
        },
        "1044": {
            "order_id": "1044",
            "status": "delivered",
            "items": ["Mechanical Keyboard", "Mouse Pad"],
            "total": 159.99,
            "placed_at": "2026-03-20",
        },
    }
)

MOCK_TRACKING: MappingProxyType[str, dict] = MappingProxyType(
    {
        "1042": {
            "order_id": "1042",
            "carrier": "FedEx",
            "tracking_number": "FX-9876543210",
            "estimated_delivery": "2026-04-01",
            "current_location": "Distribution Center, Chicago IL",
        },
        "1044": {
            "order_id": "1044",
            "carrier": "UPS",
            "tracking_number": "1Z-5678901234",
            "estimated_delivery": "2026-03-22",
            "current_location": "Delivered",
        },
    }
)


@tool
def get_order_status(order_id: str) -> dict:
    """Look up the current status of an order by order ID."""
    if order_id in MOCK_ORDERS:
        return dict(MOCK_ORDERS[order_id])
    return {"error": f"Order {order_id} not found", "order_id": order_id}


@tool
def get_tracking_info(order_id: str) -> dict:
    """Get shipping and tracking information for an order."""
    if order_id in MOCK_TRACKING:
        return dict(MOCK_TRACKING[order_id])
    return {"error": f"No tracking information for order {order_id}", "order_id": order_id}