from setuphelpers import *
import re

r"""
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
# https://support.microsoft.com/en-us/topic/-operation-is-not-supported-error-installing-a-post-checkpoint-update-by-double-clicking-the-msu-package-86b89ef4-d5d3-4a2d-b471-3d67c8ea4f0e

def install():

    with EnsureWUAUServRunning():

        kb_files = [
            'windows11.0-kb5043080-x64_953449672073f8fb99badb4cc6d5d7849b9c83e8.msu',
            'windows11.0-kb5050009-x64_97aac2ab4f607b11d50ad2fd88a5841ee0b18dd5.msu'
            ]

        for kb_file in kb_files:
            kb_guess = re.findall(r'^.*-(KB.*)-',kb_file)
            if not kb_guess or not is_kb_installed(kb_guess[0]):
                print("Installing {}".format(kb_file))
                run('dism.exe /Online /Quiet /NoRestart /Add-Package /PackagePath:"{}"'.format(kb_file), accept_returncodes=[0, 3010, 2359302, -2145124329], timeout=1800)
            else:
                print("{} already installed".format(kb_file))

    if is_pending_reboot():
        print("A reboot is needed !")

                                

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


def update_package():
    # Initializing variables
    url_dl = [
    "http://dl.delivery.mp.microsoft.com/filestreamingservice/files/66557ecc-650e-4c70-8baf-c624c4de912f/public/windows11.0-kb5050009-x64_97aac2ab4f607b11d50ad2fd88a5841ee0b18dd5.msu",
    "https://catalog.sf.dl.delivery.mp.microsoft.com/filestreamingservice/files/d8b7f92b-bd35-4b4c-96e5-46ce984b31e0/public/windows11.0-kb5043080-x64_953449672073f8fb99badb4cc6d5d7849b9c83e8.msu"
    ]

    proxies = get_proxies()

    for url in url_dl:
        latest_bin = url.rsplit('/',1)[-1]

        # Downloading latest binaries
        if not isfile(latest_bin):
            print("Downloading: %s" % latest_bin)
            wget(url, latest_bin, proxies=proxies)

    # Changing 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)