From a575d2229fa75c431c11212d0bc4a7aeed23b6ae Mon Sep 17 00:00:00 2001
From: Kroese <kroese@users.noreply.github.com>
Date: Sun, 28 Apr 2024 19:25:05 +0200
Subject: [PATCH] Update install.sh

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

diff --git a/src/install.sh b/src/install.sh
index 0979925..f05b9a3 100644
--- a/src/install.sh
+++ b/src/install.sh
@@ -328,9 +328,11 @@ downloadImage() {
 
   if [ -z "$url" ]; then
 
-    doMido "$iso" "$version" "$desc" && return 0
+    if isMido "$version"; then
+      doMido "$iso" "$version" "$desc" && return 0
+    fi
 
-    if [[ "${version,,}" == "win10${PLATFORM,,}" ]] || [[ "$version" == "win11${PLATFORM,,}" ]]; then
+    if isESD "$version"; then
 
       info "Failed to download $desc using Mido, will try a different method now..."
 
@@ -353,7 +355,11 @@ downloadImage() {
   [[ "${version,,}" == "http"* ]] && return 1
 
   url=$(getLink "$version")
-  [ -z "$url" ] && return 1
+  
+  if [ -z "$url" ]; then
+    ! isMido && error "Invalid version specified: $version"
+    return 1
+  fi
     
   info "Failed to download $desc from Microsoft, will try another mirror now..."