From 99ba0edf7933ec2449018f13fd7b85e6b2dc4869 Mon Sep 17 00:00:00 2001
From: Kroese <kroese@users.noreply.github.com>
Date: Sun, 5 May 2024 16:57:10 +0200
Subject: [PATCH] Update install.sh

---
 src/install.sh | 32 ++++++++++++++++++++------------
 1 file changed, 20 insertions(+), 12 deletions(-)

diff --git a/src/install.sh b/src/install.sh
index bb4fe5c..6a63c19 100644
--- a/src/install.sh
+++ b/src/install.sh
@@ -72,9 +72,9 @@ startInstall() {
   if [ -z "$CUSTOM" ]; then
     rm -f "$ISO"
   else
-    if [[ "$ISO" != "$CUSTOM_ORG" ]]; then
+    if [ -n "$ORIGINAL" ]; then
       rm -f "$ISO"
-      ISO="$CUSTOM_ORG"
+      ISO="$ORIGINAL"
       CUSTOM="$ISO"
     fi
   fi
@@ -145,11 +145,14 @@ abortInstall() {
 
   [[ "${iso,,}" == *".esd" ]] && exit 60
 
-  [ -n "$CUSTOM" ] && BOOT="$iso"
-
-  if [[ "$iso" != "$BOOT" ]]; then
-    if ! mv -f "$iso" "$BOOT"; then
-      error "Failed to move ISO file: $iso" && return 1
+  if [ -n "$CUSTOM" ]; then
+    BOOT="$iso"
+    REMOVE="N"
+  else
+    if [[ "$iso" != "$BOOT" ]]; then
+      if ! mv -f "$iso" "$BOOT"; then
+        error "Failed to move ISO file: $iso" && return 1
+      fi
     fi
   fi
 
@@ -163,7 +166,7 @@ detectCustom() {
   local size base
 
   CUSTOM=""
-  CUSTOM_ORG=""
+  ORIGINAL=""
 
   if [[ "${VERSION,,}" != "http"* ]]; then
     base="${VERSION/\/storage\//}"
@@ -186,12 +189,17 @@ detectCustom() {
   size="$(stat -c%s "$file")"
   [ -z "$size" ] || [[ "$size" == "0" ]] && return 0
 
-  CUSTOM_ORG="$file"
-
   base="$STORAGE/windows.$size.iso"
-  [ -f "$base" ] && [ -s "$base" ] && file="$base"
 
-  CUSTOM="$file"
+  if [ -f "$base" ] && [ -s "$base" ]; then
+    CUSTOM="$base"
+    ORIGINAL="$file"
+  else
+    REMOVE="N"
+    CUSTOM="$file"
+    rm -f "$base"
+  fi
+
   return 0
 }