main.py

import datetime
from contextlib import asynccontextmanager
from pathlib import Path

import graphdoc  # type: ignore
import graphql
import uvicorn
from graphql import GraphQLSyntaxError

from fastapi import Depends, FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from fastapi_mcp import AuthConfig, FastApiMCP
from starlette.responses import Response

from src.api.rest_routes import router as rest_router
from src.config import settings
from src.graphql_api import graphql_app
from src.middleware.auth import verify_auth


@asynccontextmanager
async def lifespan(_app: FastAPI):
    """Application lifespan (replaces deprecated on_event handlers)."""
    _app.state.public_paths = {"/", "/graph/"}
    yield


app = FastAPI(
    title=settings.APP_NAME,
    description=settings.APP_DESCRIPTION,
    lifespan=lifespan,
)
app.include_router(rest_router)
app.mount("/graph", graphql_app)

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=False,
    allow_methods=["*"],
    allow_headers=["X-Forwarded-For", "Authorization", "Content-Type"],
)


@app.get("/")
async def root():
    return {
        "message": {
            "app_name": settings.APP_NAME,
            "system_time": datetime.datetime.now(),
            "discovery_endpoint": "/mcp",
        }
    }


app.mount(
    "/doc_templates",
    StaticFiles(directory=Path(__file__).parent.absolute() / "./src/doc_templates"),
    name="doc_templates",
)


@app.get("/graphql/docs", include_in_schema=False)
async def get_graphql_docs():
    """Handler for graphql docs."""
    path = "./schema.graphql"

    with open(path, "r", encoding="utf-8") as graphql_file:
        schema = graphql_file.read()

    try:
        graphql.parse(schema)
    except GraphQLSyntaxError as exc:
        raise Exception(path, str(exc)) from exc
    return Response(
        content=graphdoc.to_doc(
            schema, templates_path="src/doc_templates", use_cache=False
        ),
        media_type="text/html",
    )


if settings.AUTH_ENABLED:
    mcp = FastApiMCP(
        app,
        name=settings.APP_NAME,
        description=settings.APP_DESCRIPTION,
        auth_config=AuthConfig(dependencies=[Depends(verify_auth)]),
    )
else:
    mcp = FastApiMCP(
        app,
        name=settings.APP_NAME,
        description=settings.APP_DESCRIPTION,
    )

mcp.mount_http()
mcp.setup_server()

if __name__ == "__main__":
    uvicorn.run(
        app,
        host=settings.HOST,
        port=settings.PORT,
        reload=settings.DEBUG_MODE,
    )