#!/bin/sh

pgrep -x dmenu && exit
COLS="name,type,size,mountpoint"

drives="$(lsblk -rpo "$COLS" | awk '$2=="part"&&$4==""{printf "%s (%s)\n",$1,$3}')"

[ -z "$drives" ] && exit 1
chosen="$(echo "$drives" | dmenu -i -p "What to mount?" | awk '{print $1}')"
[ -z "$chosen" ] && exit 1
sudo -A mount "$chosen" && notify-send "$chosen mounted on $(mount | grep $chosen | awk '{print $3}')" && exit 0

mp="$(find /mnt /media /home -type d -maxdepth 5 2>/dev/null | dmenu -i -p "Type in mount point")"
[ "$mp" = "" ] && exit 1

if [ ! -d "$mp" ]; then
    mkdiryn=$(printf "No\\nYes" | dmenu -i -p "$mp does not exist. Create it?")
    [ "$mkdiryn" = "Yes" ] && sudo -A mkdir -p "$mp"
fi

sudo -A mount "$chosen" "$mp" && notify-send "$chosen mounted to $mp"