"""MQTT router — exposes MQTT entity state and publish endpoint."""

from __future__ import annotations

import logging
from typing import Any, Dict, Optional

from fastapi import APIRouter, HTTPException
from pydantic import BaseModel

from server.services.mqtt_service import mqtt_service

logger = logging.getLogger(__name__)

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


class PublishRequest(BaseModel):
    topic: str
    payload: Any
    retain: bool = False


@router.get("")
async def get_mqtt_state() -> Dict[str, Any]:
    """Return all stored MQTT entities."""
    return {
        "connected": mqtt_service.connected,
        "entities": mqtt_service.get_entities(),
        "topic_count": len(mqtt_service.store),
    }


@router.get("/topic/{topic:path}")
async def get_mqtt_topic(topic: str) -> Dict[str, Any]:
    """Return value for a specific MQTT topic."""
    msg = mqtt_service.store.get(topic)
    if not msg:
        raise HTTPException(status_code=404, detail=f"Topic '{topic}' not found")
    return {
        "topic": msg.topic,
        "value": msg.payload,
        "timestamp": msg.timestamp,
    }


@router.post("/publish")
async def publish_mqtt(req: PublishRequest) -> Dict[str, str]:
    """Publish a message to the MQTT broker."""
    try:
        await mqtt_service.publish(req.topic, req.payload, retain=req.retain)
        return {"status": "ok", "topic": req.topic}
    except RuntimeError as exc:
        raise HTTPException(status_code=503, detail=str(exc))