from contextlib import asynccontextmanager

from fastapi import FastAPI

from app.api.v1.entities import router as entities_router
from app.config import load_settings
from app.core.exception_handlers import register_exception_handlers
from app.ha.client import HaClient, HaClientSettings
from app.ha.reader import HaReader
from backend.routes.ml import init_ml_routes


@asynccontextmanager
async def lifespan(app: FastAPI):  # type: ignore[no-untyped-def]
    settings = app.state.settings
    if hasattr(app.state, "ha_reader"):
        del app.state.ha_reader
    if settings.ha_configured:
        client = HaClient(
            settings=HaClientSettings(
                url=settings.ha_url,
                token=settings.ha_token,
            )
        )
        app.state.ha_reader = HaReader(client=client)
    yield


app = FastAPI(
    title="SillyHome Next API",
    description="Lokales Smart-Home-Intelligenzsystem für Home Assistant.",
    version="0.1.0",
    lifespan=lifespan,
)
app.state.settings = load_settings()
register_exception_handlers(app)
app.include_router(entities_router)
init_ml_routes(app, model_store=app.state.settings.model_store)


@app.get("/health")
def health() -> dict[str, str]:
    return {"status": "ok"}


@app.get("/")
def root() -> dict[str, str]:
    return {"service": "sillyhome-next", "docs": "/docs"}