"""
Contains browser objects.
"""
from selenium import webdriver

# from selenium.webdriver import Firefox as FirefoxBrowser
# from selenium.webdriver.firefox.options import Options
# from selenium.webdriver.firefox.firefox_binary import FirefoxBinary

from project_settings import SETTINGS


def firefox_options():
    """ Set Firefox options """
    options = webdriver.firefox.options.Options()
    options.headless = False

    # Set download options
    download_dir = SETTINGS["browsers"]["firefox"]["download_path"]
    options.set_preference("browser.download.folderList", 2)
    options.set_preference("browser.download.manager.showWhenStarting", False)
    options.set_preference("browser.download.dir", download_dir)
    for whitelisted_file_type in SETTINGS["browsers"]["firefox"][
        "whitelisted_file_types"
    ]:
        options.set_preference(
            "browser.helperApps.neverAsk.saveToDisk", whitelisted_file_type
        )

    return options


class Firefox(webdriver.Firefox):
    """ Represents a firefox browser controlled by selenium """

    def __init__(
        self,
        firefox_binary=webdriver.firefox.firefox_binary.FirefoxBinary(
            SETTINGS["browsers"]["firefox"]["binary_path"]
        ),
        executable_path=SETTINGS["browsers"]["firefox"]["driver_path"],
        firefox_options=firefox_options(),
    ):
        super().__init__(
            firefox_binary=firefox_binary,
            executable_path=executable_path,
            firefox_options=firefox_options,
        )