# -*- coding: utf-8 -*-
from setuphelpers import *
import psutil
import time
import datetime
import ctypes

try:
    from waptenterprise.waptservice.enterprise import get_active_sessions, start_interactive_process
except:
    from waptservice.enterprise import get_active_sessions, start_interactive_process
import win32api
import sys
import platform

# Usable WAPT package functions: install(), uninstall(), session_setup(), audit(), update_package()
# Declaring global variables - Warnings: 1) WAPT context is only available in package functions; 2) Global variables are not persistent between calls
allow_force_reboot = False
boot_limit = 7  # In days
boot_limit_max = 30  # In days
force_reboot_timeout = 7200  # In seconds
message_box_title = "Information message about your PC uptime - The IT Support Team"
message_box_title_fr = "Message d'information à propos de la durée d'allumage du PC - Le Support Informatique"
# boot_limit_message = "This machine must be rebooted because it has been up for too long, Would you like to reboot now?"
boot_limit_message_string = "This machine must be rebooted because it has been up for: %s, Would you like to reboot now?"
boot_limit_message_fr_string = "Ce PC doit être redémarré, car il est allumé depuis : %s, Voulez-vous redémarrer maintenant ?"
# boot_limit_max_message = "This PC will be FORCE reboot in 2 hours because he's up since too long"
boot_limit_max_message_string = "Reboot will be forced in %s hours because this machine has been up for: %s"
boot_limit_max_message_fr_string = "Ce PC va être redémarré de manière FORCÉE dans %s heures, car il est allumé depuis : %s"


def install():
    pass
    r""" # Initializing variables
    wapt_enterprise_dir = makepath(WAPT.wapt_base_dir,'waptenterprise')

    # Checking that the WAPT Agent is well in Enterprise version
    if isdir(wapt_enterprise_dir):
        print('Continuing: WAPT Agent is well installed in Enterprise version')
    else:
        error('WAPT Agent is NOT installed in Enterprise version, please upgrade')
    """


def session_setup():
    print("This machine has been up for:")
    print(convert_time_to_date(uptime_seconds()))

    # Translating message
    if get_language() == "fr":
        message_box_title = message_box_title_fr
        boot_limit_message_string = boot_limit_message_fr_string
        boot_limit_max_message_string = boot_limit_max_message_fr_string

    # Avoiding user pop-up spamming
    check_timeout = 1800
    date_format = "%Y-%m-%d %H:%M:%S"  # datetime.datetime.strptime(date_pwsh, "%Y/%m/%d %H:%M:%S")
    date = run_powershell('Get-Date -Format "yyyy-MM-dd HH:mm:ss"')  # datetime.datetime.now()
    date_reg = registry_readstring(HKEY_CURRENT_USER, r"SOFTWARE\WAPT\Shutdown Information", "Last Check")
    if date_reg != "":
        date_diff = datetime.datetime.strptime(date, date_format) - datetime.datetime.strptime(date_reg, date_format)
        if date_diff.total_seconds() < check_timeout:
            return "Already checked"
    registry_set(HKEY_CURRENT_USER, r"SOFTWARE\WAPT\Shutdown Information", "Last Check", date)

    # Notify the user
    if allow_force_reboot:
        if uptime_days() > boot_limit_max:
            message_box(
                message_box_title,
                boot_limit_max_message_string % (convert_time_to_hours(force_reboot_timeout), convert_time_to_date(uptime_seconds(), translate=True)),
            )
            reboot_machine(force=1, timeout=force_reboot_timeout)
            return "ERROR"
    if uptime_days() > boot_limit:
        message_box_ask_reboot(message_box_title, boot_limit_message_string % convert_time_to_date(uptime_seconds(), translate=True))
        return "WARNING"
    if uptime_days() < boot_limit:
        return "OK"


def audit():
    print("This PC has been up for:")
    print(convert_time_to_date(uptime_seconds()))

    if allow_force_reboot:
        if uptime_days() > boot_limit_max:
            run_session_setup(control.package)
            print("Forcing reboot after the defined timeout.")
            reboot_machine(force=1, timeout=force_reboot_timeout)
            return "ERROR"
    if uptime_days() > boot_limit:
        run_session_setup(control.package)
        return "WARNING"
    if uptime_days() < boot_limit:
        return "OK"


def run_session_setup(package_name):
    for session_id in get_active_sessions():
        start_interactive_process("wapt-get", "--hide session-setup %s -f" % package_name, session_id=session_id)  # , minimize=True


def message_box(title, text, style=0):
    if platform.python_version_tuple()[0] == "3":
        import importlib

        importlib.reload(sys)
    else:
        reload(sys)
        sys.setdefaultencoding("utf-8")

    MB_OK = 0
    MB_OKCANCEL = 1
    MB_YESNOCANCEL = 3
    MB_YESNO = 4
    ctypes.windll.user32.MessageBoxW(0, "%s" % text, "%s" % title, style)
    return


def message_box_ask_reboot(title, text):
    if platform.python_version_tuple()[0] == "3":
        import importlib

        importlib.reload(sys)
    else:
        reload(sys)
        sys.setdefaultencoding("utf-8")

    ID_OK = 0
    ID_CANCEL = 2
    ID_ABORT = 3
    ID_YES = 6
    ID_NO = 7

    response = ctypes.windll.user32.MessageBoxW(0, "%s" % text, "%s" % title, 4)
    if response == ID_YES:
        reboot_machine()
    else:
        pass


def uptime_seconds():
    return time.time() - psutil.boot_time()


def uptime_minutes():
    return (time.time() - psutil.boot_time()) / 60


def uptime_hours():
    return (time.time() - psutil.boot_time()) / 3600


def uptime_days():
    return ((time.time() - psutil.boot_time()) / 3600) / 24


def convert_time_to_hours(seconds):
    return seconds / 3600


def convert_time_to_date(seconds, translate=False):
    from datetime import datetime, timedelta

    sec = timedelta(seconds=seconds)
    d = datetime(1, 1, 1) + sec
    # print("DAYS:HOURS:MIN:SEC")
    # up 23 weeks, 3 days, 23 hours, 16 minutes
    # up 23 weeks 3 days 23 hours 16 minutes
    if translate:
        if get_language() == "fr":
            return "%d jours %d heures %d minutes %d secondes" % (d.day - 1, d.hour, d.minute, d.second)
    return "%d days %d hours %d minutes %d seconds" % (d.day - 1, d.hour, d.minute, d.second)

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


def update_package():
    # Incrementing version of the package
    control.version = "%s-%s" % (control.get_software_version(), int(control.version.split("-")[-1]) + 1)
    control.save_control_to_wapt()
    print("Changing version to: %s in WAPT\\control" % control.version)