#!/bin/bash

#PS3="Please select a disk to partition:"
echo "Please select a disk to partition:"

# Don't separate values by space.
IFS=$'\n'

# Set variable containing name and size of all disks.
disks=( $(lsblk -d | tail -n+2 | awk '{print $1" "$4}') )

# Declare the array.
declare -a disks

# Select value on array.
select disk in ${disks[*]}
do
IFS=' '
array=($disk)
dev="${array[0]}"

echo "Partitioning /dev/$dev."
echo "Clearing existing partitioning table."
sgdisk -Z /dev/$dev &&
echo -e

size="${array[1]%?}"
echo "Creating boot partition of 256 MB."
sgdisk -n 1:0:+256M /dev/$dev &&
echo -e

echo "Creating system partition of $size GB."
sgdisk -n 2:0:0 /dev/$dev

echo "This is your new partition table:"
lsblk
break
done