from io import BytesIO

from aiohttp import ClientSession


class DetaStorage:
    def __init__(self, deta_base_url: str, deta_id: str, deta_key: str):
        self.deta_url = deta_base_url
        self.deta_id = deta_id
        self.deta_key = deta_key
        self.headers = {
            "X-Api-Key": self.deta_key,
        }

    async def download(self, file_name: str) -> BytesIO | None:
        """

        :param file_name: The name of the file to be downloaded
        :return: Either bytes f successfull download or None if failed
        """
        async with ClientSession(headers=self.headers) as session:
            async with session.get(f"{self.deta_url}/files/download?name={file_name}") as response:
                if response.status == 200:
                    return BytesIO(await response.read())
                elif response.status == 404:
                    return None
                else:
                    raise Exception("Download failed")

    async def upload(self, file: bytes, file_name: str) -> None:
        """
        :param file: The file in bytes
        :param file_name: The name of the file
        :return:
        """
        async with ClientSession(headers=self.headers) as session:
            async with session.post(f"{self.deta_url}/files?name={file_name}", data=file) as response:
                if response.status == 201:
                    return None
                else:
                    raise Exception("Upload failed")

    async def delete(self, file_names: [str]) -> None:
        async with ClientSession(headers=self.headers) as session:
            async with session.delete(f"{self.deta_url}/files", json={"names": file_names}) as response:
                if response.status == 200:
                    return None
                else:
                    raise Exception("Delete failed")