"""The FastAPI app. `packetstation.main:app` is built on first access so imports need no settings.""" from __future__ import annotations import asyncio import contextlib import logging from collections.abc import AsyncIterator from pathlib import Path from typing import Any from fastapi import FastAPI from fastapi.responses import FileResponse, HTMLResponse, JSONResponse, Response from starlette.concurrency import run_in_threadpool from . import db, guard from .api import errors as api_errors from .api.origin import OriginCheckMiddleware from .auth.ldap import DirectoryProtocol from .events import EventHub from .proxy import TrustedProxies, TrustedProxyMiddleware from .security import RequestLogMiddleware, SecurityHeadersMiddleware from .settings import Settings, get_settings log = logging.getLogger(__name__) PLACEHOLDER = """ Packet Station

Packet Station

The web client is not built into this image (STATIC_DIR missing).

""" NO_FALLBACK = ("api", "_blob") def _static_file(static_dir: Path, path: str) -> Path | None: """The file under static_dir for a URL path, or None. Never escapes static_dir.""" if not path: return None root = static_dir.resolve() candidate = (root / path).resolve() if not candidate.is_relative_to(root) or not candidate.is_file(): return None return candidate def _load_departments(s: Settings) -> None: """Rebuild the group map from the departments table; the built-in seed stays when the DB is not there.""" from .auth import groups try: with db.connect_as("app", s, autocommit=True) as conn: deps = groups.load_departments(conn) except Exception as e: log.warning("group map from the built-in departments (database: %s)", type(e).__name__) return if deps: groups.use_departments(deps) def create_app( settings: Settings | None = None, *, run_guard: bool = True, directory: DirectoryProtocol | None = None ) -> FastAPI: s = settings or get_settings() if run_guard: guard.startup_checks(s) trusted = TrustedProxies(s.trusted_proxy_host) hub = EventHub(connect=lambda: db.connect_as("app", s, autocommit=True)) @contextlib.asynccontextmanager async def lifespan(_: FastAPI) -> AsyncIterator[None]: await asyncio.to_thread(trusted.refresh) task = asyncio.create_task(trusted.run()) if trusted.host else None deps_task = asyncio.create_task(asyncio.to_thread(_load_departments, s)) hub.start() # connects in its own thread and retries with backoff; never blocks startup log.info("packetstation app starting", extra={"ps": {"build": s.ps_build, "stack": s.stack}}) try: yield finally: if task: task.cancel() deps_task.cancel() await asyncio.to_thread(hub.stop) await run_in_threadpool(db.close_pool) app = FastAPI( title="Packet Station", version=s.ps_build, lifespan=lifespan, docs_url=None, redoc_url=None, openapi_url=None ) app.state.settings = s app.state.trusted = trusted app.state.directory = directory # None: built from settings on first use app.state.events = hub app.state.session_key = None app.state.groups_cache = None api_errors.install(app) from .api import account, admin, auth, departments from .api import events as events_api for module in (auth, account, admin, departments, events_api): app.include_router(module.router) @app.get("/healthz") def healthz() -> dict[str, Any]: return {"ok": True, "build": s.ps_build} @app.get("/api/version") def version() -> dict[str, Any]: return {"build": s.ps_build, "stack": s.stack, "site_tz": s.site_tz} @app.get("/api/db-health") def db_health() -> Response: try: db.ping(s) except db.DatabaseUnavailable: log.warning("db-health: database unavailable") return JSONResponse({"ok": False, "error": "database unavailable"}, status_code=503) return JSONResponse({"ok": True}) @app.api_route("/{path:path}", methods=["GET", "HEAD"], include_in_schema=False) def static(path: str) -> Response: if path.split("/", 1)[0] in NO_FALLBACK: return JSONResponse({"detail": "Not Found"}, status_code=404) static_dir = s.static_dir index = static_dir / "index.html" if not index.is_file(): return HTMLResponse(PLACEHOLDER) found = _static_file(static_dir, path) if found is not None: return FileResponse(found) return FileResponse(index, headers={"Cache-Control": "no-cache"}) # Last added runs first: proxy rewrite, then the log (sees the real ip), then headers, then the Origin check app.add_middleware(OriginCheckMiddleware) app.add_middleware(SecurityHeadersMiddleware) app.add_middleware(RequestLogMiddleware, build=s.ps_build) app.add_middleware(TrustedProxyMiddleware, trusted=trusted) return app _app: FastAPI | None = None def __getattr__(name: str) -> Any: global _app if name == "app": if _app is None: _app = create_app() return _app raise AttributeError(name)