import asyncio
from contextlib import asynccontextmanager, suppress

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

from app.api.routes import api_router
from app.config import get_settings
from app.db.base import init_db
from app.fitness.watcher import fitness_watcher_loop
from app.homelab.watcher import homelab_watcher_loop
from app.pomodoro.watcher import pomodoro_watcher_loop
from app.reminders.watcher import reminders_watcher_loop


@asynccontextmanager
async def lifespan(_: FastAPI):
    init_db()
    pomodoro_task = asyncio.create_task(pomodoro_watcher_loop())
    fitness_task = asyncio.create_task(fitness_watcher_loop())
    homelab_task = asyncio.create_task(homelab_watcher_loop())
    reminders_task = asyncio.create_task(reminders_watcher_loop())
    yield
    pomodoro_task.cancel()
    fitness_task.cancel()
    homelab_task.cancel()
    reminders_task.cancel()
    with suppress(asyncio.CancelledError):
        await pomodoro_task
    with suppress(asyncio.CancelledError):
        await fitness_task
    with suppress(asyncio.CancelledError):
        await homelab_task
    with suppress(asyncio.CancelledError):
        await reminders_task


def create_app() -> FastAPI:
    settings = get_settings()
    app = FastAPI(title="Home AI Assistant", lifespan=lifespan)

    app.add_middleware(
        CORSMiddleware,
        allow_origins=settings.cors_origins_list,
        allow_credentials=True,
        allow_methods=["*"],
        allow_headers=["*"],
    )

    app.include_router(api_router)
    return app


app = create_app()