import os
import pytest
import logging
from logging.handlers import RotatingFileHandler
from collections import Mapping
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities

from .live_server import LiveServer
from .browsers import BROWSERSTACK_CONFIG


@pytest.fixture(scope="function", autouse=True)
def session(db, request):
    """
    Override base test session
    """
    pass


@pytest.fixture(scope="session")
def live_app(app):
    handler = RotatingFileHandler("log/acceptance.log", maxBytes=10000, backupCount=1)
    handler.setLevel(logging.INFO)
    app.logger.addHandler(handler)

    runnable = LiveServer(app, port=8943, timeout=10)
    runnable.spawn_live_server()
    app.server_url = runnable.server_url

    yield app

    runnable.terminate()


class DriverCollection(Mapping):
    """
    Allows access to drivers with dictionary syntax. Keeps track of which ones
    have already been initialized. Allows teardown of all existing drivers.
    """

    def __init__(self):
        self._drivers = {}

    def __iter__(self):
        return iter(self._drivers)

    def __len__(self):
        return len(self._drivers)

    def __getitem__(self, name):
        if name in self._drivers:
            return self._drivers[name]

        elif name in BROWSERSTACK_CONFIG:
            self._drivers[name] = self._build_driver(name)
            return self._drivers[name]

        else:
            raise AttributeError("Driver {} not found".format(name))

    def _build_driver(self, config_key):
        return webdriver.Remote(
            command_executor="http://{}:{}@hub.browserstack.com:80/wd/hub".format(
                os.getenv("BROWSERSTACK_EMAIL"), os.getenv("BROWSERSTACK_TOKEN")
            ),
            desired_capabilities=BROWSERSTACK_CONFIG.get(config_key),
        )

    def teardown(self):
        for driver in self._drivers.values():
            driver.quit()


@pytest.fixture(scope="session")
def drivers():
    driver_collection = DriverCollection()

    yield driver_collection

    driver_collection.teardown()