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

os_name = "%(os_name)s"


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():
    waptagent_package_file = glob.glob('*.pkg')[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,'/opt/wapt/waptsetup.pkg')
        subprocess.Popen("sleep 60 && installer -package /opt/wapt/waptsetup.pkg -target / && rm -f /opt/wapt/waptsetup.pkg && launchctl kickstart -k system/it.tranquil.waptservice",shell=True, executable='/bin/bash', start_new_session=True)
    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 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()