Enable booting to PE

OS/init.cmd

@@ -33,6 +33,7 @@ set StagingDir=%TEMP%\SRBW
 set RemoteURL="https://git.mrmeeb.stream/MrMeeb/sure-recover-boot-wiper/raw/branch/main/payload"
 
 mkdir %StagingDir%
+mkdir C:\Sources
 
 powershell -Command Invoke-WebRequest %RemoteURL%/run.ps1 -OutFile %StagingDir%\run.ps1
 

payload/run.ps1

@@ -7,15 +7,17 @@ $StagingDir = $PSScriptRoot
 $ProgressPreference = 'SilentlyContinue'
 
 # Download compressed boot dir to Staging Dir
+Write-Host "Downloading boot.zip"
 Invoke-WebRequest -Uri $RemoteURL/boot.zip -OutFile $StagingDir\boot.zip
 
 # Expand boot.zip to C:\Boot
 Expand-Archive -Path $StagingDir\boot.zip -DestinationPath C:\ -Force
 
 # Download WinPE wim to C:\Sources\boot.wim
-New-Item -ItemType "Directory" -Path C:\Sources -Force
+Write-Host "Downloading boot.wim"
 Invoke-WebRequest -Uri $RemoteURL/boot.wim -OutFile C:\Sources\boot.wim
 
+# Setup booting to PE next boot
 
 function BootToPE {
 
@@ -77,4 +79,9 @@ function BootToPE {
     
 }
 
-Remove-Item -Path $StagingDir -Recurse
+$answer = read-host "Demo armed. Ready to reboot? (y/n)"
+if ($answer -eq 'y') { 
+    BootToPE
+} else {
+  Remove-Item -Path $StagingDir -Recurse
+}