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 starlette.responses import Response

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

from src.config import settings
from src.graphql_api import graphql_app


@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, lifespan=lifespan)
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(),
        }
    }


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 __name__ == "__main__":
    uvicorn.run(
        "main:app",
        host=settings.HOST,
        reload=settings.DEBUG_MODE,
        port=settings.PORT,
    )