# -*- coding: utf-8 -*-
##################################################
# This file is part of WAPT Enterprise
# All right reserved, (c) Tranquil IT Systems 2024
# For more information please refer to
# https://wapt.tranquil.it/store/licences.html
##################################################
from setuphelpers import *
from setupdevhelpers import *

app_name = "Postman"
app_dir = makepath(programfiles, "Postman")
app_path = makepath(app_dir, "Postman.exe")
installed_version = get_file_properties(app_path).get("ProductVersion", "")

def install():

    bin_name = glob.glob("Postman*.exe")[0]
    nupkg_file = f'Postman-{control.get_software_version()}-full.nupkg'

    # Uninstall older version before new one
    if installed_version and Version(installed_version) < Version(control.get_software_version()):
        uninstall()

    unzip_with_7zip(bin_name, target=basedir, filenames=[nupkg_file])
    unzip(nupkg_file, target=app_name)

    copytree2(makepath(app_name, "lib" ,"net45"), app_dir) # Copy the portable version to programfiles

    create_desktop_shortcut(app_name, target=app_path)
    create_programs_menu_shortcut(app_name, target=app_path)

    remove_tree(app_name, ignore_errors=True)
    if isfile(nupkg_file):
        remove_file(nupkg_file)

def session_setup():
    storage_dir =  makepath(user_appdata, 'Postman', 'storage')
    settings_file = makepath(storage_dir, 'settings.json')

    print(storage_dir)
    print(settings_file)

    settings = dict()

    if isfile(settings_file):
        settings = json_load_file(settings_file)
    mkdirs(storage_dir)

    settings['downloadedVersion'] = None
    settings['lastKnownVersion'] = control.get_software_version()

    json_write_file(settings_file, settings)
        

def audit():
    # Auditing software
    audit_status = "OK"
    if Version(installed_version) < Version(control.get_software_version()):
        print(f"{app_name} is installed in version ({installed_version}) instead of ({control.get_software_version()})")
        audit_status = "WARNING"
    elif isdir(app_dir) and not dir_is_empty(app_dir):
        print(f"{app_name} ({installed_version}) is installed")
        audit_status = "OK"
    else:
        print(f"{app_name} is not installed")
        audit_status = "ERROR"
    return audit_status

def uninstall():
    # Uninstalling software
    killalltasks(ensure_list(control.impacted_process))
    if isdir(app_dir):
        remove_tree(app_dir)

    # Removing shortcuts
    remove_desktop_shortcut(app_name)
    remove_programs_menu_shortcut(app_name)
                                

# -*- coding: utf-8 -*-
##################################################
# This file is part of WAPT Enterprise
# All right reserved, (c) Tranquil IT Systems 2024
# For more information please refer to
# https://wapt.tranquil.it/store/licences.html
##################################################
from setuphelpers import *
from setupdevhelpers import *
import waptlicences


def update_package():
    # Declaring local variables
    package_updated = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    app_name = control.name

    download_dict = {
        "windows-x64" : "https://dl.pstmn.io/download/latest/win64",
        "darwin-x64" : "https://dl.pstmn.io/download/latest/osx_64",
        "darwin-arm" : "https://dl.pstmn.io/download/latest/osx_arm64",
        "debian_based-x64" : "https://dl.pstmn.io/download/latest/linux_64",
        "debian_based-arm" : "https://dl.pstmn.io/download/latest/linux_arm64",
    }

    os_type = control.target_os + "-" + ensure_list(control.architecture)[0]

    download_url = download_dict[os_type]
    latest_bin = requests.head(download_url, proxies=proxies).headers['Content-Disposition'].split("filename=")[-1]
    latest_bin = latest_bin.replace('%20', ' ')
    latest_bin_extension = latest_bin.rsplit('.', 1)[-1]

    print(f"Download URL is: {download_url}")
    print(f"Downloading: {latest_bin}")

    wget(download_url, latest_bin, proxies=proxies)
    version = get_version_from_binary(latest_bin)

    print(f"Latest {app_name} version is: {version}")

    expected_issuer = 'Postman, Inc.'
    sign_name = waptlicences.check_exe_signing_certificate(latest_bin)[0]
    if sign_name != expected_issuer:
        error(f'Bad issuer {sign_name} != {expected_issuer} ')

    # Changing version of the package
    if Version(version) > Version(control.get_software_version()):
        print(f"Software version updated (from: {control.get_software_version()} to: {Version(version)})")
        package_updated = True
    else:
        print(f"Software version up-to-date ({Version(version)})")

    for f in glob.glob(f'*.{latest_bin_extension}'):
        if f != latest_bin:
            remove_file(f)

    control.set_software_version(version)
    control.save_control_to_wapt()

    return package_updated