# -*- coding: utf-8 -*-
from setuphelpers import *
import psutil
import time
import threading

"""
  {
   "key":"{ce044972-8c61-4a19-8f29-08e5c75f2674}",
   "name":"SeaDrive 1.0.12",
   "version":"1.0.12",
   "install_date":"",
   "install_location":"",
   "uninstall_string":"\"C:\\ProgramData\\Package Cache\\{ce044972-8c61-4a19-8f29-08e5c75f2674}\\seadrive.exe\"  /uninstall",
   "publisher":"Seafile ltd.",
   "system_component":0,
   "win64":false
  }
  }
   "key":"{7FD14F30-762C-48CF-830B-063CD4F99188}",
   "name":"SeaDrive 3.0.8",
   "version":"3.0.8",
   "install_date":"2024-03-18 00:00:00",
   "install_location":"",
   "uninstall_string":"MsiExec.exe /X{7FD14F30-762C-48CF-830B-063CD4F99188}",
   "publisher":"HaiWenHuZhi ltd.",
   "system_component":0,
   "win64":true
  }
"""
# seadriveRoot
# old=C:\Users\username\seadrive\data
# new=C:/Users/username/seadrive_root


def install():
    bin_name = glob.glob("seadrive-*.msi")[0]
    uninstall_seadrive_v1()
    install_msi_if_needed(
        bin_name,
        min_version=control.get_software_version(),
        timeout=600,
    )

    # Clear the uninstallkeys to avoid the no-code audit and uninstall
    uninstallkey.clear()


def session_setup():
    print("Disabling: auto-update check and configuring base options")
    registry_setstring(HKEY_CURRENT_USER, r"SOFTWARE\SeaDrive", "ShellExtDisabled", "1")
    registry_setstring(HKEY_CURRENT_USER, r"SOFTWARE\SeaDrive\Seafile Drive Client\Language", "current", "en")
    registry_setstring(HKEY_CURRENT_USER, r"SOFTWARE\SeaDrive\Seafile Drive Client\Misc", "SparkleAlreadyEnableUpdateByDefault", "true")
    registry_setstring(HKEY_CURRENT_USER, r"SOFTWARE\SeaDrive\Seafile Drive Client\WinSparkle", "CheckForUpdates", "0")
    registry_setstring(HKEY_CURRENT_USER, r"Software\SeaDrive\Seafile Drive Client\Behavior", "hideMainWindowWhenStarted", "true")
    #registry_setstring(HKEY_CURRENT_USER, r"SOFTWARE\SeaDrive\Seafile Drive Client\Settings", "diskLetter", "S:")
    
    #Since version 3, there is no more drive letter assigned. Here is how to configure it.
    mount_point = registry_readstring(HKEY_CURRENT_USER, r"SOFTWARE\SeaDrive\Seafile Drive Client\Settings", "seadriveRoot")
    if isdir(mount_point):
        for folder in glob.glob(f'{mount_point}\*\Shared with groups'):
            if isdir(folder):
                registry_setstring(HKEY_CURRENT_USER, r"SOFTWARE\Microsoft\Windows\CurrentVersion\Run", "Seadrive_letter" , rf'''C:\Windows\System32\cmd.exe /C "subst S: {folder.split('Shared with groups')[0]}"''')
                #run(f'subst S: {folder.split("Shared with groups")[0]}')
                break
    
    #Seadrive is now installed in programfiles, not programfiles32
    registry_setstring(HKEY_CURRENT_USER, r"SOFTWARE\Microsoft\Windows\CurrentVersion\Run", "Seadrive", r"C:\Program Files\SeaDrive\bin\seadrive-gui.exe")


def audit():
    # Declaring local variables
    audit_status = "OK"
    app_check = installed_softwares("SeaDrive")
    if app_check:
        installed_version = app_check[0]["version"]
        control.name = app_check[0]["name"]
    else:
        installed_version = ""
        control.name = control.package.split("-", 1)[-1].replace("-", " ").title()
    audit_version = False

    # Auditing software
    if not installed_version:
        print(f"{control.name} is not installed.")
        audit_status = "ERROR"
    elif audit_version and Version(installed_version, 4) < Version(control.get_software_version(), 4):
        print(f"{control.name} ({installed_version}) installed version should be: {control.get_software_version()}")
        audit_status = "WARNING"
    else:
        print(f"{control.name} is installed." if installed_version in control.name else f"{control.name} ({installed_version})  is installed.")
        audit_status = "OK"

    return audit_status


def uninstall():
    # "C:\Program Files\SeaDrive\bin\seadrive-gui.exe" --remove-user-data
    for app_check in installed_softwares("SeaDrive"):
        try:
            run(uninstall_cmd(app_check["key"]), timeout=60)
        except:
            killalltasks("seadrive-gui.exe")
            killalltasks(ensure_list(control.impacted_process))
            wait_uninstallkey_absent(keywords="SeaDrive")
            time.sleep(60)
            if installed_softwares("SeaDrive"):
                error("Uninstalling SeaDrive was unsuccessful.")


def uninstall_seadrive_v1():
    app_uninstallkey = "{82442898-8FE9-44E8-8F50-D2EACBB80848}"
    # Uninstalling the SeaDrive dependency called Dokan Library
    for uninstall in installed_softwares("Dokan Library"):
        print("Removing: %s" % uninstall["name"])
        run(uninstall_cmd(uninstall["key"]))

    # Uninstalling SeaDrive
    for uninstall in installed_softwares(name="SeaDrive 1"):
        # if uninstall["key"] == app_uninstallkey:
        #     continue
        print("Removing: %s" % uninstall["name"])
        t = threading.Thread(target=killseadrive)
        t.start()
        run(uninstall_cmd(uninstall["key"]))


def killseadrive():
    time.sleep(15)
    killalltasks("seadrive-gui.exe")
    time.sleep(15)
    killalltasks("seadrive-gui.exe")
    time.sleep(15)
    killalltasks("seadrive-gui.exe")

                                

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


def update_package():
    # Declaring local variables
    package_updated = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    url = "https://www.seafile.com/en/download/"

    # Getting latest version from official sources
    print("URL used is: %s" % url)
    for bs_search in bs_find_all(url, "a", "class", "download-op", proxies=proxies):
        if "seadrive-" in bs_search.get("href", ""):
            version = bs_search.text
            download_url = bs_search["href"]
            latest_bin = download_url.split("/")[-1]
            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)