Files
manage/backend/src/media_library_viewer_api/main.py
T
2026-05-06 21:25:12 +02:00

113 lines
3.7 KiB
Python

"""FastAPI application entrypoint."""
from __future__ import annotations
import logging
import time
from contextlib import asynccontextmanager
import uvicorn
from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware
from media_library_viewer_api.auth import require_jwt_auth, validate_auth_settings
from media_library_viewer_api.config import get_settings
from media_library_viewer_api.logging_utils import configure_logging, describe_settings
from media_library_viewer_api.routers import dashboard, monitoring, media, files, jobs, users, tasks
from media_library_viewer_api.routers.settings import router as settings_router
from media_library_viewer_api.dependencies import get_mail_queue, get_monitoring_poller, get_settings_store
from media_library_viewer_api.services.known_hosts import ensure_known_hosts_for_machines
logger = logging.getLogger(__name__)
@asynccontextmanager
async def lifespan(app: FastAPI):
"""Application lifespan — startup/shutdown."""
settings = get_settings()
configure_logging(settings.log_level)
validate_auth_settings(settings)
logger.info("Backend startup complete: %s", describe_settings(settings))
store = get_settings_store()
try:
changed = ensure_known_hosts_for_machines(store.list_machines(), settings.ssh_known_hosts_file)
logger.info("Managed known_hosts updated entries=%s file=%s", changed, settings.ssh_known_hosts_file)
except Exception:
logger.exception("Failed to synthesize managed known_hosts file")
mail_queue = get_mail_queue()
monitoring_poller = get_monitoring_poller()
mail_queue.start()
monitoring_poller.start()
yield
monitoring_poller.stop()
mail_queue.stop()
logger.info("Backend shutdown complete")
app = FastAPI(
title="Manage API",
version="0.1.0",
description="Manage API for Jellyfin media browsing, SSH file inspection, server monitoring, and JWT-protected access.",
lifespan=lifespan,
)
# CORS for development (Vite runs on :5173)
app.add_middleware(
CORSMiddleware,
allow_origins=[
"http://localhost:5173",
"http://localhost:3000",
"http://127.0.0.1:5173",
"http://127.0.0.1:3000",
],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
@app.middleware("http")
async def enforce_jwt_auth(request: Request, call_next):
return await require_jwt_auth(request, call_next)
@app.middleware("http")
async def log_requests(request: Request, call_next):
"""Log every API request with timing and outcome."""
start = time.perf_counter()
client_host = request.client.host if request.client else "unknown"
logger.info("request start %s %s client=%s", request.method, request.url.path, client_host)
try:
response = await call_next(request)
except Exception:
logger.exception("request error %s %s client=%s", request.method, request.url.path, client_host)
raise
elapsed_ms = (time.perf_counter() - start) * 1000.0
logger.info(
"request end %s %s status=%s elapsed_ms=%.1f",
request.method,
request.url.path,
response.status_code,
elapsed_ms,
)
return response
# Register routers
app.include_router(dashboard.router)
app.include_router(monitoring.router)
app.include_router(media.router)
app.include_router(files.router)
app.include_router(jobs.router)
app.include_router(users.router)
app.include_router(tasks.router)
app.include_router(settings_router)
@app.get("/api/health")
def health_check() -> dict[str, str]:
"""Simple health check endpoint."""
logger.debug("health check requested")
return {"status": "ok"}
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=8000)