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

rsat_capabilities = [
    "Rsat.ActiveDirectory.DS-LDS.Tools~~~~0.0.1.0",
    "Rsat.BitLocker.Recovery.Tools~~~~0.0.1.0",
    "Rsat.CertificateServices.Tools~~~~0.0.1.0",
    "Rsat.DHCP.Tools~~~~0.0.1.0",
    "Rsat.Dns.Tools~~~~0.0.1.0",
    "Rsat.FailoverCluster.Management.Tools~~~~0.0.1.0",
    "Rsat.FileServices.Tools~~~~0.0.1.0",
    "Rsat.GroupPolicy.Management.Tools~~~~0.0.1.0",
    "Rsat.IPAM.Client.Tools~~~~0.0.1.0",
    "Rsat.LLDP.Tools~~~~0.0.1.0",
    "Rsat.NetworkController.Tools~~~~0.0.1.0",
    "Rsat.NetworkLoadBalancing.Tools~~~~0.0.1.0",
    "Rsat.RemoteAccess.Management.Tools~~~~0.0.1.0",
    "Rsat.RemoteDesktop.Services.Tools~~~~0.0.1.0",
    "Rsat.ServerManager.Tools~~~~0.0.1.0",
    "Rsat.Shielded.VM.Tools~~~~0.0.1.0",
    "Rsat.StorageMigrationService.Management.Tools~~~~0.0.1.0",
    "Rsat.StorageReplica.Tools~~~~0.0.1.0",
    "Rsat.SystemInsights.Management.Tools~~~~0.0.1.0",
    "Rsat.VolumeActivation.Tools~~~~0.0.1.0",
    "Rsat.WSUS.Tools~~~~0.0.1.0",
]
# rsat_capabilities_to_install = rsat_capabilities
rsat_capabilities_to_install = [
    "Rsat.ActiveDirectory.DS-LDS.Tools~~~~0.0.1.0",
    "Rsat.BitLocker.Recovery.Tools~~~~0.0.1.0",
    "Rsat.Dns.Tools~~~~0.0.1.0",
    "Rsat.GroupPolicy.Management.Tools~~~~0.0.1.0",
    "Rsat.LLDP.Tools~~~~0.0.1.0",
    "Rsat.RemoteAccess.Management.Tools~~~~0.0.1.0",
]


def install():
    # Declaring local variables
    try:
        rsat_binaries_zip = glob.glob("rsat_*.zip")[0]
        unzip(rsat_binaries_zip, ".")
        rsat_binaries_dir = rsat_binaries_zip.split(".")[0]
    except:
        rsat_binaries_dir = glob.glob("rsat_*")[0]
    rsat_source_path = makepath(os.getcwd(), rsat_binaries_dir, "LanguagesAndOptionalFeatures")
    app_error = False

    # Installing the package
    with EnsureWUAUServRunning():
        for capability in rsat_capabilities_to_install:
            print("Installing: %s" % capability)
            try:
                run_powershell('Add-WindowsCapability -Online -Name "%s" -LimitAccess -Source "%s"' % (capability, rsat_source_path))
            except:
                print("Installation failed for: %s" % capability)
                app_error = True

    if app_error:
        error("RSAT capabilities to install do not succeed completely. You may need to restart the computer and try again")


def uninstall():
    # Declaring local variables
    app_error = False

    # Uninstalling the package
    print("Uninstalling all RSAT features")
    with EnsureWUAUServRunning():
        for capability in rsat_capabilities:
            print("Uninstalling: %s" % capability)
            try:
                run_powershell('Remove-WindowsCapability -Online -Name "%s"' % capability)
            except:
                print("Uninstallation failed on: %s" % capability)
                app_error = True

    if app_error and not params.get("running_as_luti"):
        error("RSAT capabilities uninstallation do not succeed completely. You may need to restart the computer and try again")


def audit():
    # Manually check in PowerShell: Get-WindowsCapability -online | ? Name -like Rsat* | FT
    print("Showing installed RSAT features:")
    pprint.pprint(run_powershell("Get-WindowsCapability -Online | Where Name -like Rsat* | Where State -Like Installed | Select Name,RestartNeeded"))
    print("Showing NOT installed RSAT features:")
    pprint.pprint(
        run_powershell("Get-WindowsCapability -Online | Where Name -like Rsat* | Where State -NotLike Installed | Select Name,RestartNeeded")
    )
    return "OK"

                                

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


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

