35 lines
1.5 KiB
Python
35 lines
1.5 KiB
Python
from fastapi import APIRouter, HTTPException, Query, Response
|
|
from fastapi.responses import FileResponse
|
|
|
|
from ..media_catalog import KNOWN_MEDIA_ROLES, media_manifest_version, published_assets, published_file
|
|
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/api/v1/media/catalog")
|
|
def media_catalog(
|
|
response: Response,
|
|
entity_type: str | None = Query(None, pattern="^(fish|waterbody|tackle|reference)$"),
|
|
media_role: str | None = Query(None, pattern="^(waterbody_cover|waterbody_map|waterbody_depth_map|waterbody_screenshot|tackle_card|tackle_detail|rig_diagram|tackle_screenshot)$"),
|
|
) -> list[dict]:
|
|
if media_role and entity_type not in {"waterbody", "tackle"}:
|
|
raise HTTPException(status_code=422, detail="media_role requires waterbody or tackle entity_type")
|
|
if media_role and media_role not in KNOWN_MEDIA_ROLES:
|
|
raise HTTPException(status_code=422, detail="unknown media role")
|
|
response.headers["Cache-Control"] = "public, max-age=60, stale-while-revalidate=60"
|
|
response.headers["X-Media-Manifest-Version"] = str(media_manifest_version())
|
|
return published_assets(entity_type, media_role)
|
|
|
|
|
|
@router.get("/api/v1/media/assets/{digest}", response_class=FileResponse)
|
|
def media_asset(digest: str) -> FileResponse:
|
|
item = published_file(digest)
|
|
if not item:
|
|
raise HTTPException(status_code=404, detail="Media asset not found")
|
|
path, media_type = item
|
|
return FileResponse(path, media_type=media_type, headers={
|
|
"Cache-Control": "public, max-age=31536000, immutable",
|
|
"ETag": f'"{digest}"',
|
|
})
|