import datetime
from typing import Any, Dict, List

import aiohttp
from bs4 import BeautifulSoup

from weather.api import WeatherApi
from weather.model import WeatherResponse, WeatherValue

from . import datehelp
from .location import LOCATION_BUNDLE
from .parser import DAYS_PARSER, LOCATION_PARSER, ONE_DAY_PARSER, ROW_PARSERS


class GismeteoApi(WeatherApi):
    BASE_URL = "https://www.gismeteo.ru"

    USER_AGENT = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.0.0 Safari/537.36"
    COOKIE = "cf_clearance=U28mYVC0ENu88vorlL_CWmWOoevvXp0vb4xCqfqYC9s-1722273367-1.0.1.1-IDV73azTHY0V.NAnmEvok3zf5HHEkvF098pmya7IiqRRB5nk3FhbLCb0AeWm_kpTFqi1niFk2mYN_ramGTSl0A"

    async def _request(self, endpoint: str) -> str:
        url = f"{self.BASE_URL}/{endpoint}"
        async with aiohttp.ClientSession(
            headers={
                "User-Agent": self.USER_AGENT,
                "Cookie": self.COOKIE,
            },
            raise_for_status=True,
        ) as session:
            async with session.request("GET", url) as response:
                return await response.text()

    def _parse_oneday(self, date: datetime.date, data: str) -> WeatherResponse:
        result: List[Dict[str, Any]] = []
        soup = BeautifulSoup(data, features="html.parser")
        location = LOCATION_PARSER.parse_location(data)
        widget = ONE_DAY_PARSER.parse_widget(soup)
        for parser in ROW_PARSERS:
            for index, value in enumerate(parser.parse_row(widget)):
                while len(result) < index + 1:
                    result.append({})
                result[index][parser.KEY] = value
        values = [WeatherValue(**item) for item in result]
        return WeatherResponse(
            location=location or "n/a",
            date=date,
            period="day",
            values=values,
        )

    def _parse_manydays(self, data: str) -> WeatherResponse:
        result: List[Dict[str, Any]] = []
        soup = BeautifulSoup(data, features="html.parser")
        location = LOCATION_PARSER.parse_location(data)
        widget = DAYS_PARSER.parse_widget(soup)
        for parser in ROW_PARSERS:
            for index, value in enumerate(parser.parse_row(widget)):
                while len(result) < index + 1:
                    result.append({})
                result[index][parser.KEY] = value
        print(">", result)
        values = [WeatherValue(**item) for item in result]
        return WeatherResponse(
            location=location or "n/a",
            date=datetime.date.today(),
            period="days",
            values=values,
        )

    async def get_day(self, location_id: str, date: datetime.date) -> WeatherResponse:
        location = LOCATION_BUNDLE.parse(location_id)
        data = await self._request(f"weather-{location}/{datehelp.dump(date)}")
        return self._parse_oneday(date, data)

    async def get_days(self, location_id: str, days: int) -> WeatherResponse:
        location = LOCATION_BUNDLE.parse(location_id)
        data = await self._request(f"weather-{location}/{days}-days")
        return self._parse_manydays(data)