Procedure:
1- Download Windows 10+ FoD/LoF ISO on Microsoft VLSC
2- Run the update_package()
3- Adapt package host capabilities then build-upload the package

"""


def update_package():
    # Declaring local variables
    capabilities_list = [
        "ActiveDirectory",
        "BitLocker",
        "CertificateServices",
        "DHCP",
        "DNS",
        "FailoverCluster",
        "FileServices",
        "GroupPolicy",
        "IPAM",
        "LLDP",
        "NetworkController",
        "NetworkLoadBalancing",
        "RemoteAccess",
        "RemoteDesktop",
        "ServerManager",
        "Shielded",
        "StorageMigrationService",
        "StorageReplica",
        "SystemInsights",
        "VolumeActivation",
        "WSUS",
    ]
    mandatory_files = ["FoDMetadata_", "Conditions", "Neutral"]
    mandatory_capabilities_files = ["~~"]
    # default_lang_list = ["fr-fr", "en-us"]

    # Interacting differently with Luti
    if params.get("running_as_luti"):
        latest_bin = "rsat_windows11v24h2_fr-fr.zip"
        download_url = "http://srvtemplates.ad.tranquil.it/binary_cache/" + latest_bin
        sha256 = "4994f1ccd42243931b7aa0b36cdf1efcbbe71ee44E8fdfb7d8b502af26e90718"
        wget(download_url, latest_bin, sha256=sha256)
        return "OK"

    # Asking for iso path
    iso_path = waptguihelper.filename_dialog(
        "Please provide the desired FoD iso file for the requested OS Version - %s" % control.package,
        "",
        "FoD_LoF.ISO",
        "ISO Files|*.ISO",
    )  # os.getcwd()
    # iso_path = r"C:\temp\SW_DVD9_Win_11_21H2_x64_MultiLang_LangPackAll_LIP_LoF_X22-62148.ISO"

    # Asking for langs
    lang_json = waptguihelper.grid_dialog(
        "Please selected the desired language(s). You can duplicate lines with CTRL + C, CTRL + V",
        json.dumps([{"lang": "en-us"}, {"lang": "fr-fr"}]),
        waptguihelper.GRT_SELECTED,
    )
    lang_list = []
    for l in lang_json:
        lang_list.append(l["lang"])

    if iso_path is None:
        error("Please provide an ISO")

    # Extract with conditions
    conditionnal_include_list = mandatory_capabilities_files + lang_list
    rsat_binaries_zip = "rsat_%s_%s.zip" % (basedir.split(os.path.sep)[-1], "_".join(i for i in lang_list))
    rsat_binaries_dir = rsat_binaries_zip.split(".")[0]

    for d in glob.glob("*rsat_*"):
        if isdir(d):
            remove_tree(d)

    for f in glob.glob("*rsat_*.zip"):
        if isfile(f):
            remove_tree(f)

    if Version(control.min_os_version) >= Version(WindowsVersions.Windows11):
        for f in mandatory_files:
            unzip_with_7zip(iso_path, rsat_binaries_dir, "**/*%s*.cab".lower() % f, recursive=True)

        for f in lang_list:
            unzip_with_7zip(iso_path, rsat_binaries_dir, "LanguagesAndOptionalFeatures/**/*%s*.cab".lower() % f, recursive=True)

        for fc in conditionnal_include_list:
            for cap in capabilities_list:
                unzip_with_7zip(iso_path, rsat_binaries_dir, "**/*%s*%s*.cab".lower() % (cap, fc), recursive=True)
    else:
        for f in mandatory_files:
            unzip_with_7zip(iso_path, rsat_binaries_dir, "*%s*.cab".lower() % f, recursive=True)

        for f in lang_list:
            unzip_with_7zip(iso_path, rsat_binaries_dir, "**/*%s*.cab".lower() % f, recursive=True)

        for fc in conditionnal_include_list:
            for cap in capabilities_list:
                unzip_with_7zip(iso_path, rsat_binaries_dir, "*%s*%s*.cab".lower() % (cap, fc), recursive=True)

    print("You may want to create manually a zip called: %s" % rsat_binaries_zip)