from __future__ import annotations

import time
from typing import Any

DEFAULT_CACHE_TTL_SEC = 3600


class SentMessageCache:
    def __init__(self, ttl_sec: int = DEFAULT_CACHE_TTL_SEC):
        self._ttl_sec = ttl_sec
        self._cache: dict[str, dict[str, Any]] = {}

    def put(self, message_id: str, recipient: str, content: str, metadata: dict[str, Any] | None = None):
        self._cache[message_id] = {
            "message_id": message_id,
            "recipient": recipient,
            "content": content,
            "sent_at": time.time(),
            "metadata": metadata or {},
        }
        self._cleanup()

    def get(self, message_id: str) -> dict[str, Any] | None:
        entry = self._cache.get(message_id)
        if not entry:
            return None
        if time.time() - entry["sent_at"] > self._ttl_sec:
            del self._cache[message_id]
            return None
        return entry

    def remove(self, message_id: str):
        self._cache.pop(message_id, None)

    def _cleanup(self):
        boundary = time.time() - self._ttl_sec
        expired = [k for k, v in self._cache.items() if v["sent_at"] < boundary]
        for k in expired:
            del self._cache[k]

    @property
    def size(self) -> int:
        self._cleanup()
        return len(self._cache)