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


# Declaring global variables - Warnings: 1) WAPT context is only available in package functions; 2) Global variables are not persistent between calls

folder_base = r"%s\Windows Kits\10\Assessment and Deployment Kit\Windows Preinstallation Environment" % programfiles32
list_files = []

list_arch = ["x86"]
# list_arch = ["amd64","arm","arm64","x86"]

dict_boot = {"amd64": "bootx64.efi", "x86": "bootia32.efi", "arm": "bootarm.efi", "arm64": "bootaa64.efi"}

for arch in list_arch:
    list_files.extend(
        [
            r"%s\%s\Media\Boot\BCD" % (folder_base, arch),
            r"%s\%s\Media\Boot\boot.sdi" % (folder_base, arch),
            r"%s\%s\Media\EFI\Microsoft\Boot\BCD" % (folder_base, arch),
            r"%s\%s\en-us\winpe.wim" % (folder_base, arch),
            r"%s\%s\Media\bootmgr" % (folder_base, arch),
            r"%s\%s\Media\bootmgr.efi" % (folder_base, arch),
            r"%s\%s\Media\EFI\Boot\%s" % (folder_base, arch, dict_boot[arch]),
        ]
    )


def install():
    for file in list_files:
        parent_path = file.rsplit("\\", 1)[0]
        if not isdir(parent_path):
            mkdirs(parent_path)
        if not isfile(file) or force:
            filecopyto(file.replace(folder_base + "\\", ""), file)

    # for windows7
    if not isfile(r"C:\Windows\System32\Bootsect.exe"):
        filecopyto("Bootsect.exe", r"C:\Windows\System32\Bootsect.exe")

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

folder_base = r"%s\Windows Kits\10\Assessment and Deployment Kit\Windows Preinstallation Environment" % programfiles32
list_files = []

list_arch = ["x86"]
# list_arch = ["amd64","arm","arm64","x86"]

dict_boot = {"amd64": "bootx64.efi", "x86": "bootia32.efi", "arm": "bootarm.efi", "arm64": "bootaa64.efi"}

for arch in list_arch:
    list_files.extend(
        [
            r"%s\%s\Media\Boot\BCD" % (folder_base, arch),
            r"%s\%s\Media\Boot\boot.sdi" % (folder_base, arch),
            r"%s\%s\Media\EFI\Microsoft\Boot\BCD" % (folder_base, arch),
            r"%s\%s\en-us\winpe.wim" % (folder_base, arch),
            r"%s\%s\Media\bootmgr" % (folder_base, arch),
            r"%s\%s\Media\bootmgr.efi" % (folder_base, arch),
            r"%s\%s\Media\EFI\Boot\%s" % (folder_base, arch, dict_boot[arch]),
        ]
    )


def update_package():

    for file in list_files:
        mkdirs(file.replace(folder_base + "\\", "").rsplit("\\", 1)[0])
        filecopyto(file, file.replace(folder_base + "\\", ""))

    # for winpe32
    if isdir(makepath(programfiles32, "7-Zip")):
        copytree2(makepath(programfiles32, "7-Zip"), "7-Zip")
    #################

    filecopyto(r"C:\Windows\System32\Bootsect.exe", "Bootsect.exe")

    prefixpkg = control.package.split("-", 1)[0]
    WAPT= Wapt()
    f = WAPT.is_installed("%s-adkwinpe" % prefixpkg)
    version = f.version.split("-")[0]
    control.version = "%s-%s" % (version, control.version.split("-", 1)[-1])
    control.save_control_to_wapt()