DISMTools 0.7 Preview 3 (#260)

* Update assembly information

* [Fix] Fixed update checker issue

* [Info Dialogs] Small refactor

* [PE Helper] Add README

* [PE Helper] Show restart message in a cleaner fashion

* [PE Helper] Create a file for diskpart list disk

* [Enhance] Save DISM output on operation finish

* [Theme Designer] Fixed deployment issues

* [Theme Engine] Add color theme switch capabilities, add 0.4 Enhanced Themes

* [Unattended answer file] Add option to hide PWSH windows

* [Settings] Add option to hide clock

* [Unattended answer file] Modify welcome message

* [Fix] Prevent user writes to console control

* [Fix] Fixed image edition tasks never finishing

* [DynaLog] Allow logging of additional, non-DISM related exceptions for bgprocs

* [PE Helper] Improve WDS Helper Server Component (client coming soon)

* [Unattended answer file] Fix anchor for checkbox

* [PE Helper] WDS Helper Client and more goodies

- WDS Helper Client
- DT PE Menu

* [Enhance] Disable close capability with Window Helper

* [PE Helper] Finalized WDS Helper Client for TP1

- Fixed possible race conditions
- Separated requests using folders with GUID names
- WDSH server can now block devices with requests denied by WDS
- Fatal errors no longer exit the script, but reboot the computer after 5 seconds

* [Library] Update DISM API to 3.3.12

* [REL] Update What's New section

Author: CodingWonders

DISMTools.vbproj

@@ -84,7 +84,7 @@
       <HintPath>packages\Markdig.0.41.0\lib\net462\Markdig.dll</HintPath>
     </Reference>
     <Reference Include="Microsoft.Dism, Version=1.0.0.0, Culture=neutral, PublicKeyToken=5bba90053b345501, processorArchitecture=MSIL">
-      <HintPath>packages\Microsoft.Dism.3.3.0\lib\net40\Microsoft.Dism.dll</HintPath>
+      <HintPath>packages\Microsoft.Dism.3.3.12\lib\net40\Microsoft.Dism.dll</HintPath>
     </Reference>
     <Reference Include="Microsoft.WindowsAPICodePack.Core, Version=8.0.6.0, Culture=neutral, PublicKeyToken=8afb38e9204fc0a9, processorArchitecture=MSIL">
       <HintPath>packages\WindowsAPICodePack.8.0.6\lib\net48\Microsoft.WindowsAPICodePack.Core.dll</HintPath>
@@ -1198,15 +1198,15 @@
     <None Include="Helpers\extps1\PE_Helper\files\dim_start\dimstart.bat">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
-    <None Include="Helpers\extps1\PE_Helper\files\diskpart\dp_listdisk.dp">
-      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
-    </None>
     <None Include="Helpers\extps1\PE_Helper\files\DISMTools-PE.zip">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
     <None Include="Helpers\extps1\PE_Helper\files\HotInstall.zip">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
+    <None Include="Helpers\extps1\PE_Helper\files\startup\menu.ps1">
+      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
     <None Include="Helpers\extps1\PE_Helper\files\startup\StartInstall.ps1">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
@@ -1222,6 +1222,9 @@
     <None Include="Helpers\extps1\PE_Helper\pxehelpers\wds\prepare_wds.ps1">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
+    <None Include="Helpers\extps1\PE_Helper\pxehelpers\wds\wdshelper.ps1">
+      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
     <None Include="Helpers\extps1\PE_Helper\pxehelpers\wds\wdshelper_server.ps1">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
@@ -1268,10 +1271,12 @@
     <None Include="Tools\UnattendGen\win-x86\publish\UnattendGen.runtimeconfig.json" />
     <None Include="Tools\UnattendGen\win-x86\UnattendGen.deps.json" />
     <None Include="Tools\UnattendGen\win-x86\UnattendGen.runtimeconfig.json" />
-    <None Include="Utilities\ThemeUtilities\ThemeFiles\DT05Dark.ini" />
-    <None Include="Utilities\ThemeUtilities\ThemeFiles\DT05Light.ini" />
     <None Include="Utilities\ThemeUtilities\ThemeFiles\default_dark.ini" />
     <None Include="Utilities\ThemeUtilities\ThemeFiles\default_light.ini" />
+    <None Include="Utilities\ThemeUtilities\ThemeFiles\DT04Dark.ini" />
+    <None Include="Utilities\ThemeUtilities\ThemeFiles\DT04Light.ini" />
+    <None Include="Utilities\ThemeUtilities\ThemeFiles\DT05Dark.ini" />
+    <None Include="Utilities\ThemeUtilities\ThemeFiles\DT05Light.ini" />
   </ItemGroup>
   <ItemGroup>
     <None Include="Panels\readme.md" />
@@ -1812,6 +1817,7 @@
   </ItemGroup>
   <ItemGroup>
     <Folder Include="bin\Debug\" />
+    <Folder Include="Helpers\extps1\PE_Helper\files\diskpart\" />
     <Folder Include="Panels\Set_Ops\" />
     <Folder Include="Tools\UnpEax\" />
     <Folder Include="Updater\" />
@@ -1918,7 +1924,8 @@ IF EXIST "$(SolutionDir)Utilities\ThemeUtilities\ThemeFiles\*.ini" (
 
 :: Copy Theme Designer
 IF EXIST "$(SolutionDir)Tools\DT_ThemeDesigner\out\*.exe" (
-	xcopy "$(SolutionDir)Tools\DT_ThemeDesigner\out\*.exe" "Tools\ThemeDesigner" /cehyi
+	xcopy "$(SolutionDir)Tools\DT_ThemeDesigner\out\*.*" "Tools\ThemeDesigner" /cehyi
+	del "Tools\ThemeDesigner\*.zip" /F /S /Q
 )
 
 IF NOT EXIST AutoUnattend (

Helpers/extps1/PE_Helper/PE_Helper.ps1

@@ -632,13 +632,26 @@ function Start-PECustomization
                 Set-Content -Path "$imagePath\Windows\system32\startnet.cmd" -Value $contents -Force
             }
             Copy-Item -Path "$((Get-Location).Path)\files\startup\StartInstall.ps1" -Destination "$imagePath\StartInstall.ps1" -Force
-            Copy-Item -Path "$((Get-Location).Path)\files\dim_start\dimstart.bat" -Destination "$imagePath\dimstart.bat"
+            Copy-Item -Path "$((Get-Location).Path)\files\dim_start\dimstart.bat" -Destination "$imagePath\dimstart.bat" -Force
+            Copy-Item -Path "$((Get-Location).Path)\files\startup\menu.ps1" -Destination "$imagePath\menu.ps1" -Force
             Write-Host "Startup commands changed"
         }
         catch
         {
             Write-Host "Could not change startup commands"
         }
+        try
+        {
+            Write-Host "CUSTOMIZATION STEP - Prepare System for Network-based Installations" -BackgroundColor DarkGreen
+            Write-Host "Preparing NetInstall..."
+            New-Item -Path "$imagePath\pxehelpers" -ItemType Directory | Out-Null
+            Copy-Item -Path "$((Get-Location).Path)\pxehelpers\*" -Destination "$imagePath\pxehelpers" -Verbose -Force -Recurse -Container -ErrorAction SilentlyContinue
+            Write-Host "The target system is now ready for network-based installations"
+        }
+        catch
+        {
+            Write-Host "Could not prepare the system for network-based installations"
+        }
         Write-Host "CUSTOMIZATION STEP - Set Scratch Size" -BackgroundColor DarkGreen
         Write-Host "Setting scratch size..."
         dism /English /image="$imagePath" /set-scratchspace=512 | Out-Host
@@ -836,7 +849,12 @@ function Start-OSApplication
         Write-Host "No Windows image has been found on this drive. An installation image is required. Exiting..."
         exit 1
     }
