app/utils/file_utils.py

import hashlib
import mimetypes
from pathlib import Path
from uuid import uuid4


SUPPORTED_EXTENSIONS = {".pdf", ".jpg", ".jpeg", ".png", ".webp"}


def normalize_extension(filename: str) -> str:
    return Path(filename or "").suffix.lower()


def safe_original_filename(filename: str) -> str:
    name = Path(filename or "upload").name
    return name or "upload"


def build_safe_stored_filename(extension: str) -> str:
    return f"{uuid4().hex}{extension}"


def ensure_child_path(parent: Path, candidate: Path) -> Path:
    resolved_parent = parent.resolve()
    resolved_candidate = candidate.resolve()
    if resolved_parent != resolved_candidate.parent:
        raise ValueError("Path traversal attempt rejected")
    return resolved_candidate


def sha256_hex(data: bytes) -> str:
    return hashlib.sha256(data).hexdigest()


def guess_mime_type(filename: str) -> str:
    return mimetypes.guess_type(filename)[0] or "application/octet-stream"