#!/bin/sh
mntpnt="/mnt"

# pick out the drive but not sda because thats the main drive
choice=`lsblk -lp | \
	grep -e 'part $'  -e "part /mnt/*"| \
	awk '{print $1, $4}' | \
	dmenu -i -p 'Drive to (un)mount'`
if [ -z "$choice" ]; then exit; fi

dev=`echo $choice | awk '{print $1}'`
name="`basename $dev`"
if [ -z "$dev" ]; then exit; fi

# Check if it's already mounted / something was even picked
if grep -qs "$dev" /proc/mounts;
then
	# unmount the drive 
	umount "$HOME/.mounts/$name"
	rm -d "$HOME/.mounts/$name"
	# check if we still need the .mounts directory or nah
	if [ "`ls ~/.mounts/`" ]
	then
		rm -d ~/.mounts/
	fi
	umount $dev
	rm -d "$mntpnt/$name"
else
	mkdir -p "$mntpnt/$name"
	mount "$dev" "$mntpnt/$name"
fi