From f5c20d724e5c7de327d5208e2bb7b162a596fcae Mon Sep 17 00:00:00 2001
From: pixrobot <pixrobot8@4d2.org>
Date: Tue, 2 Apr 2024 20:56:19 +0100
Subject: [PATCH] assign partition variables

---
 installer.sh | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/installer.sh b/installer.sh
index 876a878..baccdca 100755
--- a/installer.sh
+++ b/installer.sh
@@ -47,6 +47,17 @@ insmod /root/vfat-modules/vfat.ko.zst
 dialog --title "Welcome" --msgbox "Welcome to the EvolutionOS installer! \n\nThis installer is meant to be straightforward, no need for technical skill. \n\nPress enter to select buttons, tab to move between text boxes, and left / right to move between buttons." 0 0
 choice=$(dialog --menu "Select an installation drive:" 20 60 10 $(create_menu $(parted -l | grep -oE '^Disk /dev/[a-zA-Z0-9]+' | cut -d' ' -f2- | cut -c 6-)) 2>&1 >/dev/tty)
 
+if [[ ${choice} =~ "nvme" ]]; then
+    part1="p1"
+    part2="p2"
+elif [[ ${choice} =~ "mmcblk" ]]; then
+    part1="p1"
+    part2="p2"
+else
+    part1="1"
+    part2="2"
+fi
+
 if [[ "$(cat /sys/block/${choice}/ro)" = "1" ]]; then
     dialog --title "Scan disk" --msgbox "You are not able to install evolutionOS on a non-writable disk" 0 0
     failinstall