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

"""
sudo ./install.sh --help

Usage: ./install.sh [options]

Options:

    --uninstall                      only uninstall from the installation prefix
    --destdir=[<none>]               set installation root
    --prefix=[/usr/local]            set installation prefix
    --without=[<none>]               comma-separated list of components to not install
    --components=[<none>]            comma-separated list of components to install
    --list-components                list available components
    --sysconfdir=[/etc]              install system configuration files
    --bindir=[/bin]                  install binaries
    --libdir=[/lib]                  install libraries
    --datadir=[/share]               install data
    --mandir=[/share/man]            install man pages in PATH
    --docdir=[\<default\>]           install documentation in PATH
    --disable-ldconfig               don't run ldconfig after installation (Linux only)
    --disable-verify                 don't obsolete
    --verbose                        run with verbose output


sudo ./install.sh --list-components

# Available components

* rustc
* rust-std-x86_64-unknown-linux-gnu
* rust-docs
* rust-docs-json-preview
* rust-demangler-preview
* cargo
* rustfmt-preview
* rls-preview
* rust-analyzer-preview
* llvm-tools-preview
* clippy-preview
* rust-analysis-x86_64-unknown-linux-gnu

"""
components_to_install = ["rustc", "cargo", "rust-std-armv7-unknown-linux-gnueabihf"]
app_dir = "/opt/rust"
app_sys_path = makepath(app_dir, "bin")


def install():
    zip_name = glob.glob("rust-*.tar.gz")[0]
    sub_dir = zip_name.rsplit(".tar", 1)[0]

    # Get components list (corresponding to subfolders)
    with tarfile.open(zip_name, "r") as tar:
        file = tar.extractfile(f"{sub_dir}/components")
        if file:
            content = file.read()
    components = content.decode().split("\n")

    print("Installing: %s to: %s" % (zip_name, app_dir))
    # dirs_list = str(run(f'tar -tf {zip_name} --exclude="*/*/*"')).split('\n')
    # run(f'tar -xvf {zip_name} --exclude="{sub_dir}/rust-docs" --exclude "{sub_dir}/rust-docs-json-preview"')
    exclude_list = [f'--exclude="{sub_dir}/{c}"' for c in components if c and c not in components_to_install]
    run(f"tar -xvf {zip_name} " + " ".join([a for a in exclude_list]))
    # print(run(f'{makepath(sub_dir, "install.sh")} --destdir="{app_dir}" --components="{",".join([a for a in components_to_install])}"'))
    print(run(f'{makepath(sub_dir, "install.sh")} --prefix="{app_dir}" --components="{",".join([a for a in components_to_install])}"'))
    # move_directory(sub_dir, app_dir)
    print('Linking rustc binary')
    if os.path.isfile('/opt/rust/bin/rustc'):
        print('ln -s /opt/rust/bin/rustc /usr/bin/rustc')
        remove_file('/usr/bin/rustc')
        run('ln -s /opt/rust/bin/rustc /usr/bin/rustc')

    print('Linking cargo binary')
    if os.path.isfile('/opt/rust/bin/cargo'):
        print('ln -s /opt/rust/bin/cargo /usr/bin/cargo')
        remove_file('/usr/bin/cargo')
        run('ln -s /opt/rust/bin/cargo /usr/bin/cargo')


def uninstall():
    if isfile(makepath(app_dir, "/lib/rustlib/uninstall.sh")):
        run(makepath(app_dir, "/lib/rustlib/uninstall.sh"))
    if isdir(app_dir):
        remove_tree(app_dir)
    remove_file('/usr/bin/cargo')
    remove_file('/usr/bin/rustc')

                                

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


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


    git_repo = "rust-lang/rust"
    url_api = "https://api.github.com/repos/%s/releases/latest" % git_repo

    # Getting latest informations from Github API
    json_load = json.loads(wgets(url_api, proxies=proxies))

    version = json_load["tag_name"]


    download_dict = {
        "windows-x64": f"https://static.rust-lang.org/dist/rust-{version}-x86_64-pc-windows-msvc.msi",
        "windows-x86": f"https://static.rust-lang.org/dist/rust-{version}-i686-pc-windows-msvc.msi",
        "windows-all": f"https://static.rust-lang.org/dist/rust-{version}-i686-pc-windows-msvc.msi",
        "linux-x64": f"https://static.rust-lang.org/dist/rust-{version}-x86_64-unknown-linux-gnu.tar.gz",
        "linux-arm64": f"https://static.rust-lang.org/dist/rust-{version}-aarch64-unknown-linux-gnu.tar.gz",
        "darwin-x64": f"https://static.rust-lang.org/dist/rust-{version}-x86_64-apple-darwin.tar.gz",
        "darwin-arm": f"https://static.rust-lang.org/dist/rust-{version}-aarch64-apple-darwin.tar.gz",
        "linux-arm": f"https://static.rust-lang.org/dist/rust-{version}-armv7-unknown-linux-gnueabihf.tar.gz"
    }
    download_url = download_dict[control.target_os + "-" + ensure_list(control.architecture)[0]]
    latest_bin = download_url.split("/")[-1]

    # Downloading latest binaries
    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)

    # 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()

    # Deleting outdated binaries
    remove_outdated_binaries(latest_bin, ["exe", "msi", "deb", "rpm", "dmg", "pkg", ".tar.gz", ".tar"])

    # Validating or not update-package-sources
    return package_updated