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

uninstallkey = []

# Declaring specific app values (TO CHANGE)
bin_name_63 = 'Windows8.1-KB2693643-x64.msu'
bin_name_62 = 'Windows6.2-KB2693643-x64.msu'
url_dl_63 = 'https://download.microsoft.com/download/1/8/E/18EA4843-C596-4542-9236-DE46F780806E/Windows8.1-KB2693643-x64.msu'
url_dl_62 = 'https://download.microsoft.com/download/4/3/F/43FAC524-F605-426E-A2AE-EAD03D264FDE/Windows6.2-KB2693643-x64.msu'


def install():
	# Getting the used storage on programfiles before installation (place it on the top)
    get_disk_free_space_before = get_disk_free_space(programfiles)

    # Installing the package
    if windows_version() >= '6.2' and windows_version() < '6.2.9999':
        with EnsureWUAUServRunning():
            try:
                run(r'wusa.exe "%s" /quiet /norestart' % bin_name_62)
            except CalledProcessError as e:
                if e.returncode == 2359302:
                    pass
    elif windows_version() >= '6.3' and windows_version() < '6.3.9999':
        with EnsureWUAUServRunning():
            try:
                run(r'wusa.exe "%s" /quiet /norestart' % bin_name_63)
            except CalledProcessError as e:
                if e.returncode == 2359302:
                    pass
    else:
        print('This Windows version is not compatible with this package')
        pass

    print('Activating features')
    print ('AD')
    features = [
        'RemoteServerAdministrationTools',
        'RemoteServerAdministrationTools-Roles',
        'RemoteServerAdministrationTools-Roles-AD',
        ]
    run(r'c:\windows\sysnative\dism.exe /online /enable-feature %s' % " ".join(['/featurename:%s' % f for f in features]))

    print ('AD-DS')
    features = [
        'RemoteServerAdministrationTools-Roles-AD-DS']
    run(r'c:\windows\sysnative\dism.exe /online /enable-feature %s' % " ".join(['/featurename:%s' % f for f in features]))

    print ('AD-DS-Snapins')
    features = [
        'RemoteServerAdministrationTools-Roles-AD-DS-SnapIns']
    run(r'c:\windows\sysnative\dism.exe /online /enable-feature %s' % " ".join(['/featurename:%s' % f for f in features]))

    print ('DNS')
    features = [
        'RemoteServerAdministrationTools-Roles-DNS',
        ]
    run(r'c:\windows\sysnative\dism.exe /online /enable-feature %s' % " ".join(['/featurename:%s' % f for f in features]))

    print ('GP')
    features = [
        'RemoteServerAdministrationTools-Features',
        'RemoteServerAdministrationTools-Features-GP',
        ]
    run(r'c:\windows\sysnative\dism.exe /online /enable-feature %s' % " ".join(['/featurename:%s' % f for f in features]))

	# Return used storage of the program. (place it on the bottom)
    get_disk_free_space_after = get_disk_free_space(programfiles)
    free_space_after_diff = get_disk_free_space_before - get_disk_free_space_after
    print("Storage used: " + str(free_space_after_diff))


def uninstall():
    with EnsureWUAUServRunning():
        run('wusa.exe /uninstall /kb:2693643  /quiet /norestart')


def update_package():
    print('Download/Update package content from upstream binary sources')

    # Getting proxy informations from WAPT settings
    proxy = {}
    if isfile(makepath(application_data(),'waptconsole','waptconsole.ini')):
        proxywapt = inifile_readstring(makepath(user_local_appdata(),'waptconsole','waptconsole.ini'),'global','http_proxy')
        if proxywapt :
            proxy = {'http':proxywapt,'https':proxywapt}

    # Downloading latest binaries
    if not isfile(bin_name_63):
        print('Downloading: ' + bin_name_63)
        wget(url_dl_63,bin_name_63,proxies=proxy)
        print('Update package done. You can now build-upload your package')

    if not isfile(bin_name_62):
        print('Downloading: ' + bin_name_62)
        wget(url_dl_62,bin_name_62,proxies=proxy)
        print('Update package done. You can now build-upload your package')


if __name__ == '__main__':
    update_package()