#!/bin/bash

Define variables

DISKS=$(lsblk -d -o NAME,SIZE | awk '$2!~/M$/ && $2!=0 {print $1}') PARTITIONS=$(lsblk -d -o NAME,SIZE | awk '$2~/M$/ && $2!=0 {print $1}') MOUNT_POINTS=$(df -h | awk '{print $6}' | sed '1d') UUIDS=$(blkid -o value -s UUID $PARTITIONS)

Define functions

select_disk() { DISK=$(whiptail --menu "Select a disk to partition and format:" 0 0 0
$(for disk in $DISKS; do echo "$disk $(( $(lsblk -d -o SIZE -b /dev/$disk) / 1024 / 1024 / 1024 ))GB"; done)
3>&1 1>&2 2>&3) if [ $? -ne 0 ]; then exit 1; fi }

create_partitions() { whiptail --yesno "Do you want to create a single partition on $DISK?" 0 0
--yes-button "Yes" --no-button "No" if [ $? -eq 0 ]; then PARTITION="$DISK"1 parted /dev/$DISK mklabel gpt mkpart primary ext4 0% 100% else whiptail --msgbox "You will need to manually create the partitions." 0 0 exit 1 fi }

format_partitions() { FILESYSTEM=$(whiptail --menu "Select a filesystem for the partition:" 0 0 0
"ext2" ""
"ext3" ""
"ext4" ""
"fat32" ""
"ntfs" ""
3>&1 1>&2 2>&3) if [ $? -ne 0 ]; then exit 1; fi mkfs.$FILESYSTEM /dev/$PARTITION }

mount_partitions() { MOUNT_POINT=$(whiptail --inputbox "Enter a mount point for the partition:" 0 0 "/"
3>&1 1>&2 2>&3) if [ $? -ne 0 ]; then exit 1; fi if [ -d $MOUNT_POINT ]; then whiptail --yesno "The directory $MOUNT_POINT already exists. Do you want to use it as the mount point?" 0 0
--yes-button "Yes" --no-button "No" if [ $? -ne 0 ]; then exit 1; fi else mkdir -p $MOUNT_POINT fi echo "/dev/$PARTITION $MOUNT_POINT $FILESYSTEM defaults 0 0" >> /etc/fstab mount -a }

uuid_mount() { UUID=$(whiptail --menu "Select a partition to mount by UUID:" 0 0 0
$(for uuid in $UUIDS; do echo "$uuid $(lsblk -d -o NAME,SIZE | grep $(blkid -o value -s UUID=$uuid) | awk '{print $1}')"; done)
3>&1 1>&2 2>&3) if [ $? -ne 0 ]; then exit 1; fi if grep -q "$UUID" /etc/fstab; then whiptail --msgbox "The partition is already set to mount by UUID." 0 0 exit 1 else echo "UUID=$UUID $(whiptail --inputbox "Enter a mount point for the partition:" 0 0 "/"
3>&1 1>&2 2>&3) $FILESYSTEM defaults 0 0" >> /etc/fstab mount -a fi }

Main program

select_disk create_partitions format_partitions mount_partitions uuid_mount

exit