from fastapi import FastAPI, HTTPException, status
from fastapi.responses import HTMLResponse
from pydantic import BaseModel, Field
from scalar_fastapi import get_scalar_api_reference

app: FastAPI = FastAPI()


class ShipmentId(BaseModel):
    shipment_id: int = Field(strict=True, description="Id of the shipment.")


class Shipment(BaseModel):
    content: str = Field(strict=True, max_length=256, description="Content of the shipment")
    weight_in_grams: int = Field(strict=True, ge=1, le=10_000, description="Weight in grams")
    destination: str = Field(strict=True, max_length=64, description="Shipment Destination country")


shipments: dict[int, dict[str, str | int]] = {
    1: {"content": "Soap", "destination": "USA", "weight_in_grams": 200},
    2: {"content": "Eggs", "destination": "India", "weight_in_grams": 219},
}


@app.get(path="/shipment/get")
def get_shipment(shipment_id: int) -> Shipment:
    """."""
    shipment_details: dict[str, str | int] | None = shipments.get(shipment_id)

    if shipment_details is None:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail="shipment id does not exist",
        )

    return Shipment.model_validate(obj=shipment_details)


@app.post(path="/shipments/create", response_model=ShipmentId)
def create_shipment(shipment: Shipment) -> ShipmentId:
    new_id: int = max(shipments.keys(), default=0) + 1

    shipments[new_id] = shipment.model_dump()

    return ShipmentId(shipment_id=new_id)


@app.get(path="/scalar-ui", include_in_schema=False)
async def get_scalar_ui() -> HTMLResponse:
    """."""
    return get_scalar_api_reference(
        openapi_url=app.openapi_url,
        title="Test Fastapi App",
    )