"""
CouncilOS — FastAPI application entrypoint.

Start the server:
    uvicorn main:app --reload --port 8000

API Overview:
    POST   /api/councils/              — Create a blueprint
    GET    /api/councils/              — List all blueprints
    GET    /api/councils/{id}          — Get specific blueprint
    PUT    /api/councils/{id}          — Update a blueprint
    DELETE /api/councils/{id}          — Delete a blueprint
    POST   /api/councils/run           — Start a run (Phase 1 hard-coded graph)
    POST   /api/councils/{id}/run      — Start a run from a blueprint (Phase 3)
    GET    /api/councils/run/{run_id}  — Poll run status/result
    GET    /api/health                 — Health check
    WS     /ws/council/{run_id}        — Real-time agent status events
"""

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

from api.routes import router
from api.blueprint_routes import blueprint_router
from api.websocket import ws_router
from database import init_db, close_db


@asynccontextmanager
async def lifespan(app: FastAPI):
    """Application lifespan: startup and shutdown logic."""
    print("CouncilOS API starting up...")
    await init_db()
    print("Database initialized.")
    yield
    await close_db()
    print("CouncilOS API shutting down...")


app = FastAPI(
    title="CouncilOS API",
    description=(
        "Backend for the CouncilOS multi-agent AI pipeline platform. "
        "Orchestrates LangGraph council runs and streams real-time agent "
        "status via WebSockets."
    ),
    version="0.2.0",
    lifespan=lifespan,
)

# CORS — allow all origins in development; tighten in production
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# Mount REST routes under /api prefix
app.include_router(router, prefix="/api")
app.include_router(blueprint_router, prefix="/api")

# Mount WebSocket routes (no prefix — path is /ws/council/{run_id})
app.include_router(ws_router)


if __name__ == "__main__":
    import uvicorn
    uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)