+    $diskGetterDpScript = @'
+    lis dis
+    exit
+'@
     New-Item -Path "X:\files\diskpart" -ItemType Directory -Force | Out-Null
+    $diskGetterDpScript | Out-File "X:\files\diskpart\dp_listdisk.dp" -Force -Encoding utf8
     $drive = Get-Disks
     if ($drive -eq "ERROR")
     {
@@ -1004,6 +1022,8 @@ function Start-OSApplication
     }
     New-BootFiles -drLetter $driveLetter -bootPart "auto" -diskId $drive -cleanDrive $($partition -eq 0)
     Start-Sleep -Milliseconds 250
+    Clear-Host
+    Write-Host "`n`n`n`n`n`n`n`n`n`n"
     Write-Host "The first stage of Setup has completed, and your system will reboot automatically."
     Write-Host "If there are any bootable devices, remove those before proceeding, as your system may boot to this environment again."
     Write-Host "When your computer restarts, Setup will continue."
@@ -1055,9 +1075,9 @@ function Get-Disks
     #>
 
     # Show disk list with diskpart
-    if (Test-Path .\files\diskpart\dp_listdisk.dp -PathType Leaf)
+    if (Test-Path "$env:SYSTEMDRIVE\files\diskpart\dp_listdisk.dp" -PathType Leaf)
     {
-        diskpart /s ".\files\diskpart\dp_listdisk.dp" | Out-Host
+        diskpart /s "$env:SYSTEMDRIVE\files\diskpart\dp_listdisk.dp" | Out-Host
     }
     else
     {

Helpers/extps1/PE_Helper/README.MD

@@ -0,0 +1,33 @@
+# The DISMTools Preinstallation Environment Helper (PE Helper)
+
+The Preinstallation Environment Helper is a set of scripts and applications designed to help during test deployments of your Windows images. It's used to create the ISO file and install your Windows image, whether it's a local deployment or a network-based one.
+
+To learn more about the PE Helper in detail, refer to the DISMTools Help documentation.
+
+## Folder Contents
+
+| Script | Description |
+|:--|:--:|
+| `PE_Helper.ps1` | The main event. The script generates the ISO file and installs the operating system |
+| `backgrounds` | The folder containing the background images used in the DISMTools Preinstallation Environment |
+| `files\dimstart.bat` | A script that manually starts the Driver Installation Module |
+| `files\startup\StartInstall.ps1` | A script that manually starts operating system installation from within the Extensibility Suite |
+| `files\startup\startnet.cmd` | **Required for Windows PE**. A script that starts the preinstallation environment |
+| `files\DISMTools-PE.zip` | A ZIP file containing Visual Studio solution files for the Extensibility Suite |
+| `files\HotInstall.zip` | DISMTools HotInstall, which lets you install the operating system from a full Windows installation |
+| `pxehelpers` | The DISMTools PXE Helpers, currently for Windows Deployment Services (WDS) |
+| `pxehelpers\wds` | The DISMTools PXE Helpers for Windows Deployment Services (WDS) |
+| `tools\DIM` | The folder containing the source code for the Driver Installation Module |
+
+## Script disclaimer
+
+The scripts present in this repository are provided **AS IS, WITHOUT ANY WARRANTY**. We AREN'T RESPONSIBLE for any damage done to your device.
+
+Just like any script, you should read the source code of these files if you're a programmer who knows Batch and PowerShell. Otherwise, run these with caution, or ask a programmer friend to examine the source code and provide a more detailed description of what a script does for you.
+
+## Credits/Copyright information
+
+- **Preinstallation Environment Helper, Extensibility Suite, Driver Installation Module:** (c) 2024-2025 CodingWonders Software
+- **HotInstall:** (c) 2025 CodingWonders Software
+- **PXE Helpers:** (c) 2025 CodingWonders Software
+- **Compilation Scripts:** (c) 2022 CT Tech Group LLC. Licensed under MIT

Helpers/extps1/PE_Helper/files/diskpart/dp_listdisk.dp

@@ -0,0 +1,39 @@
+Write-Host "Choose your preferred installation method:`n"
+Write-Host "  1 - Local Installation"
+Write-Host "      Choose this method if you started the Preinstallation Environment using local media, such as"
+Write-Host "      DVD or USB drives. This is recommended for newcomers."
+Write-Host "  2 - Network Installation"
+Write-Host "      Choose this method if you started the Preinstallation Environment using a network-based"
+Write-Host "      deployment solution. This is recommended for system administrators that want to deploy a system"
+Write-Host "      image to multiple computers at once."
+Write-Host "      Note that WDS is only supported at the moment." -ForegroundColor White
+Write-Host "  S - Shut down my computer"
+Write-Host "  R - Restart my computer`n"
+Write-Host "You will not be able to go back to choose another option after making your decision. You must reboot your"
+Write-Host "computer and select the correct option. You can also restart your computer by closing this window.`n"
+$option = Read-Host -Prompt "Choose an installation method by typing the option and pressing ENTER"
+switch ($option) {
+	"2" {
+		Clear-Host
+		Write-Host "Welcome to the Windows Deployment Services Technology Preview.`n"
+		Write-Host "The WDS Technology Preview 1 allows you to perform basic operating system deployment using a Windows Deployment"
+		Write-Host "Services server. The WDS Helper takes care of the installation process.`n"
+		Write-Host "The WDS Helper follows a client-server architecture. You are about to launch the client component, and it's required"
+		Write-Host "that you start the server component beforehand. You can find it in the `"pxehelpers\wds`" folder of"
+		Write-Host "this DVD/USB drive, which you can eject now. The required installation components are already loaded into memory.`n"
+		Write-Host "Since this is a technology preview, only basic testing has been made in order to make sure that you have a smooth"
+		Write-Host "experience. Expect to find issues, so don't hesitate to report feedback on this technology to make it better in"
+		Write-Host "future releases of DISMTools. If you didn't want to select this option, restart your computer.`n"
+		Write-Host "Once again, thanks for trying the WDS Technology Preview.`n`n"
+		Write-Host "                                  - CodingWonders Software`n`n"
+		Write-Host "Press ENTER to start the WDS Helper..."
+		Read-Host | Out-Null
+		New-Item -Path "$env:SYSTEMDRIVE\netinstall" | Out-Null
+	}
+	"S" {
+		Start-Process -FilePath "$env:WINDIR\system32\wpeutil.exe" -ArgumentList "shutdown" -NoNewWindow -Wait
+	}
+	"R" {
+		Start-Process -FilePath "$env:WINDIR\system32\wpeutil.exe" -ArgumentList "reboot" -NoNewWindow -Wait
+	}
+}

Helpers/extps1/PE_Helper/files/startup/menu.ps1

@@ -14,6 +14,13 @@ if %debug% equ 1 (
 	taskmgr
 )
 powershell -command Set-ExecutionPolicy Unrestricted
+if not exist "%sysdrive%\HotInstall" (
+	powershell -file "%sysdrive%\menu.ps1"
+	if exist "%sysdrive%\netinstall" (
+		cd /d "%sysdrive%"\
+		powershell -file ".\pxehelpers\wds\wdshelper.ps1"
+	)
+)
 if %debug% neq 2 if exist "%sysdrive%\HotInstall" (
 	echo Please insert the disc image and press ENTER...
 	pause > nul

Helpers/extps1/PE_Helper/files/startup/startnet.cmd

@@ -0,0 +1,42 @@
+param (
+    [switch]$Debug,
+    [switch]$Run,
+    [switch]$SkipPreprocessing,
+    [string]$Arguments
+)
+
+$workingdir = $PSScriptRoot
+
+Push-Location
+Set-Location $workingdir
+
+# Variable to sync between runspaces
+$sync = [Hashtable]::Synchronized(@{})
+$sync.PSScriptRoot = $workingdir
+$sync.configs = @{}
+
+function Update-Progress {
+    param (
+        [Parameter(Mandatory, position=0)]
+        [string]$StatusMessage,
+
+        [Parameter(Mandatory, position=1)]
+        [ValidateRange(0,100)]
+        [int]$Percent,
+
+        [Parameter(position=2)]
+        [string]$Activity = "Compiling"
+    )
+
+    Write-Progress -Activity $Activity -Status $StatusMessage -PercentComplete $Percent
+}
+
+Update-Progress "Pre-req: Running Preprocessor..." 0
+
+# Dot source the 'Invoke-Preprocessing' Function from 'tools/Invoke-Preprocessing.ps1' Script
+$preprocessingFilePath = ".\tools\Invoke-Preprocessing.ps1"
+. $preprocessingFilePath
+
+$excludedFiles = @("$preprocessingFilePath", '*.png', '*.exe')
+$msg = "Pre-req: Code Formatting"
+Invoke-Preprocessing -WorkingDir "$workingdir" -ExcludedFiles $excludedFiles -ProgressStatusMessage $msg -ThrowExceptionOnEmptyFilesList -SkipExcludedFilesValidation

Helpers/extps1/PE_Helper/pxehelpers/Compile.ps1

@@ -1,3 +1,5 @@
+using namespace System.Collections.Generic
+
 $global:product = ""
 $global:description = ""