# -*- coding: utf-8 -*-
from setuphelpers import *
task_name = 'HPIA'
CommandVendor = 'wmic systemenclosure get manufacturer'
Vendor = str(run(CommandVendor))

def install():
    print('Vérification de la marque de l\'ordinateur...')

    if Vendor.find('HP') != -1:
        app_dir = makepath(programdata,"HP","HPIA")

        if isdir(app_dir):
            killalltasks(ensure_list(control.impacted_process))
            #Wait for process to end
            time.sleep(5)
            remove_tree(app_dir)

        bin_name = glob.glob("hp-hpia*.exe")[0]
        mkdirs(app_dir)

        run(
            bin_name + f" /s /e /f \"{app_dir}\"",
            accept_returncodes = [0, 3010, 1168]
        )

        print('copie des fichiers')
        mkdirs('C:\Windows\Wapt\HPIA')
        filecopyto('HPIA.ps1','C:\Windows\Wapt\HPIA')
        filecopyto('HPIA.xml','C:\Windows\Wapt\HPIA')

        print('création de la tâche planifiée')
        if not task_exists(task_name):
            run(r'schtasks /create /xml C:\Windows\Wapt\HPIA\HPIA.xml /tn HPIA')
        else:
            print("task %s exist and is active" % task_name)
    else:
        print('Ce n\'est pas un PC HP...')


def uninstall():

    app_dir = makepath(programdata,"HP","HPIA")
    task_dir = makepath("C:","Windows","Wapt","HPIA")
    task_name = "HPIA"

    killalltasks(ensure_list(control.impacted_process))
    time.sleep(5)

    if isdir(app_dir):
        remove_tree(app_dir)

    if isdir(task_dir):
        remove_tree(task_dir)

    # Removing the scheduled task
    if task_exists(task_name):
        try:
            delete_task(task_name)
        except:
            print('Unable to remove the task: %s' % task_name)

def audit():
    if Vendor.find('HP') != -1:
        if not task_exists(task_name):
            print("task %s don't exist, WAPT is installing package to fix that" % task_name)
            install()
            return "WARNING"
        if not enable_task(task_name):
            print("task %s is disable, WAPT is going to enable the task" % task_name)
            enable_task(task_name)
            return "WARNING"
        else:
            print("task %s exist and is active" % task_name)
            return "OK"
    else:
        print('Ce n\'est pas un PC HP...')
        return "OK"



                                

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


def update_package():
    # Declaring local variables
    package_updated = False
    proxies = get_proxies_from_wapt_console()
    if not proxies:
        proxies = get_proxies()
    update_dict = {
        "bin_contains": {
            "windows-all": ".exe",
        },
    }
    url_base = "https://ftp.ext.hp.com/pub/caps-softpaq/cmit/HPIA.html"

    # Getting latest version from official sources
    links = bs_find_all(url_base, "a", proxies=proxies)
    for link in links:
        if '.exe' in link['href']:
            href = link['href']
            download_url = href
            latest_bin = download_url.split('/')[-1]
            version = latest_bin.split("-")[-1].replace(".exe","")
            break


    # Downloading latest binaries
    print("Latest %s version is: %s" % (control.name, version))
    print("Download URL is: %s" % download_url)
    if not isfile(latest_bin):
        print("Downloading: %s" % latest_bin)
        wget(download_url, latest_bin, proxies=proxies)
    else:
        print("Binary is present: %s" % latest_bin)

    # Deleting outdated binaries
    remove_outdated_binaries(latest_bin)
    # arch_list = ensure_list(control.architecture)
    # remove_outdated_binaries(version, filename_contains=("x64" if "x64" in arch_list else "x86" if "x86" in arch_list else []))

    # Checking version from file
    if get_os_name() == "Windows" and "windows" in control.target_os.lower():
        version_from_file = get_version_from_binary(latest_bin)
        if Version(version_from_file, 4) == Version(version, 4):
            print(f"INFO: Binary file version ({version_from_file}) corresponds to online version ({version})")
        else:
            error(f"ERROR: Binary file version ({version_from_file}) do NOT corresponds to online version ({version})")

    # Changing version of the package
    if Version(version, 4) > Version(control.get_software_version(), 4):
        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)
    control.save_control_to_wapt()

    # Validating or not update-package-sources
    return package_updated

    # # Changing version of the package and validating update-package-sources
    # return complete_control_version(control, version)