port-ui/app/utils/cache_manager.py

import os
import hashlib
import requests
import mimetypes

class CacheManager:
    def __init__(self, cache_dir="static/cache"):
        self.cache_dir = cache_dir
        self._ensure_cache_dir_exists()

    def _ensure_cache_dir_exists(self):
        if not os.path.exists(self.cache_dir):
            os.makedirs(self.cache_dir)

    def clear_cache(self):
        if os.path.exists(self.cache_dir):
            for filename in os.listdir(self.cache_dir):
                path = os.path.join(self.cache_dir, filename)
                if os.path.isfile(path):
                    os.remove(path)

    def cache_file(self, file_url):
        # generate a short hash for filename
        hash_suffix = hashlib.blake2s(file_url.encode('utf-8'), digest_size=8).hexdigest()
        parts = file_url.rstrip("/").split("/")
        base = parts[-2] if parts[-1] == "download" else parts[-1]

        try:
            resp = requests.get(file_url, stream=True, timeout=5)
            resp.raise_for_status()
        except requests.RequestException:
            return None

        content_type = resp.headers.get('Content-Type', '')
        ext = mimetypes.guess_extension(content_type.split(";")[0].strip()) or ".png"
        filename = f"{base}_{hash_suffix}{ext}"
        full_path = os.path.join(self.cache_dir, filename)

        if not os.path.exists(full_path):
            with open(full_path, "wb") as f:
                for chunk in resp.iter_content(1024):
                    f.write(chunk)

        # return path relative to /static/
        return f"cache/{filename}"