From b0db11cb4acca0897c3a58345c32ac379ab3cb7d Mon Sep 17 00:00:00 2001
From: Alan Mason <1923621+2Shirt@users.noreply.github.com>
Date: Thu, 30 Nov 2017 16:50:14 -0800
Subject: [PATCH] Added wimlib-imagex

Woops, it's been missing for a bit..
---
 .bin/Scripts/build_pe.ps1 | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/.bin/Scripts/build_pe.ps1 b/.bin/Scripts/build_pe.ps1
index 631d6d9e..fcc01980 100644
--- a/.bin/Scripts/build_pe.ps1
+++ b/.bin/Scripts/build_pe.ps1
@@ -162,7 +162,10 @@ if ($MyInvocation.InvocationName -ne ".") {
             @("qdir32.zip", "https://www.softwareok.com/Download/Q-Dir_Portable.zip"),
             # TestDisk / PhotoRec
             @("testdisk64.zip", "https://www.cgsecurity.org/testdisk-7.1-WIP.win64.zip"),
-            @("testdisk32.zip", "https://www.cgsecurity.org/testdisk-7.1-WIP.win.zip")
+            @("testdisk32.zip", "https://www.cgsecurity.org/testdisk-7.1-WIP.win.zip"),
+            # wimlib-imagex
+            @("wimlib64.zip", "https://wimlib.net/downloads/wimlib-1.12.0-windows-x86_64-bin.zip"),
+            @("wimlib32.zip", "https://wimlib.net/downloads/wimlib-1.12.0-windows-i686-bin.zip")
         )
         foreach ($Tool in $ToolSources) {
             DownloadFile -Path $Temp -Name $Tool[0] -Url $Tool[1]
@@ -444,6 +447,22 @@ if ($MyInvocation.InvocationName -ne ".") {
         catch {
             Write-Host ("  ERROR: Failed to extract files." ) -ForegroundColor "Red"
         }
+        
+        # wimlib-imagex
+        try {
+            $ArgumentList = @(
+                "x", "$Temp\wimlib64.zip", "-o$Root\WK\amd64\wimlib",
+                "-aoa", "-bso0", "-bse0", "-bsp0")
+            Start-Process -FilePath $SevenZip -ArgumentList $ArgumentList -NoNewWindow -Wait
+            $ArgumentList = @(
+                "x", "$Temp\wimlib32.zip", "-o$Root\WK\x86\wimlib",
+                "-aoa", "-bso0", "-bse0", "-bsp0")
+            Start-Process -FilePath $SevenZip -ArgumentList $ArgumentList -NoNewWindow -Wait
+            Remove-Item "$Temp\wimlib*"
+        }
+        catch {
+            Write-Host ("  ERROR: Failed to extract files." ) -ForegroundColor "Red"
+        }
     }
     
     ## Build ##