#!/bin/sh

######################################################################
# @author       : swytch (adapted form Luke Smith - lukesmith.xyz)
# @file         : dmenuumount
# @license      : MIT
# @created      : Wednesday May 20, 2020 18:11:17 CEST
#
# @description  : dmenu prompt to unmount drives
######################################################################


unmountusb() {
	chosen=$(echo "$usbdrives" | dmenu -i -p "Unmount which drive?" | awk '{print $1}')
if [ -z "$chosen" ]; then
	exit
else
	if sudo -A umount "$chosen" ; then
		notify-send "禍 USB unmounting" "$chosen unmounted."
	else
		notify-send -u "critical" -t 3000 "禍 USB ERROR" "Unmounting failed."
	fi
fi
}

usbdrives=$(lsblk -nrpo "name,type,size,mountpoint" | awk '$2=="part"&&$4!~/\/boot|\/home$|SWAP/&&length($4)>1{printf "%s (%s)\n",$4,$3}')

if [ -z "$usbdrives" ]; then
	notify-send -u "critical" -t 3000 "禍 USB ERROR" "No drive to unmount."
else
	unmountusb
fi