import os
import tarfile
import time
from collections import defaultdict
from pathlib import Path

import wget
from app.core.utils import logger, timed_cache
from app.settings import BASE_DIR, DRIVER_SESSION_TTL, GECKO_DRIVER_VERSION
from selenium import webdriver
from selenium.common.exceptions import (
    NoSuchElementException,
    StaleElementReferenceException,
    WebDriverException,
)
from selenium.webdriver.firefox import options
from selenium.webdriver.firefox.service import Service
from selenium.webdriver.firefox.webdriver import RemoteWebDriver, WebDriver


def download_gecko_driver() -> None:
    gecko_driver = (
        f'https://github.com/mozilla/geckodriver/releases/download/v{GECKO_DRIVER_VERSION}/'
        f'geckodriver-v{GECKO_DRIVER_VERSION}-linux64.tar.gz'
    )

    if not Path(BASE_DIR / 'geckodriver').exists():
        logger.info(f'Downloading gecodriver v {GECKO_DRIVER_VERSION}...')
        geckodriver_file = wget.download(
            url=gecko_driver, out=BASE_DIR.resolve().as_posix()
        )

        with tarfile.open(geckodriver_file) as tar:
            tar.extractall(BASE_DIR)
        os.remove(f'{BASE_DIR / "geckodriver"}-v{GECKO_DRIVER_VERSION}-linux64.tar.gz')
        logger.info(f'\ngeckodriver has been downloaded to folder {BASE_DIR}')


def configure_firefox_driver(private_window: bool = False) -> WebDriver | None:
    opt = options.Options()
    opt.headless = True
    opt.add_argument('-profile')
    opt.add_argument(f'{Path.home()}/snap/firefox/common/.mozilla/firefox')
    if private_window:
        opt.set_preference("browser.privatebrowsing.autostart", True)
    service = Service(executable_path=(BASE_DIR / 'geckodriver').as_posix())
    try:
        firefox_driver = webdriver.Firefox(service=service, options=opt)
        return firefox_driver
    except WebDriverException:
        logger.error('Error configuring webdriver. Possible it already configured')
        return None


def parse_yandex_maps(
    url: str, message: str, driver: RemoteWebDriver | None = None
) -> str:
    if not driver:
        logger.error('Driver is not configured')
        return 'Что-то пошло не так. :( Драйвер Firefox не сконфигурирован.'

    driver.get(url)
    time.sleep(1)

    bus_arrival: dict[str, str | None] = defaultdict(str)

    try:
        web_elements = driver.find_elements(
            by='class name', value='masstransit-vehicle-snippet-view'
        )
        for web_element in web_elements:
            bus = web_element.find_element(
                by='class name', value='masstransit-vehicle-snippet-view__main-text'
            )
            if bus:
                bus_arrival_time = web_element.find_element(
                    by='class name',
                    value='masstransit-prognoses-view__title-text',
                )
                match bus.text:
                    case "300":
                        bus_arrival["Автобус 300"] = (
                            bus_arrival_time.text if bus_arrival_time else None
                        )
                    case "т19":
                        bus_arrival["Автобус Т19"] = (
                            bus_arrival_time.text if bus_arrival_time else None
                        )
    except NoSuchElementException:
        pass
    except StaleElementReferenceException:
        pass

    if not any(bus_arrival.values()):
        return 'Автобусов 300 или Т19 не найдено. \n\nСмотри на карте :)'

    answer = f'{message}\n\n'
    for bus_name, arrival_time in bus_arrival.items():
        answer += f'{bus_name} - {arrival_time}\n'
    return answer


@timed_cache(seconds=DRIVER_SESSION_TTL)
def get_driver() -> RemoteWebDriver:
    opt = options.Options()
    opt.headless = True
    driver = RemoteWebDriver(
        command_executor='http://selenoid_host:4444/wd/hub', options=opt
    )
    return driver