package           : tis-rustdesk-config-template
version           : 25
architecture      : all
section           : base
priority          : optional
name              : RustDesk Config
categories        : Utilities
maintainer        : WAPT Team,Tranquil IT
description       : Config for RustDesk
depends           : tis-rustdesk
conflicts         : 
maturity          : PROD
locale            : all
target_os         : all
min_wapt_version  : 2.3
sources           : 
installed_size    : 
impacted_process  : 
description_fr    : Config rustdesk
description_pl    : 
description_de    : 
description_es    : 
description_pt    : 
description_it    : 
description_nl    : 
description_ru    : 
audit_schedule    : 30d
editor            : 
keywords          : RustDesk
licence           : 
homepage          : https://rustdesk.com/
package_uuid      : fe6b79ff-907d-46c3-99fe-70d1e5f12ac5
valid_from        : 
valid_until       : 
forced_install_on : 
changelog         : https://github.com/rustdesk/rustdesk/releases
min_os_version    : 
max_os_version    : 
icon_sha256sum    : 2435ab1227e19048fc0d782821cc7258a41b69dc1e22914aedddd288a38b6ccc
signer            : Tranquil IT
signer_fingerprint: 8c5127a75392be9cc9afd0dbae1222a673072c308c14d88ab246e23832e8c6bb
signature_date    : 2025-01-27T14:08:35.000000
signed_attributes : package,version,architecture,section,priority,name,categories,maintainer,description,depends,conflicts,maturity,locale,target_os,min_wapt_version,sources,installed_size,impacted_process,description_fr,description_pl,description_de,description_es,description_pt,description_it,description_nl,description_ru,audit_schedule,editor,keywords,licence,homepage,package_uuid,valid_from,valid_until,forced_install_on,changelog,min_os_version,max_os_version,icon_sha256sum,signer,signer_fingerprint,signature_date,signed_attributes
signature         : TgnKQ6UvtSKtt/75BMyrMmsyjpDwHs+2S2Jz6XzNqMRSS02YWchHdkoxUMr5bdXZ3hn0xgVkjgM93kSYmDqxRbPjRfuL0Z62F7wi41mM3R3pXV0O4Lr9cU7u/btFbV0RPPYag91da9rMmwbBn6cec0LpqqvAB5oG/0dAWa+wZzFVftdkFCCFSnrq9WJS+Atu9KtL7S6zyFGMJOxwo6CGBzooaIRorRSWL1VD38z+AfE/BZiHUBubvavRS3lN0zvDwK+NhQlnnbLI//45V0HUz2EWfnE0HQFR20sI/dMGZArbXG4gOcLdFzX/oPntm8HWEhRW+OludBep/XSs6E+UaQ==

# -*- coding: utf-8 -*-
from setuphelpers import *
import string, secrets
import platform
import json
import base64

r"""
Usable WAPT package functions: install(), uninstall(), session_setup(), audit(), update_package()

"""

if platform.system() == 'Windows':
    if uninstall_key_exists('RustDesk'):
        RustDeskBin = makepath(install_location('RustDesk'),'rustdesk.exe')
elif platform.system() == 'Darwin':
    RustDeskBin = "/Applications/RustDesk.app/Contents/MacOS/RustDesk"
else:
    RustDeskBin = "/usr/bin/rustdesk"


def install():
    rustdesk_cfg = open('rustdesk.cfg').read()

    alphabet = string.ascii_letters + string.digits
    rustdesk_pw = ''.join(secrets.choice(alphabet) for i in range(12))

    run_notfatal('"%s" --config %s' % (RustDeskBin,rustdesk_cfg))
    run('"%s" --password %s' % (RustDeskBin,rustdesk_pw ))

    if platform.system() == 'Windows':
        if service_is_running("rustdesk"):
            service_stop("rustdesk")
        service_start("rustdesk")

    elif platform.system() == 'Darwin':
        print('todo')
    else:
        systemd_restart_service('rustdesk')

def audit():
    rustdesk_id = run('"%s" --get-id' % RustDeskBin)
    if rustdesk_id.endswith('\n'):
        rustdesk_id = rustdesk_id.split('\n')[-2]

    alphabet = string.ascii_letters + string.digits
    rustdesk_pw = ''.join(secrets.choice(alphabet) for i in range(12))
    run('"%s" --password %s' % (RustDeskBin, rustdesk_pw))

    WAPT.write_audit_data_if_changed("rustdesk", 'rustdesk-id', rustdesk_id, max_count=365)
    WAPT.write_audit_data_if_changed("rustdesk", 'password_encryption', rsa_encrypted_data_str(rustdesk_pw, WAPT.public_certs_dir), max_count=365)
    
    return "OK"

def encode_rustdesk_config(config):
    rustdesk_list = list(base64.b64encode(json.dumps(config, separators=(',', ':')).encode('utf-8')).decode('utf-8').replace('=',''))
    rustdesk_list.reverse()
    return ''.join(rustdesk_list)

def update_package():
    old_conf = ""
    if isfile('rustdesk.cfg'):
        old_conf = open('rustdesk.cfg').read()
        
    newconf = ""
    if not params.get("install_with_luti", False):
        from waptguihelper import message_dialog, memo_dialog,input_dialog, MB_YESNO, MB_ICONQUESTION, ID_YES

        is_cfg_config = message_dialog("RUSTDESK Config", "Do you have an cfg config file ?", MB_YESNO + MB_ICONQUESTION)
        if is_cfg_config == ID_YES:
            newconf = memo_dialog('RUSTDESK Config', "Please give your rustdesk cfg config that you can retrieved in the 'Audit data' of your host server", old_conf)
        else:
            conf = {"api":""}
            conf['host'] = input_dialog('Host','Please give ip or dns name of HOST rustdesk server :','rustdesk.mydomain.lan')
            conf['relay'] = input_dialog('Host','Please give ip or dns name of RELAY rustdesk server :',conf['host'])
            conf['key'] = input_dialog('Host','Please give the public key of the rustdesk server (content of the file /opt/rustdesk/*.pub) :','')

            newconf = encode_rustdesk_config(conf)

        control.package = control.package.replace('-template', '')
        control.save_control_to_wapt()

    with open('rustdesk.cfg', 'w') as f:
        f.write(newconf)