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

from api.routes import router
from mcp_client import get_github_mcp_client

app = FastAPI(title="AI Codebase Assistant")

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

app.include_router(router, prefix="/api")

mcp_client_instance = get_github_mcp_client()


@app.on_event("startup")
async def startup_event():
    await mcp_client_instance.connect()


@app.on_event("shutdown")
async def shutdown_event():
    await mcp_client_instance.disconnect()


@app.get("/health")
def health_check():
    return {"status": "ok"}