import aiohttp

from aiohttp import ClientResponse

from attrs import define


HTTP_HEADERS = {
    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 "
                  "(KHTML, like Gecko) Chrome/67.0.3396.99 Safari/537.36",
    "Content-Language": "en-US",
    "Cache-Control": "max-age=0",
    "Accept": "*/*",
    "Accept-Charset": "utf-8,ISO-8859-1;q=0.7,*;q=0.3",
    "Accept-Language": "en-US,en;q=0.9,en-US;q=0.8,en;q=0.7",
    "Connection": 'keep-alive'
}


@define
class DeezerEngine:
    cookies: dict
    arl: str = None
    token: str = None

    @classmethod
    async def from_arl(cls, arl: str):
        cookies = {'arl': arl}
        data, cookies = await cls(cookies).call_api(
            'deezer.getUserData', get_cookies=True
        )

        data = data['results']
        token = data['checkForm']

        return cls(
            cookies=cookies,
            arl=arl,
            token=token
        )

    async def call_legacy_api(
            self, request_point: str, params: dict = None
    ):
        async with aiohttp.ClientSession(cookies=self.cookies) as session:
            async with session.get(
                    f"https://api.deezer.com/{request_point}",
                    params=params,
                    headers=HTTP_HEADERS
            ) as r:
                return await r.json()

    @staticmethod
    async def _iter_exact_chunks(response: ClientResponse, chunk_size: int = 2048):
        buffer = b""
        async for chunk in response.content.iter_any():
            buffer += chunk
            while len(buffer) >= chunk_size:
                yield buffer[:chunk_size]
                buffer = buffer[chunk_size:]
        if buffer:
            yield buffer

    async def get_data_iter(self, url: str):
        async with aiohttp.ClientSession(
                cookies=self.cookies,
                headers=HTTP_HEADERS
        ) as session:
            r = await session.get(
                url,
                allow_redirects=True
            )
            async for chunk in self._iter_exact_chunks(r):
                yield chunk

    async def call_api(
            self, method: str, params: dict = None,
            get_cookies: bool = False
    ):
        async with aiohttp.ClientSession(cookies=self.cookies) as session:
            async with session.post(
                    f"https://www.deezer.com/ajax/gw-light.php",
                    params={
                        'method': method,
                        'api_version': '1.0',
                        'input': '3',
                        'api_token': self.token or 'null',
                    },
                    headers=HTTP_HEADERS,
                    json=params
            ) as r:
                if not get_cookies:
                    return await r.json()
                else:
                    return await r.json(), r.cookies