Added update section

* Needs testing

.bin/Scripts/build_pe.ps1

@@ -10,7 +10,7 @@ $Host.UI.RawUI.WindowTitle = "Wizard Kit: Windows PE Build Tool"
 $WD = $(Split-Path $MyInvocation.MyCommand.Path)
 $Bin = (Get-Item $WD -Force).Parent.FullName
 $Root = (Get-Item $Bin -Force).Parent.FullName
-$Temp = "{0}\tmp" -f $Bin
+$Temp = "$Bin\tmp"
 $Date = Get-Date -UFormat "%Y-%m-%d"
 $Host.UI.RawUI.BackgroundColor = "Black"
 $Host.UI.RawUI.ForegroundColor = "White"
@@ -106,30 +106,29 @@ function MakeClean {
 function DownloadFile ($Path, $Name, $Url) {
     $OutFile = "{0}\{1}" -f $Path, $Name
 
-    Write-Host ("Downloading: {0}" -f $Name)
+    Write-Host ("Downloading: $Name")
     New-Item -Type Directory $Path 2>&1 | Out-Null
     try {
-        Invoke-Webrequest -Uri $Url -OutFile $OutFile
+        Invoke-WebRequest -Uri $Url -OutFile $OutFile
     }
     catch {
         Write-Host ("  ERROR: Failed to download file." ) -ForegroundColor "Red"
+        $DownloadErrors += 1
     }
 }
 function FindDynamicUrl ($SourcePage, $RegEx) {
-    $Url = ""
-
     # Get source page
     Invoke-Webrequest -Uri $SourcePage -OutFile "tmp_page"
 
     # Search for real url
     $Url = Get-Content "tmp_page" | Where-Object {$_ -imatch $RegEx}
-    $Url = $Url -ireplace '.*(a |)href="([^"]+)".*', "$2"
+    $Url = $Url -ireplace '.*(a |)href="([^"]+)".*', '$2'
-    $Url = $Url -ireplace ".*(a |)href='([^']+)'.*", "$2"
+    $Url = $Url -ireplace ".*(a |)href='([^']+)'.*", '$2'
 
     # Remove tmp_page
     Remove-Item "tmp_page"
 
-    return $Url
+    $Url | Select-Object -First 1
 }
 function WKPause ($Message = "Press Enter to continue... ") {
     Write-Host $Message -NoNewLine
@@ -143,7 +142,7 @@ function WKPause ($Message = "Press Enter to continue... ") {
 # Asked by:     https://stackoverflow.com/users/65164/mark-mascolino
 # Answer by:    https://stackoverflow.com/users/696808/bacon-bits
 if ($MyInvocation.InvocationName -ne ".") {
-    # Clear-Host
+    Clear-Host
     Write-Host "Wizard Kit: Windows PE Build Tool`n"
     
     ## Prep ##
@@ -157,6 +156,147 @@ if ($MyInvocation.InvocationName -ne ".") {
     Push-Location "$WD"
     MakeClean
     
+    if (Ask-User "Update Tools?") {
+        $DownloadErrors = 0
+        $Path = $Temp
+        
+        ## Download Tools ##
+        # 7-Zip
+        DownloadFile -Path $Path -Name "7z-installer.msi" -Url "http://www.7-zip.org/a/7z1701.msi"
+        DownloadFile -Path $Path -Name "7z-extra.7z" -Url "http://www.7-zip.org/a/7z1701-extra.7z"
+
+        # ConEmu
+        $Url = "https://github.com/Maximus5/ConEmu/releases/download/v17.11.09/ConEmuPack.171109.7z"
+        DownloadFile -Path $Path -Name "ConEmuPack.7z" -Url $Url
+
+        # Notepad++
+        $Url = "https://notepad-plus-plus.org/repository/7.x/7.5.2/npp.7.5.2.bin.minimalist.x64.7z"
+        DownloadFile -Path $Path -Name "nppamd64.7z" -Url $Url
+        $Url = "https://notepad-plus-plus.org/repository/7.x/7.5.2/npp.7.5.2.bin.minimalist.7z"
+        DownloadFile -Path $Path -Name "nppx86.7z" -Url $Url
+
+        # Python
+        $Url = "https://www.python.org/ftp/python/3.6.3/python-3.6.3-embed-win32.zip"
+        DownloadFile -Path $Path -Name "python32.zip" -Url $Url
+        $Url = "https://www.python.org/ftp/python/3.6.3/python-3.6.3-embed-amd64.zip"
+        DownloadFile -Path $Path -Name "python64.zip" -Url $Url
+
+        # Python: psutil
+        $DownloadPage = "https://pypi.python.org/pypi/psutil"
+        $RegEx = "href=.*-cp36-cp36m-win32.whl"
+        $Url = FindDynamicUrl $DownloadPage $RegEx
+        DownloadFile -Path $Path -Name "psutil32.whl" -Url $Url
+        $RegEx = "href=.*-cp36-cp36m-win_amd64.whl"
+        $Url = FindDynamicUrl $DownloadPage $RegEx
+        DownloadFile -Path $Path -Name "psutil64.whl" -Url $Url
+    
+        ## Bail ##
+        # If errors were encountered during downloads
+        if ($DownloadErrors -gt 0) {
+            Abort
+        }
+        
+        ## Extract ##
+        # 7-Zip
+        Write-Host "Extracting: 7-Zip"
+        try {
+            $ArgumentList = @("/a", "$Temp\7z-installer.msi", "TARGETDIR=$Temp\7zi", "/qn")
+            Start-Process -FilePath "$System32\msiexec.exe" -ArgumentList $ArgumentList -Wait
+            $SevenZip = "$Temp\7zi\Files\7-Zip\7z.exe"
+            $ArgumentList = @(
+                "e", "$Temp\7z-extra.7z", "-o$Root\WK\amd64\7-Zip",
+                "-aoa", "-bso0", "-bse0", "-bsp0",
+                "x64\7za.exe", "*.txt")
+            Start-Process -FilePath $SevenZip -ArgumentList $ArgumentList -NoNewWindow -Wait
+            $ArgumentList = @(
+                "e", "$Temp\7z-extra.7z", "-o$Root\WK\x86\7-Zip",
+                "-aoa", "-bso0", "-bse0", "-bsp0",
+                "7za.exe", "*.txt")
+            Start-Process -FilePath $SevenZip -ArgumentList $ArgumentList -NoNewWindow -Wait
+            Remove-Item "$Temp\7z*" -Recurse
+            $SevenZip = "$Root\WK\x86\7-Zip\7za.exe"
+        }
+        catch {
+            Write-Host ("  ERROR: Failed to extract files." ) -ForegroundColor "Red"
+        }
+
+        # Notepad++
+        Write-Host "Extracting: Notepad++"
+        try {
+            $ArgumentList = @(
+                "x", "$Temp\nppamd64.7z", "-o$Root\WK\amd64\NotepadPlusPlus",
+                "-aoa", "-bso0", "-bse0", "-bsp0")
+            Start-Process -FilePath $SevenZip -ArgumentList $ArgumentList -NoNewWindow -Wait
+            $ArgumentList = @(
+                "x", "$Temp\nppx86.7z", "-o$Root\WK\x86\NotepadPlusPlus",
+                "-aoa", "-bso0", "-bse0", "-bsp0")
+            Start-Process -FilePath $SevenZip -ArgumentList $ArgumentList -NoNewWindow -Wait
+            Remove-Item "$Temp\npp*.7z"
+            Move-Item "$Root\WK\amd64\NotepadPlusPlus\notepad++.exe" "$Root\WK\amd64\NotepadPlusPlus\notepadplusplus.exe"
+            Move-Item "$Root\WK\x86\NotepadPlusPlus\notepad++.exe" "$Root\WK\x86\NotepadPlusPlus\notepadplusplus.exe"
+        }
+        catch {
+            Write-Host ("  ERROR: Failed to extract files." ) -ForegroundColor "Red"
+        }
+
+        # ConEmu
+        Write-Host "Extracting: ConEmu"
+        try {
+            $ArgumentList = @(
+                "x", "$Temp\ConEmuPack.7z", "-o$Root\WK\amd64\ConEmu",
+                "-aoa", "-bso0", "-bse0", "-bsp0")
+            Start-Process -FilePath $SevenZip -ArgumentList $ArgumentList -NoNewWindow -Wait
+            Remove-Item "$Root\WK\amd64\ConEmu\ConEmu.exe"
+            Remove-Item "$Root\WK\amd64\ConEmu\ConEmu.map"
+            Move-Item "$Root\WK\amd64\ConEmu\ConEmu64.exe" "$Root\WK\amd64\ConEmu\ConEmu.exe"
+            Move-Item "$Root\WK\amd64\ConEmu\ConEmu64.map" "$Root\WK\amd64\ConEmu\ConEmu.map"
+            $ArgumentList = @(
+                "x", "$Temp\ConEmuPack.7z", "-o$Root\WK\x86\ConEmu",
+                "-aoa", "-bso0", "-bse0", "-bsp0")
+            Start-Process -FilePath $SevenZip -ArgumentList $ArgumentList -NoNewWindow -Wait
+            Remove-Item "$Root\WK\x86\ConEmu\ConEmu64.exe"
+            Remove-Item "$Root\WK\x86\ConEmu\ConEmu64.map"
+            Remove-Item "$Temp\ConEmuPack.7z"
+        }
+        catch {
+            Write-Host ("  ERROR: Failed to extract files." ) -ForegroundColor "Red"
+        }
+
+        # Python
+        Write-Host "Extracting: Python"
+        try {
+            $ArgumentList = @(
+                "x", "$Temp\python64.zip", "-o$Root\WK\amd64\python",
+                "-aoa", "-bso0", "-bse0", "-bsp0")
+            Start-Process -FilePath $SevenZip -ArgumentList $ArgumentList -NoNewWindow -Wait
+            $ArgumentList = @(
+                "x", "$Temp\python32.zip", "-o$Root\WK\x86\python",
+                "-aoa", "-bso0", "-bse0", "-bsp0")
+            Start-Process -FilePath $SevenZip -ArgumentList $ArgumentList -NoNewWindow -Wait
+            Remove-Item "$Temp\python*.zip"
+        }
+        catch {
+            Write-Host ("  ERROR: Failed to extract files." ) -ForegroundColor "Red"
+        }
+
+        # Python: psutil
+        Write-Host "Extracting: Python"
+        try {
+            $ArgumentList = @(
+                "x", "$Temp\psutil64.whl", "-o$Root\WK\amd64\python",
+                "-aoa", "-bso0", "-bse0", "-bsp0")
+            Start-Process -FilePath $SevenZip -ArgumentList $ArgumentList -NoNewWindow -Wait
+            $ArgumentList = @(
+                "x", "$Temp\psutil32.whl", "-o$Root\WK\x86\python",
+                "-aoa", "-bso0", "-bse0", "-bsp0")
+            Start-Process -FilePath $SevenZip -ArgumentList $ArgumentList -NoNewWindow -Wait
+            Remove-Item "$Temp\*.whl"
+        }
+        catch {
+            Write-Host ("  ERROR: Failed to extract files." ) -ForegroundColor "Red"
+        }
+    }
+    
     ## Build ##
     foreach ($Arch in @("amd64", "x86")) {
         $Drivers = "$Root\Drivers\%arch"
@@ -166,7 +306,7 @@ if ($MyInvocation.InvocationName -ne ".") {
         # Copy WinPE files
         Write-Host "Copying files..."
         $Cmd = ("{0}\copype.cmd" -f $Env:WinPERoot)
-        Start-Process $Cmd -ArgumentList @($Arch, $PEFiles) -NoNewWindow -Wait
+        Start-Process -FilePath $Cmd -ArgumentList @($Arch, $PEFiles) -NoNewWindow -Wait
         
         # Remove unwanted items
         foreach ($SubDir in @("media", "media\Boot", "media\EFI\Microsoft\Boot")) {
@@ -195,7 +335,7 @@ if ($MyInvocation.InvocationName -ne ".") {
             ('/Image:"{0}"' -f $Mount),
             "/Set-ScratchSpace:512"
         )
-        Start-Process $DISM -ArgumentList $ArgumentList -NoNewWindow -Wait
+        Start-Process -FilePath $DISM -ArgumentList $ArgumentList -NoNewWindow -Wait
         
         # Add WK tools
         Write-Host "Copying tools..."
@@ -211,30 +351,33 @@ if ($MyInvocation.InvocationName -ne ".") {
         Copy-Item -Path "$Root\Scripts" -Destination "$Mount\WK\Scripts" -Recurse -Force
         
         # Add System32 items
+        $HostSystem32 = "{0}\System32" -f $Env:SystemRoot
         Copy-Item -Path "$Root\System32\*" -Destination "$Mount\Windows\System32" -Recurse -Force
         $ArgumentList = @("/f", "$Mount\Windows\System32\winpe.jpg", "/a")
-        Start-Process "C:\Windows\System32\takeown.exe" -ArgumentList $ArgumentList -NoNewWindow -Wait
+        Start-Process -FilePath "$HostSystem32\takeown.exe" -ArgumentList $ArgumentList -NoNewWindow -Wait
         $ArgumentList = @("$Mount\Windows\System32\winpe.jpg", "/grant", "Administrators:F")
-        Start-Process "C:\Windows\System32\icacls.exe" -ArgumentList $ArgumentList -NoNewWindow -Wait
+        Start-Process -FilePath "$HostSystem32\icacls.exe" -ArgumentList $ArgumentList -NoNewWindow -Wait
         Copy-Item -Path "$Root\WinPE.jpg" -Destination "$Mount\Windows\System32\winpe.jpg" -Recurse -Force
         
-        # Update registry
+        # Load registry hives
         Write-Host "Updating Registry..."
-        $Reg = "C:\Windows\System32\reg.exe"
+        $Reg = "$HostSystem32\reg.exe"
-        Start-Process $Reg -ArgumentList @("load", "HKLM\WinPE-SW", "$Mount\Windows\System32\config\SOFTWARE") -NoNewWindow -Wait
+        $ArgumentList = @("load", "HKLM\WinPE-SW", "$Mount\Windows\System32\config\SOFTWARE")
-        Start-Process $Reg -ArgumentList @("load", "HKLM\WinPE-SYS", "$Mount\Windows\System32\config\SYSTEM") -NoNewWindow -Wait
+        Start-Process -FilePath $Reg -ArgumentList $ArgumentList -NoNewWindow -Wait
+        $ArgumentList = @("load", "HKLM\WinPE-SYS", "$Mount\Windows\System32\config\SYSTEM")
+        Start-Process -FilePath $Reg -ArgumentList $ArgumentList -NoNewWindow -Wait
         
-            # Add 7-Zip and Python to path
+        # Add tools to path
-            $RegPath = "HKLM:\WinPE-SYS\ControlSet001\Control\Session Manager\Environment"
+        $RegPath = "HKLM:\WinPE-SYS\ControlSet001\Control\Session Manager\Environment"
-            $RegKey = Get-ItemProperty -Path $RegPath
+        $RegKey = Get-ItemProperty -Path $RegPath
-            $NewValue = "{0};%SystemDrive%\WK\7-Zip;%SystemDrive%\WK\python;%SystemDrive%\WK\wimlib" -f $RegKey.Path
+        $NewValue = "{0};%SystemDrive%\WK\7-Zip;%SystemDrive%\WK\python;%SystemDrive%\WK\wimlib" -f $RegKey.Path
-            Set-ItemProperty -Path $RegPath -Name "Path" -Value $NewValue -Force | Out-Null
+        Set-ItemProperty -Path $RegPath -Name "Path" -Value $NewValue -Force | Out-Null
-            
+        
-            # Replace Notepad
+        # Replace Notepad
-            $RegPath = "HKLM:\WinPE-SW\Microsoft\Windows NT\CurrentVersion\Image File Execution Options\notepad.exe"
+        $RegPath = "HKLM:\WinPE-SW\Microsoft\Windows NT\CurrentVersion\Image File Execution Options\notepad.exe"
-            $NewValue = 'wscript "X:\WK\NotepadPlusPlus\npp.vbs"'
+        $NewValue = 'wscript "X:\WK\NotepadPlusPlus\npp.vbs"'
-            New-Item -Path $RegPath -Force | Out-Null
+        New-Item -Path $RegPath -Force | Out-Null
-            New-ItemProperty -Path $RegPath -Name "Debugger" -Value $NewValue -Force | Out-Null
+        New-ItemProperty -Path $RegPath -Name "Debugger" -Value $NewValue -Force | Out-Null
         
         # Run garbage collection to release potential stale handles
         ## Credit: https://jrich523.wordpress.com/2012/03/06/powershell-loading-and-unloading-registry-hives/
@@ -243,8 +386,8 @@ if ($MyInvocation.InvocationName -ne ".") {
 
         # Unload registry hives
         Start-Sleep -Seconds 2
-        Start-Process $Reg -ArgumentList @("unload", "HKLM\WinPE-SW") -NoNewWindow -Wait
+        Start-Process -FilePath $Reg -ArgumentList @("unload", "HKLM\WinPE-SW") -NoNewWindow -Wait
-        Start-Process $Reg -ArgumentList @("unload", "HKLM\WinPE-SYS") -NoNewWindow -Wait
+        Start-Process -FilePath $Reg -ArgumentList @("unload", "HKLM\WinPE-SYS") -NoNewWindow -Wait
         
         # Unmount image
         Write-Host "Dismounting image..."
@@ -253,7 +396,7 @@ if ($MyInvocation.InvocationName -ne ".") {
         # Create ISO
         $ArgumentList = @("/iso", $PEFiles, "$Root\wk-winpe-$Date-$Arch.iso")
         $Cmd = "{0}\MakeWinPEMedia.cmd" -f $Env:WinPERoot
-        Start-Process $Cmd -ArgumentList $ArgumentList -NoNewWindow -Wait
+        Start-Process -FilePath $Cmd -ArgumentList $ArgumentList -NoNewWindow -Wait
     }
 
     ## Done ##