# -*- coding: utf-8 -*-
from setuphelpers import *

allow_arbitrary_install = True
package_depend_order_list_any_prefix = [
    "tis-selenium-chrome-driver",
    "tis-selenium-firefox-driver",
]


def install():
    unzip("selenium.zip")
    path_selenium = makepath(WAPT.wapt_base_dir, "selenium")
    mkdirs(path_selenium)
    copytree2("selenium", path_selenium)

    # Meta package part
    package_prefix = control.package.split("-", 1)[0]
    package_depend_order_list = [package_prefix + "-" + x.split("-", 1)[1] for x in package_depend_order_list_any_prefix]
    package_depend_found = False

    if [y for y in package_depend_order_list if WAPT.is_installed(y)]:
        package_depend_found = True
        print("OK: A WAPT Selenium Driver is already installed with WAPT")
    else:
        if allow_arbitrary_install:
            print(
                "WARNING: A WAPT Selenium Driver will be installed with WAPT since this package allow arbitrary install based on available package in a specific order"
            )
            for pkg_name in package_depend_order_list:
                if WAPT.is_available(pkg_name):
                    package_depend_found = True
                    print("Installing: %s" % pkg_name)
                    WAPT.install(pkg_name)
                    WAPT.audit(pkg_name)
                    break
        if not package_depend_found:
            error(
                "ERROR: No WAPT Selenium Driver have been deployed on this system, please import and deploy the package adapted to your organization"
            )


def uninstall():
    path_selenium = makepath(WAPT.wapt_base_dir, "selenium")
    if isdir(path_selenium):
        remove_tree(path_selenium)

                                

# -*- coding: utf-8 -*-
from setuphelpers import *
from waptutils import create_recursive_zip
from waptutils import CustomZipFile


def update_package():
    # Declaring local variables
    package_updated = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    path_selenium = makepath(basedir, "selenium")

    if isdir(path_selenium):
        remove_tree(path_selenium)
    mkdirs(path_selenium)

    if get_os_name() == "Windows":
        pathwaptpython = makepath(WAPT.wapt_base_dir, "waptpython.exe")
    else:
        pathwaptpython = makepath(WAPT.wapt_base_dir, "waptpython.sh")

    if proxies:
        run('"%s" -m pip install selenium --target="%s" --proxy="%s"' % (pathwaptpython, path_selenium, proxies["http"]))
    else:
        run('"%s" -m pip install selenium --target="%s"' % (pathwaptpython, path_selenium))

    # Get Selenium version from dirname
    version = glob.glob(makepath(path_selenium, "selenium-*.dist-info"))[0].rsplit("-", 2)[-2].rsplit(".", 1)[0]

    # Create Selenium zip
    create_recursive_zip(CustomZipFile(r"selenium.zip", "w"), path_selenium)
    remove_tree(path_selenium)

    # Changing version of the package
    if Version(version) > Version(control.get_software_version()):
        print("Software version updated (from: %s to: %s)" % (control.get_software_version(), Version(version)))
        package_updated = True
    else:
        print("Software version up-to-date (%s)" % Version(version))
    control.set_software_version(version, inc_build=True)
    control.save_control_to_wapt()

    # Validating or not update-package-sources
    return package_updated