main.py

from fastapi import FastAPI
from fastapi.concurrency import asynccontextmanager
from fastapi.routing import APIRoute
from fastapi.middleware.cors import CORSMiddleware
import database
from database import init_db, session_factory
from seed import seed_if_empty
from routes import main_router
from user.permission import Permission
import os
from fastapi.openapi.utils import get_openapi
import redis.asyncio as redis


def generate_unique_id(route: APIRoute):
    if len(route.tags) > 0:
        return f"{route.tags[0]}-{route.name}"
    else:
        return f"{route.path}{route.name}"


@asynccontextmanager
async def lifespan(app: FastAPI):
    if not os.getenv("ENVIRONMENT") == "testing":
        database.redis_client = redis.from_url(
            os.getenv("REDIS_URL"),
            decode_responses=True,
        )
        await database.redis_client.ping()

    if os.getenv("ENVIRONMENT") == "development":
        # Not needed if you setup a migration system like Alembic
        init_db()
        with session_factory() as db:
            seed_if_empty(app, db)
    elif os.getenv("ENVIRONMENT") == "testing":
        init_db()

    yield
    # after yield comes shutdown logic
    if os.getenv("ENVIRONMENT") != "testing":
        if database.redis_client:
            await database.redis_client.close()
            database.redis_client = None


# No Swagger/OpenAPI page for production
no_docs = os.getenv("ENVIRONMENT") == "production"

dev_origins = [
    "http://localhost:3000",  # If we don't specify port, it defaults to 80
    "http://127.0.0.1:3000",
    "http://localhost:3001",  # Sometimes VSCode thinks 3000 is used, so the frontend uses a higher port
    "http://127.0.0.1:3001",  # Wildcard ports don't work with cookies so we need this cursed mess :(
    "http://localhost:3002",
    "http://127.0.0.1:3002",
    "http://localhost:3003",
    "http://127.0.0.1:3003",
]

stage_origins = [
    "https://stage.frontend.fsektionen.se",
    "http://localhost:3000",  # If we don't specify port, it defaults to 80
    "http://127.0.0.1:3000",
    "http://localhost:3001",  # Sometimes VSCode thinks 3000 is used, so the frontend uses a higher port
    "http://127.0.0.1:3001",
    "http://localhost:3002",
    "http://127.0.0.1:3002",
    "http://localhost:3003",
    "http://127.0.0.1:3003",
]

production_origins = ["https://fsektionen.se"]

app = FastAPI(
    lifespan=lifespan,
    redoc_url=None if no_docs else "/redoc",
    docs_url=None if no_docs else "/docs",
    generate_unique_id_function=generate_unique_id,
)

if os.getenv("ENVIRONMENT") == "development":
    app.add_middleware(
        CORSMiddleware,
        allow_origins=dev_origins,
        allow_credentials=True,
        allow_methods=["*"],
        allow_headers=["*"],
    )

if os.getenv("ENVIRONMENT") == "stage":
    app.add_middleware(
        CORSMiddleware,
        allow_origins=stage_origins,
        allow_credentials=True,
        allow_methods=["*"],
        allow_headers=["*"],
    )

if os.getenv("ENVIRONMENT") == "production":
    app.add_middleware(
        CORSMiddleware,
        allow_origins=production_origins,
        allow_credentials=True,
        allow_methods=["*"],
        allow_headers=["*"],
    )

app.include_router(router=main_router)


def custom_openapi():  # type: ignore
    if app.openapi_schema:
        return app.openapi_schema

    # Generate the default OpenAPI schema
    openapi_schema = get_openapi(
        title=app.title,
        version=app.version,
        description=app.description,
        routes=app.routes,
    )

    # Override the servers list to include only http://localhost:8000
    # Uncomment the line below when generating with openapi-generator in flutter app
    # openapi_schema["servers"] = [{"url": "http://10.0.2.2:8000"}]

    # Cache the schema for future calls
    app.openapi_schema = openapi_schema
    return app.openapi_schema


# Override FastAPI's default OpenAPI generation with our custom function
app.openapi = custom_openapi


@app.get("/")
def hello_route():
    return {"message": "Välkommen till F-Sektionens bäckend"}


@app.get("/user-only", dependencies=[Permission.base()])
def user_only():
    return {"message": "Hello, you are a user."}


@app.get("/member-only", dependencies=[Permission.member()])
def member_only():
    return {"message": "Congratz! Only members can reach this route"}


@app.get("/manage-event-only", dependencies=[Permission.require("manage", "Event")])
def permission_route():
    return {"message": "Congratz. You reached a manage:Event route"}