Update install-arch.sh

install_scripts/install-arch.sh

@@ -5,6 +5,7 @@ YELLOW='\033[0;33m'
 RED='\033[0;31m'
 
 disk="/dev/sda"
+partition_prefix=""
 efi_part_size=500
 swap_size=8192
 hostname="arch"
@@ -46,17 +47,27 @@ partitioning_disk() {
 }
 
 formatting_partitions() {
-    mkfs.vfat -F 32 ${disk}1
+    mkfs.vfat -F 32 ${disk}${partition_prefix}1
-    mkswap ${disk}3
+    mkswap ${disk}${partition_prefix}3
-    [ "$rootfstype" == "ext4" ] && mkfs.ext4 ${disk}4
+    case "$rootfstype" in
-    [ "$rootfstype" == "btrfs" ] && mkfs.btrfs -L arch -f ${disk}4
+        ext4)
+            mkfs.ext4 -L arch ${disk}${partition_prefix}4
+            ;;
+        btrfs)
+            mkfs.btrfs -L arch -f ${disk}${partition_prefix}4
+            ;;
+        *)
+            printf "$RED ERROR:$RESET Unsupported filesystem type: $rootfstype\n"
+            exit 1
+            ;;
+    esac
 }
 
 mount_partitions() {
-    mount ${disk}4 /mnt
+    mount ${disk}${partition_prefix}4 /mnt
     mkdir -p /mnt/boot/EFI
-    mount ${disk}1 /mnt/boot/EFI
+    mount ${disk}${partition_prefix}1 /mnt/boot/EFI
-    swapon ${disk}3
+    swapon ${disk}${partition_prefix}3
 }
 
 install_base_system() {