From 2596d54e282de6f3f849ac8424d0b4428c175c0a Mon Sep 17 00:00:00 2001
From: Aaron  Honeycutt <aaronhoneycutt@protonmail.com>
Date: Mon, 17 Apr 2023 18:38:43 +0000
Subject: [PATCH] Update install.sh

---
 install.sh | 25 +++++++++++++++++++------
 1 file changed, 19 insertions(+), 6 deletions(-)

diff --git a/install.sh b/install.sh
index a282230..55fbacd 100644
--- a/install.sh
+++ b/install.sh
@@ -36,18 +36,31 @@ echo w       # write changes.
 ) | sudo fdisk $driveName -w always -W always
 
 if [ $driveName == /dev/nvme* ]; then
-  # Format the NVMe drive with the ext4 filesystem
-  mkfs.ext4 $DEVICE_PATH
+  # Set variables
+  efiName="${driveName}p1"
+  rootName="${driveName}p2"
+  swapName="${driveName}p3"
 
 # Check if the device is a SATA drive
 elif [ $driveName == /dev/sd* ]; then
-  # Format the SATA drive with the NTFS filesystem
-  mkfs.ntfs $DEVICE_PATH
+  # Set variables
+  efiName="${driveName}1"
+  rootName="${driveName}2"
+  swapName="${driveName}3"
+
+# Check if the device is a Virtual drive
+elif [ $driveName == /dev/vd* ]; then
+  # Set variables
+  efiName="${driveName}1"
+  rootName="${driveName}2"
+  swapName="${driveName}3"
 
 # Check if the device is an eMMC drive
 elif [ $driveName == /dev/mmcblk* ]; then
-  # Format the eMMC drive with the FAT32 filesystem
-  mkfs.fat -F 32 $DEVICE_PATH
+  # Set variables
+  efiName="${driveName}1"
+  rootName="${driveName}2"
+  swapName="${driveName}3"
 
 fi