import contextlib
from typing import List

from cashews import cache

from sticker import bot


class ServiceMessage:
    @staticmethod
    async def set_cache(uid: int, cid: int, mid: int):
        old = await ServiceMessage.get_cache(uid, cid)
        old.append(mid)
        await cache.set(f"service_message:{uid}:{cid}", old, expire=600)

    @staticmethod
    async def get_cache(uid: int, cid: int) -> List[int]:
        data = await cache.get(f"service_message:{uid}:{cid}")
        if data:
            return data
        return []

    @staticmethod
    async def try_delete(uid: int, cid: int):
        mid = await ServiceMessage.get_cache(uid, cid)
        if mid:
            with contextlib.suppress(Exception):
                await bot.delete_messages(cid, list(mid))