import asyncio
from typing import Any, Dict

from aiohttp import ClientConnectionError, ClientSession, ClientTimeout

from .const import _LOGGER


class APIClient:
    """Class to perform basic API requests"""

    def __init__(self, session: ClientSession = None):
        """Constructor."""
        self.session = session

    async def make_request(self, url: str):
        """Retrieve data from API."""
        internal_session: bool = False
        if self.session is None:
            internal_session = True
            self.session = ClientSession()

        _LOGGER.debug(f"Try to fetch {url}")

        try:
            async with self.session.get(url, timeout=ClientTimeout(total=9)) as res:
                if res.status != 200:
                    raise ApiError(f"Invalid response: {res.status}")

                json: Dict[str, Any] = await res.json()
                if internal_session:
                    await self.session.close()
                    self.session = None
                return json
        except (ClientConnectionError, asyncio.TimeoutError):
            if internal_session:
                await self.session.close()
                self.session = None
            raise ApiError(f"Could not connect to {url}")


class ApiError(Exception):
    """Raised when API request ended in error."""

    def __init__(self, status: str):
        """Initialize."""
        super().__init__(status)
        self.status = status