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

temp_dir = makepath("/tmp")


def create_systemd_onetime_task(name, user, cmd, delay_minutes=2):
    """
    Create cron task
    """
    run_time = time.localtime(time.time() + delay_minutes * 60)

    hour_min = time.strftime("%H:%M:%S", run_time)
    date = time.strftime("%Y-%m-%d", run_time)

    cmd = "systemd-run --on-calendar='%s %s' --description='%s' %s " % (date, hour_min, name, cmd)
    print("Create scheduled task with command : %s" % cmd)
    run(cmd)

def installed_wapt_version():
    try:
        with open(os.path.join(WAPT.wapt_base_dir, "version-full")) as fver:
            installed_wapt_version = fver.read()
    except:
        installed_wapt_version = "0.0.0.0"

    return installed_wapt_version

def install():

    if is_redhat_based():
        waptagent_package_file = glob.glob('*.rpm')[0]
    else:
        waptagent_package_file = glob.glob('*.deb')[0]

    print("Install %s" % control.asrequirement())

    if isfile(makepath(WAPT.wapt_base_dir, "waptserver", "server.py")):
        print("Wapt server installed on this host. Aborting")
        return "OK"

    package_wapt_version = control.get_software_version()

    if force or Version(installed_wapt_version()) < package_wapt_version:
        filecopyto(waptagent_package_file, temp_dir)
        waptagent_file = makepath(temp_dir, waptagent_package_file)
        install_cmd = ""
        if is_redhat_based():
            install_cmd = "yum install %s -y" % waptagent_file
        elif is_debian_based():
            install_cmd = "/usr/bin/apt install -f -y %s" % waptagent_file
        create_systemd_onetime_task("waptupgrade", "root", install_cmd)
    else:
        print("Your current wapt (%s) is same or more recent than the upgrade package (%s). Skipping..." % (installed_wapt_version, package_wapt_version))

def audit():
    # Comparing installed WAPT agent version and package version
    (package_wapt_version, package_packaging) = control.version.split("-")

    if Version(installed_wapt_version()) < control.get_software_version():
        print(
            "The installed version of the WAPT agent and the version of the package are not corresponding. The installation may have just finished. If this warning persists one day after installation, please take this into consideration."
        )
        return "WARNING"
    else:
        print("The installed version of the WAPT agent and the version of the package are corresponding.")
        return "OK"

                                

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

def update_package():
    data = json.loads(wgets('https://wapt.tranquil.it/wapt/releases/latest/wapt_agent_list.json'))

    for entry in data:
        p = PackageEntry()
        p.package = 'tis-waptupgrade'
        p.min_os_version = entry['wapt_min_os_version']
        p.max_os_version = entry['wapt_max_os_version']
        p.target_os = entry['wapt_target_os']
        p.architecture = entry['wapt_architecture']

        if "tis-waptagent-gui" in entry['url'] :
            continue

        if not WAPT.host_capabilities().is_matching_package(p):
            continue

        matchentry = entry
        break

    wget(matchentry['url'],matchentry['url'].split('/')[-1],sha256=matchentry['file_sha256'])

    control.target_os = matchentry['wapt_target_os']
    control.min_os_version = matchentry['wapt_min_os_version']
    control.max_os_version = matchentry['wapt_max_os_version']
    control.architecture = matchentry['wapt_architecture']
    control.set_software_version(matchentry['wapt_version'])
    control.save_control_to_wapt()