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

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

'https://download.microsoft.com/download/E/D/A/EDA6E3AE-31B9-449D-9E81-4E55F0881707/Windows6.1-KB958830-x64-RefreshPkg.msu'
'https://download.microsoft.com/download/E/D/A/EDA6E3AE-31B9-449D-9E81-4E55F0881707/Windows6.1-KB958830-x86-RefreshPkg.msu'

"""
# Declaring global variables - Warnings: 1) WAPT context is only available in package functions; 2) Global variables are not persistent between calls
bin_contains = 'Windows6.1-KB958830-'


def install():
    # Declaring local variables
    bin_name = glob.glob('*%s*.msu' % bin_contains)[0]

    with EnsureWUAUServRunning():
        if params.get("install_with_luti",False):
            run(r'psexec -accepteula -nobanner -s C:\Windows\System32\wusa.exe "%s" /quiet /norestart' % makepath(basedir,bin_name),accept_returncodes=[0, 3010, 2359302], timeout=3000)
        else:
            run(r'wusa.exe "%s" /quiet /norestart' % bin_name,accept_returncodes=[0, 3010, 2359302], timeout=3000)

        print('Activation des features')
        print ('AD')
        features = [
            'RemoteServerAdministrationTools',
            'RemoteServerAdministrationTools-Roles',
            'RemoteServerAdministrationTools-Roles-AD',
            ]

        run(r'dism.exe /online /enable-feature %s' % " ".join(['/featurename:%s' % f for f in features]))

        print ('AD-DS')
        features = [
            'RemoteServerAdministrationTools-Roles-AD-DS']
        run(r'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'dism.exe /online /enable-feature %s' % " ".join(['/featurename:%s' % f for f in features]))

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

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


def uninstall():
    with EnsureWUAUServRunning():
        if params.get("install_with_luti",False):
            run(r'psexec -accepteula -nobanner -s C:\Windows\System32\wusa.exe /uninstall /kb:958830  /quiet /norestart' ,accept_returncodes=[0, 3010, 2359302], timeout=3000)
        else:
            run('wusa.exe /uninstall /kb:958830  /quiet /norestart',timeout=3000)


def update_package():
    wget('http://srvtemplates.ad.tranquil.it/binary_cache/Windows6.1-KB958830-x86-RefreshPkg.msu', 'Windows6.1-KB958830-x86-RefreshPkg.msu',sha256='FCC36F0686F23BF57B0C719DFC86B06E8D2B501F93A08ED81704D15AB1726B41')