# -*- 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 *
import tarfile

app_path = "/opt/volta"
bin_files = ['volta', 'volta-migrate', 'volta-shim']

def extract_tar_gz(filename, path="."):
    with tarfile.open(filename, "r:gz") as tar:
        result = tar.extractall(path)
    print(result)

def install():
    archive_file = glob.glob("volta*linux.tar.gz")[0]
    extracted_folder = "volta"

    # Delete older version
    if isdir(app_path):
        remove_tree(app_path)

    extract_tar_gz(archive_file, path=extracted_folder)
    copytree2(extracted_folder, app_path)

    for bin_file in bin_files:
        bin_path = f"/usr/local/bin/{bin_file}"
        if isfile(bin_path) or os.path.islink(bin_path):
            print("Removing existing symlink")
            remove_file(bin_path)
            
        print(f"Creating {bin_file} alias in path")
        os.symlink(f"{app_path}/{bin_file}", bin_path)

        # Delete extracted folder
        if isdir(extracted_folder):
            remove_tree(extracted_folder)    

def uninstall():
    # Removing of symlink
    for bin_file in bin_files:
        bin_path = f"/usr/local/bin/{bin_file}"
        if isfile(bin_path) or os.path.islink(bin_path):
            remove_file(bin_path)
        
    # Removing of the software 
    if isdir(app_path):
        remove_tree(app_path)
                                

# -*- 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 *



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

    os_dict = {
        "windows": "-x86_64.msi", 
        "linux": "-linux.tar.gz",
        "darwin": "-macos.tar.gz"
    }
    api_url = "https://api.github.com/repos/volta-cli/volta/releases/latest"

    releases_dict = wgets(api_url, proxies=proxies, as_json=True)
    for asset in releases_dict["assets"]:
        if asset["browser_download_url"].endswith(os_dict[control.target_os]):
            download_url = asset["browser_download_url"]
            latest_bin = asset["name"]
            version = releases_dict["tag_name"].replace("v", "")
            break

    # Downloading latest binaries
    print(f"Latest {app_name} version is: {version}")
    print(f"Download URL is: {download_url}")
    if not isfile(latest_bin):
        print(f"Downloading: {latest_bin}")
        wget(download_url, latest_bin, proxies=proxies)
    else:
        print(f"Binary is present: {latest_bin}")

    # 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'*.tar.gz'):
        if f != latest_bin:
            remove_file(f)

    control.set_software_version(version)
    control.save_control_to_wapt()

    return package_updated