From a7d5eac8767faa22d6f1e111a50b8f77cf255160 Mon Sep 17 00:00:00 2001
From: Kroese <kroese@users.noreply.github.com>
Date: Mon, 10 Jun 2024 19:51:51 +0200
Subject: [PATCH] Update disk.sh

---
 src/disk.sh | 29 +++++++++++------------------
 1 file changed, 11 insertions(+), 18 deletions(-)

diff --git a/src/disk.sh b/src/disk.sh
index 19b87b0..c66eb17 100644
--- a/src/disk.sh
+++ b/src/disk.sh
@@ -532,19 +532,11 @@ case "${DISK_TYPE,,}" in
 esac
 
 if [ -z "${MEDIA_TYPE:-}" ]; then
-  case "${DISK_TYPE,,}" in
-    "ide" | "usb" | "scsi" )
-      if [[ "${MACHINE,,}" == "virt" ]]; then
-        MEDIA_TYPE="auto"
-      else
-        MEDIA_TYPE="$DISK_TYPE"
-      fi ;;
-    "blk" | "auto" )
-      if [[ "${MACHINE,,}" != "virt" ]] && [[ "${MACHINE,,}" != "pc-i440fx-2"* ]]; then
-        MEDIA_TYPE="ide"
-      else
-        MEDIA_TYPE="auto"
-      fi ;;
+  case "${MACHINE,,}" in
+    "virt" | "pc-q35-2"* | "pc-i440fx-2"* )
+      MEDIA_TYPE="auto" ;;
+    * )
+      [[ "${DISK_TYPE,,}" != "blk" ]] && MEDIA_TYPE="$DISK_TYPE" || MEDIA_TYPE="ide" ;;
   esac
 fi
 
@@ -561,11 +553,12 @@ DRIVERS="/drivers.iso"
 [ ! -f "$DRIVERS" ] || [ ! -s "$DRIVERS" ] && DRIVERS="$STORAGE/drivers.iso"
 
 if [ -f "$DRIVERS" ] && [ -s "$DRIVERS" ]; then
-  if [[ "${MACHINE,,}" != "virt" ]] && [[ "${MACHINE,,}" != "pc-i440fx-2"* ]]; then
-    DRIVER_TYPE="ide"
-  else
-    DRIVER_TYPE="auto"
-  fi
+  case "${MACHINE,,}" in
+    "virt" | "pc-q35-2"* | "pc-i440fx-2"* )
+      DRIVER_TYPE="auto" ;;
+    * )
+      DRIVER_TYPE="ide" ;;
+  esac
   DISK_OPTS+=$(addMedia "$DRIVERS" "$DRIVER_TYPE" "1" "" "0x6")
 fi