"""store_result / get_result — сохранение и чтение результата обработки."""

from typing import Optional

from .state import _sessions, _lock


def store_result(sid: str, zip_data: Optional[bytes]) -> bool:
    """Сохранить ZIP-результат или инвалидировать его, передав None.

    Returns:
        True если сохранено, False если сессии нет.
    """
    with _lock:
        s = _sessions.get(sid)
        if not s:
            return False
        s["result"] = zip_data
        return True


def get_result(sid: str) -> Optional[bytes]:
    """Получить результат обработки.

    Returns:
        ZIP-архив или None если сессия не найдена/результат не готов.
    """
    with _lock:
        s = _sessions.get(sid)
        return s["result"] if s else None