from typing import Optional, Tuple

from aiomcache import Client

from fastapi_cache.types import Backend


class MemcachedBackend(Backend):
    def __init__(self, mcache: Client):
        self.mcache = mcache

    async def get_with_ttl(self, key: str) -> Tuple[int, Optional[bytes]]:
        return 3600, await self.get(key)

    async def get(self, key: str) -> Optional[bytes]:
        return await self.mcache.get(key.encode())

    async def set(self, key: str, value: bytes, expire: Optional[int] = None) -> None:
        await self.mcache.set(key.encode(), value, exptime=expire or 0)

    async def clear(self, namespace: Optional[str] = None, key: Optional[str] = None) -> int:
        is_deleted = False
        if key:
            is_deleted = await self.mcache.delete(key=key.encode())
        else:
            await self.mcache.flush_all()
            is_deleted = True

        return int(is_deleted)