From b5adafafe6576d9fdbdce8af0d039a5bbed3c3e4 Mon Sep 17 00:00:00 2001
From: Uraza <uraza@users.noreply.github.com>
Date: Tue, 11 Apr 2023 20:08:59 +0000
Subject: [PATCH] Partition names should be prefixed by "p" for NVMe

---
 script/config.sh | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/script/config.sh b/script/config.sh
index 4ed7602..b73ab5b 100644
--- a/script/config.sh
+++ b/script/config.sh
@@ -87,12 +87,20 @@ else
 export ISEFI=0  # No, BIOS
 fi # is EFI firmaare? 
 
+# Detect drive type.
+# For NVME drives, partition names should be prefixed by p.
+if [[ ${DEV} == *"nvme"* ]] ; then
+export PARTPREFIX="p"  # Yes, NVME
+else
+export PARTPREFIX=""   # No, regular
+fi
+
 # Set partition number based on the firmware type
 if [  ${ISEFI} -ne 0  ] ; then 
 # EFI firmware
-export EFIPARTITION=1
-export CRYPTPARTITION=2
+export EFIPARTITION=${PARTPREFIX}1
+export CRYPTPARTITION=${PARTPREFIX}2
 else
 # BIOS firmware
-export CRYPTPARTITION=1
+export CRYPTPARTITION=${PARTPREFIX}1
 fi  # EFI firmware