Update zimaos.sh

scripts/vm/zimaos.sh

@@ -869,7 +869,6 @@ function select_storage_volume() {
 
 
 
-
 # ==========================================================
 # Create VM
 # ==========================================================
@@ -881,7 +880,7 @@ function create_vm() {
     -serial0 socket
   msg_ok "Create a $NAME"
 
-
+  BOOT_ORDER_LIST=()  # Array to store boot order for all disks
 
   # Check if UEFI (OVMF) is being used ===================
   if [[ "$BIOS_TYPE" == *"ovmf"* ]]; then
@@ -933,16 +932,11 @@ function create_vm() {
         fi
     fi
 
-
   fi
-# ==========================================================
-
 
 # Select storage volume for loader =======================
-
     LOADER_STORAGE=$(select_storage_volume $VMID "loader disk")
       
-
     #Run the command in the background and capture its PID
     qm importdisk $VMID ${LOADER_FILE} $LOADER_STORAGE > /tmp/import_log_$VMID.txt 2>&1 &
     import_pid=$!
@@ -967,10 +961,6 @@ function create_vm() {
           msg_ok "Loader imported successfully to ${CL}${BL}$LOADER_STORAGE${GN}${CL}"
     fi
 
-
-
-
- 
     STORAGE_TYPE=$(pvesm status -storage "$LOADER_STORAGE" | awk 'NR>1 {print $2}')
 
     if [[ "$STORAGE_TYPE" == "btrfs" || "$STORAGE_TYPE" == "dir" || "$STORAGE_TYPE" == "nfs" ]]; then
@@ -982,9 +972,8 @@ function create_vm() {
         if [[ -n "$IMPORTED_REF" && -n "$IMPORTED_ID" ]]; then
             if qm set "$VMID" -ide0 "$IMPORTED_REF" >/dev/null 2>&1; then
                 msg_ok "Configured loader disk as ide0"
-                qm set "$VMID" -delete "$IMPORTED_ID" >/dev/null 2>&1
             else
-                msg_error "Failed to assign loader disk to ide0"
+                msg_error "Failed to assign loader disk"
                 ERROR_FLAG=true
             fi
         else
@@ -1002,18 +991,6 @@ function create_vm() {
         fi
     fi
 
-
-
-
-    result=$(qm set "$VMID" -boot order=ide0 2>&1)
-    if [[ $? -eq 0 ]]; then
-          msg_ok "Loader configured as boot device."
-      else
-          ERROR_FLAG=true
-    fi
-
-# ==========================================================
-
     if [ "$DISK_TYPE" = "virtual" ]; then
         if [ ${#VIRTUAL_DISKS[@]} -eq 0 ]; then
             msg_error "No virtual disks configured."
@@ -1038,12 +1015,11 @@ if [ "$DISK_TYPE" = "virtual" ]; then
                     ;;
             esac
             
-
             DISK_NUM=$((i+1))
             DISK_NAME="vm-${VMID}-disk-${DISK_NUM}${DISK_EXT}"
             INTERFACE_ID="${INTERFACE_TYPE}$i"
             
-
+            # Create virtual disk
             if [[ "$STORAGE_TYPE" == "btrfs" || "$STORAGE_TYPE" == "dir" || "$STORAGE_TYPE" == "nfs" ]]; then
             
               msg_info "Creating virtual disk (format=raw) for $STORAGE_TYPE..."
@@ -1069,14 +1045,14 @@ if [ "$DISK_TYPE" = "virtual" ]; then
 
             msg_ok "Configured virtual disk as $INTERFACE_ID, ${SIZE}GB on ${CL}${BL}$STORAGE${CL} ${GN}"
 
+            BOOT_ORDER_LIST+=("$INTERFACE_ID")
             
-
+            # Add information to the description
             DISK_INFO="${DISK_INFO}<p>Virtual Disk $DISK_NUM: ${SIZE}GB on ${STORAGE} (${INTERFACE_TYPE}${i})</p>"
             CONSOLE_DISK_INFO="${CONSOLE_DISK_INFO}- Virtual Disk $DISK_NUM: ${SIZE}GB on ${STORAGE} ($INTERFACE_ID)\n"
         done
 
 
-    
     # HTML description
 HTML_DESC="<div align='center'>
 <table style='width: 100%; border-collapse: collapse;'>
@@ -1112,27 +1088,36 @@ ${DISK_INFO}
     msg_ok "Configured VM description"
 
 
-else
 
+    elif [ "$DISK_TYPE" = "passthrough" ]; then
+        if [ ${#PASSTHROUGH_DISKS[@]} -eq 0 ]; then
+            msg_error "No passthrough disks configured."
+            exit_script
+        fi
 
-      # Configure multiple passthrough disks
         DISK_INFO=""
         CONSOLE_DISK_INFO=""
 
         for i in "${!PASSTHROUGH_DISKS[@]}"; do
             DISK="${PASSTHROUGH_DISKS[$i]}"
-          DISK_MODEL=$(lsblk -ndo MODEL "$DISK" | xargs)
+            INTERFACE_ID="${INTERFACE_TYPE}$i"
-          DISK_SIZE=$(lsblk -ndo SIZE "$DISK" | xargs)
-          DISK_ID="sata$i"
             
+            # Get disk information
+            MODEL=$(lsblk -ndo MODEL "$DISK" 2>/dev/null || echo "Unknown")
+            SIZE=$(lsblk -ndo SIZE "$DISK" 2>/dev/null || echo "Unknown")
+            
+            # Configure passthrough disk
+            if qm set "$VMID" -$INTERFACE_ID "$DISK$DISCARD_OPTS" >/dev/null 2>&1; then
+                msg_ok "Configured passthrough disk as $INTERFACE_ID: $DISK ($MODEL $SIZE)"
+                BOOT_ORDER_LIST+=("$INTERFACE_ID")
                 
-          result=$(qm set $VMID -${DISK_ID} ${DISK} 2>&1)
-          if [[ $? -eq 0 ]]; then
-              msg_ok "Configured disk ${CL}${BL}($DISK_MODEL $DISK_SIZE)${CL}${GN} as $DISK_ID"
-          fi
                 # Add information to the description
-          DISK_INFO="${DISK_INFO}<p>Passthrough Disk $((i+1)): $DISK ($DISK_MODEL $DISK_SIZE)</p>"
+                DISK_INFO="${DISK_INFO}<p>Passthrough Disk $((i+1)): $DISK ($MODEL $SIZE) (${INTERFACE_TYPE}${i})</p>"
-          CONSOLE_DISK_INFO="${CONSOLE_DISK_INFO}- Passthrough Disk $((i+1)): $DISK ($DISK_MODEL $DISK_SIZE) (${DISK_ID})\n"
+                CONSOLE_DISK_INFO="${CONSOLE_DISK_INFO}- Passthrough Disk $((i+1)): $DISK ($MODEL $SIZE) ($INTERFACE_ID)\n"
+            else
+                msg_error "Failed to configure passthrough disk $DISK as $INTERFACE_ID"
+                ERROR_FLAG=true
+            fi
         done
 
 
@@ -1173,19 +1158,19 @@ ${DISK_INFO}
     fi
 
 
-if [ "$ERROR_FLAG" = true ]; then
+
-   msg_error "VM created with errors. Check configuration." 
+
+
+    if [ ${#BOOT_ORDER_LIST[@]} -gt 0 ]; then
+        BOOT_ORDER_LIST+=("ide0")  # Add IDE installer as last boot option
+        BOOT_ORDER_STRING=$(IFS=';'; echo "${BOOT_ORDER_LIST[*]}")
+        
+        if qm set "$VMID" -boot order="$BOOT_ORDER_STRING" >/dev/null 2>&1; then
+            msg_ok "Boot order configured: $BOOT_ORDER_STRING"
         else
-  msg_success "$(translate "Completed Successfully!")"
+            msg_error "Failed to configure boot order"
-
+            ERROR_FLAG=true
-  echo -e "${TAB}${GN}$(translate "Next Steps:")${CL}"
+        fi
-  echo -e "${TAB}1. $(translate "Start the VM")"
-  echo -e "${TAB}2. $(translate "Open the VM console and wait for the installer to boot")"
-  echo -e "${TAB}3. $(translate "Complete the ZimaOS installation wizard")"
-  echo -e "${TAB}4. $(translate "After installation, remove or disable the installer IDE disk")"
-  echo -e
-
-
     fi
 
 }