app.py

from contextlib import asynccontextmanager
from pathlib import Path
from typing import Any

from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
from starlette.middleware.base import BaseHTTPMiddleware
from starlette.requests import Request
from starlette.responses import Response

from core.storage.database import (
    init_db,
)
from core.routers import ROUTERS

BASE_DIR = Path(__file__).resolve().parent
STATIC_DIR = BASE_DIR / "static"
PAGE_FILES = {
    "/scan": STATIC_DIR / "scan.html",
    "/duplicates": STATIC_DIR / "duplicates.html",
    "/registry": STATIC_DIR / "registry.html",
    "/cleanup": STATIC_DIR / "cleanup.html",
    "/recycle": STATIC_DIR / "recycle.html",
    "/logs": STATIC_DIR / "logs.html",
    "/tools": STATIC_DIR / "tools.html",
    "/user-directory": STATIC_DIR / "user-directory.html",
    "/app-scan": STATIC_DIR / "app-scan.html",
}
NO_CACHE_PATHS = {"/", "/index.html", *PAGE_FILES.keys()}
NO_CACHE_SUFFIXES = (".html", ".js", ".css")


@asynccontextmanager
async def lifespan(_: FastAPI):
    init_db()
    yield


app = FastAPI(lifespan=lifespan)


class StaticNoCacheMiddleware(BaseHTTPMiddleware):
    async def dispatch(self, request: Request, call_next: Any) -> Response:
        response = await call_next(request)

        path = request.url.path
        if path in NO_CACHE_PATHS or path.endswith(NO_CACHE_SUFFIXES):
            response.headers["Cache-Control"] = "no-store, max-age=0"
            response.headers["Pragma"] = "no-cache"

        return response


for router in ROUTERS:
    app.include_router(router)

app.add_middleware(StaticNoCacheMiddleware)
app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")