"""Health check endpoints."""

from fastapi import APIRouter, Response, status

from app.db import db, redis_client

router = APIRouter()


@router.get("/healthz")
async def healthz() -> dict[str, str]:
    """Liveness probe - returns 200 if the service is running."""
    return {"status": "ok"}


@router.get("/readyz")
async def readyz(response: Response) -> dict[str, str | dict[str, bool]]:
    """
    Readiness probe - checks database and Redis connectivity.
    - Check Postgres status
    - Check Redis status
    - Return overall healthiness
    """
    checks = {
        "postgres": False,
        "redis": False,
    }

    try:
        if db.pool:
            async with db.connection() as conn:
                await conn.fetchval("SELECT 1")
                checks["postgres"] = True
    except Exception:
        pass

    checks["redis"] = await redis_client.ping()

    all_healthy = all(checks.values())
    if not all_healthy:
        response.status_code = status.HTTP_503_SERVICE_UNAVAILABLE

    return {
        "status": "ok" if all_healthy else "degraded",
        "checks": checks,
    }