# -*- coding: utf-8 -*-
from setuphelpers import *
import time
import glob
import json
import base64
# Declaring global variables - Warnings: 1) WAPT context is only available in package functions; 2) Global variables are not persistent between calls

locsrvrustdesk = makepath(programfiles,'RustDeskServer')

apiurl = ''

def install():

    #replace nssm by angelize
    if service_installed("RustDesk hbbs service"):
        service_delete('RustDesk hbbs service')

    if service_installed("RustDesk hbbr service"):
        service_delete('RustDesk hbbr service')

    bin_name = glob.glob("*.zip")[0]

    unzip(bin_name)

    if service_installed('RustDeskServer'):
        if service_is_running('RustDeskServer'):
            service_stop('RustDeskServer')

    copytree2("RustDeskServer",locsrvrustdesk)
    remove_file(makepath("rustdesk-server-windows-x86_64-unsigned",'x86_64','RustDeskServer.Setup.exe'))

    copytree2(makepath("rustdesk-server-windows-x86_64-unsigned",'x86_64'),locsrvrustdesk)
    mkdirs(makepath(locsrvrustdesk ,'log'))

    if not service_installed('RustDeskServer'):
        run(r'"%s" --install' % makepath(locsrvrustdesk,"RustDeskServer.exe"))

    run_notfatal('netsh advfirewall firewall delete rule name="%s" program="%s"'% ("hbbr", makepath(locsrvrustdesk, "hbbr.exe")))
    run_notfatal('netsh advfirewall firewall delete rule name="%s" program="%s"'% ("hbbs", makepath(locsrvrustdesk, "hbbs.exe")))


    run('netsh advfirewall firewall add rule name="%s" dir=in profile=private,domain,public action=allow program="%s" enable=yes'% ("hbbr", makepath(locsrvrustdesk, "hbbr.exe")))
    run('netsh advfirewall firewall add rule name="%s" dir=in profile=private,domain,public action=allow program="%s" enable=yes'% ("hbbs", makepath(locsrvrustdesk, "hbbs.exe")))

    if service_is_running('RustDeskServer'):
        service_stop('RustDeskServer')
    service_start('RustDeskServer')

def audit():

    hostname = get_hostname()
    nb=0
    while not glob.glob(makepath(locsrvrustdesk,'*.pub')):
        nb=nb+1
        if nb == 120:
            error("*.pub not found")
        time.sleep(1)

    pubkey = glob.glob(makepath(locsrvrustdesk,'*.pub'))[0]

    key = open(pubkey,'r').read()

    if not '.' in hostname:
        if not params.get("install_with_luti", False):
            error('please configure fqdn')

    print('fqdn use is %s' % hostname)

    config = {"host":hostname,"relay":hostname,"key":key,"api":apiurl}
    rustdesk_list = list(base64.b64encode(json.dumps(config,separators=(',', ':')).encode('utf-8')).decode('utf-8').replace('=',''))
    rustdesk_list.reverse()
    rustdesk_cfg = ''.join(rustdesk_list)

    data = {"config" : config, "rustdesk_cfg" : rustdesk_cfg}


    WAPT.write_audit_data_if_changed("rustdesk-server", 'rustdesk-server', data, max_count=365)
    return "OK"

def uninstall():
    service_delete('RustDeskServer')
    remove_tree(locsrvrustdesk)



                                

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


def update_package():
    # Declaration of local variables
    package_updated = False
    proxies = get_proxies_from_wapt_console()
    if not proxies:
        proxies = get_proxies()
    api_url = "https://api.github.com/repos/rustdesk/rustdesk-server/releases/latest"



    # Get data from API
    releases_dict = wgets(api_url, proxies=proxies, as_json=True)
    exe_found = False  # Flag pour indiquer la découverte d'un fichier .msi
    for release in releases_dict:
        if exe_found:
            break
        for asset in releases_dict["assets"]:
            if "windows-x86_64" in asset["browser_download_url"]:
                url_download = asset["browser_download_url"]
                filename = url_download.split("/")[-1]
                last_version = releases_dict["name"].replace("-",".")
                exe_found = True  # Mettre à jour le flag pour indiquer qu'un .exe a été trouvé
                break

    if not isfile(filename):
        package_updated = True
        wget(url_download, filename, proxies=proxies)

    # nettoyer les fichiers temporaires
    for f in glob.glob("*.zip"):
        if f != filename:
            remove_file(f)

    wget('https://wapt.tranquil.it/wapt/releases/latest/bin/win64/waptsvc.exe',"RustDeskServer\\RustDeskServer.exe")



    control.set_software_version(last_version)
    control.save_control_to_wapt()

    return package_updated