import json import logging from fastapi import APIRouter, Depends, Request, status from fastapi.responses import PlainTextResponse, Response from pydantic import ValidationError from sqlalchemy.orm import Session from app.config import Settings from app.dependencies import get_app_settings, get_homeassistant_client, get_poo_db from app.integrations.homeassistant import HomeAssistantClient from app.schemas.poo import PooRecordRequest from app.services.poo import publish_latest_poo_status, record_poo router = APIRouter(tags=["poo"]) logger = logging.getLogger(__name__) BAD_REQUEST_MESSAGE = "bad request" INTERNAL_SERVER_ERROR_MESSAGE = "internal server error" @router.post("/poo/record") async def create_poo_record( request: Request, db: Session = Depends(get_poo_db), settings: Settings = Depends(get_app_settings), homeassistant_client: HomeAssistantClient = Depends(get_homeassistant_client), ) -> Response: try: raw_payload = await request.body() data = json.loads(raw_payload) payload = PooRecordRequest.model_validate(data) record_poo( db, payload, settings=settings, homeassistant_client=homeassistant_client, ) except json.JSONDecodeError as exc: logger.warning("Rejected poo record request due to invalid JSON: %s", exc) return PlainTextResponse(BAD_REQUEST_MESSAGE, status_code=status.HTTP_400_BAD_REQUEST) except ValidationError as exc: logger.warning("Rejected poo record request due to validation failure: %s", exc) return PlainTextResponse(BAD_REQUEST_MESSAGE, status_code=status.HTTP_400_BAD_REQUEST) except ValueError as exc: logger.warning("Rejected poo record request due to invalid numeric input: %s", exc) return PlainTextResponse(BAD_REQUEST_MESSAGE, status_code=status.HTTP_400_BAD_REQUEST) except Exception as exc: logger.warning("Failed to store poo record: %s", exc) return PlainTextResponse( INTERNAL_SERVER_ERROR_MESSAGE, status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, ) return Response(status_code=status.HTTP_200_OK) @router.get("/poo/latest") def notify_latest_poo( db: Session = Depends(get_poo_db), settings: Settings = Depends(get_app_settings), homeassistant_client: HomeAssistantClient = Depends(get_homeassistant_client), ) -> Response: try: publish_latest_poo_status( session=db, settings=settings, homeassistant_client=homeassistant_client, ) except Exception as exc: logger.warning("Failed to publish latest poo status: %s", exc) return PlainTextResponse( INTERNAL_SERVER_ERROR_MESSAGE, status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, ) return Response(status_code=status.HTTP_200_OK)