"""Filesystem path helpers — destination derivation, hash-suffixed names.""" import re from pathlib import Path _MAX_EXT_LEN = 16 # A Patreon/gallery CDN URL embeds a 32-char hex (MD5) path segment that is the # file's stable per-file identity — the same role gallery-dl's `_filehash` # plays. It is the join key between a post body `` and the local # copy we downloaded (extract_media dedups content vs gallery images by it), so # this ONE extractor must be used for both capture-time persistence and # render-time matching — they cannot be allowed to drift. Match the FIRST 32-hex # run anywhere in the URL (path or query); real CDN URLs carry exactly one. _FILEHASH_RE = re.compile(r"([0-9a-fA-F]{32})") def filehash_from_url(url: str | None) -> str | None: """The 32-char hex (MD5) CDN identity segment of `url`, lowercased, or None when the URL is empty / carries no such segment.""" if not url: return None match = _FILEHASH_RE.search(url) return match.group(1).lower() if match else None def safe_ext(name: str | Path) -> str: """Conservatively extract a short, alphanumeric file extension. gallery-dl and Patreon CDN URLs produce basenames with URL-encoded query-string artifacts, so `Path.suffix` can return 50+ chars of base64-ish junk that blows bounded VARCHAR columns (e.g. PostAttachment.ext varchar(32)). Accept only a suffix ≤16 chars whose post-dot characters are all alphanumeric; otherwise return "" (no known extension). Operator-flagged 2026-05-25 — ONE impl for the importer and the native Patreon client. """ suffix = Path(name).suffix.lower() if not suffix or len(suffix) > _MAX_EXT_LEN: return "" if not all(c.isalnum() for c in suffix[1:]): return "" return suffix def derive_subdir(source_path: Path, import_root: Path) -> str: """Returns the relative subdirectory of source_path under import_root. The top-level folder name is treated as the 'artist' bucket. Nested paths preserve hierarchy. import_root=/import source_path=/import/Alice/sub/x.png -> "Alice/sub" source_path=/import/Alice/x.png -> "Alice" source_path=/import/x.png -> "" """ try: rel = source_path.parent.relative_to(import_root) except ValueError: return "" return str(rel) if str(rel) != "." else "" def hash_suffixed_name(stem: str, sha256_hex: str, ext: str) -> str: """Builds 'stem__'. Examples: hash_suffixed_name("photo", "abcdef1234567890...", ".png") -> "photo__abcdef1234.png" """ return f"{stem}__{sha256_hex[:10]}{ext}" def derive_top_level_artist(source_path: Path, import_root: Path) -> str | None: """Returns the top-level folder name under import_root, or None if the file is directly in import_root. """ subdir = derive_subdir(source_path, import_root) if not subdir: return None return subdir.split("/", 1)[0]