mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-06-16 01:35:36 -04:00
Mike
ac94885c84
* refactor(constants): single source of truth for data dir + merge core/src constants Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com> * docs(contributing): use named src.constants for data paths, drop core/constants references Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com> --------- Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
import os
|
|
import re
|
|
from pathlib import Path
|
|
|
|
from fastapi import HTTPException
|
|
|
|
from src.constants import GENERATED_IMAGES_DIR
|
|
|
|
|
|
GENERATED_IMAGE_DIR = Path(GENERATED_IMAGES_DIR)
|
|
GENERATED_IMAGE_RE = re.compile(
|
|
r"^[a-f0-9]{8,64}\.(png|jpg|jpeg|webp|gif|mp4|mov|webm|mkv|m4v)$"
|
|
)
|
|
GENERATED_IMAGE_HEADERS = {
|
|
"Cache-Control": "public, max-age=31536000, immutable",
|
|
"X-Content-Type-Options": "nosniff",
|
|
}
|
|
|
|
|
|
def resolve_generated_image_path(filename: str) -> Path:
|
|
if not isinstance(filename, str) or not GENERATED_IMAGE_RE.fullmatch(filename):
|
|
raise HTTPException(status_code=400, detail="Invalid filename")
|
|
root = GENERATED_IMAGE_DIR.resolve()
|
|
path = (GENERATED_IMAGE_DIR / filename).resolve()
|
|
try:
|
|
if os.path.commonpath([str(root), str(path)]) != str(root):
|
|
raise ValueError
|
|
except Exception:
|
|
raise HTTPException(status_code=400, detail="Invalid filename")
|
|
if not path.exists():
|
|
raise HTTPException(status_code=404, detail="Image not found")
|
|
return path
|