daily-briefing/server/routers/homeassistant.py

"""Home Assistant data router."""

from __future__ import annotations

import logging
from typing import Any, Dict

from fastapi import APIRouter

from server.cache import cache
from server.config import get_settings
from server.services.ha_service import fetch_ha_data

logger = logging.getLogger(__name__)

router = APIRouter(prefix="/api", tags=["homeassistant"])

CACHE_KEY = "ha"


@router.get("/ha")
async def get_ha() -> Dict[str, Any]:
    """Return Home Assistant entity data.

    The exact shape depends on what ``fetch_ha_data`` returns; on failure an
    error stub is returned instead::

        { "error": true, "message": "..." }
    """

    # --- cache hit? -----------------------------------------------------------
    cached = await cache.get(CACHE_KEY)
    if cached is not None:
        return cached

    # --- cache miss -----------------------------------------------------------
    try:
        data: Dict[str, Any] = await fetch_ha_data(
            get_settings().ha_url,
            get_settings().ha_token,
        )
    except Exception as exc:
        logger.exception("Failed to fetch Home Assistant data")
        return {"error": True, "message": str(exc)}

    await cache.set(CACHE_KEY, data, get_settings().ha_cache_ttl)
    return data