From 50d97d7adbed4913a565a829faf951d0a560ad0e Mon Sep 17 00:00:00 2001
From: Kroese <kroese@users.noreply.github.com>
Date: Wed, 2 Apr 2025 23:25:39 +0200
Subject: [PATCH] Update install.sh

---
 src/install.sh | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/install.sh b/src/install.sh
index 2f78a33..698a046 100644
--- a/src/install.sh
+++ b/src/install.sh
@@ -203,6 +203,7 @@ detectCustom() {
   local dir file base
   local fname="custom.iso"
   local boot="$STORAGE/windows.boot"
+  local previous="$STORAGE/windows.base"
 
   CUSTOM=""
 
@@ -210,9 +211,15 @@ detectCustom() {
   [ ! -d "$dir" ] && dir=$(find "$STORAGE" -maxdepth 1 -type d -iname "$fname" | head -n 1)
 
   if [ -d "$dir" ]; then
-    if ! hasDisk || [ ! -f "$boot" ]; then
+    if ! hasDisk || [ ! -f "$boot" ] || [ ! -f "$previous" ]; then
       error "The bind $dir maps to a file that does not exist!" && return 1
     fi
+    previous=$(<"$previous")
+    previous="${previous//[![:print:]]/}"
+    ISO="$dir"
+    CUSTOM="$ISO"
+    BOOT="$STORAGE/$previous"
+    return 0
   fi
 
   file=$(find / -maxdepth 1 -type f -iname "$fname" | head -n 1)