#!/bin/sh
# Gives a dmenu prompt to mount unmounted drives.
# If they're in /etc/fstab, they'll be mounted automatically.
# Otherwise, you'll be prompted to give a mountpoint from already existsing directories.
# If you input a novel directory, it will prompt you to create that directory.
pgrep -x dmenu && exit

getmount() { \
	[ -z "$chosen" ] && exit 1
	mp="$(find $1 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" ] && (mkdir -p "$mp" || sudo -A mkdir -p "$mp")
	fi
	}

mountusb() { \
	drives="$(lsblk -rpo "name,type,size,mountpoint" | awk '$2=="part"&&$4==""{printf "%s (%s)\n",$1,$3}')"
	sudo -A mount "$chosen" && exit 0
	[ -z "$drives" ] && exit 1
	chosen="$(echo "$drives" | dmenu -i -p "Mount which drive?" | awk '{print $1}')"
	getmount "/mnt /media /mount /home -type d -maxdepth 5"
	sudo -A mount "$chosen" "$mp" && pgrep -x dunst && notify-send "$chosen mounted to $mp."
	}

mountandroid() { \
	drives=$(simple-mtpfs -l)
	chosen=$(echo "$drives" | dmenu -i -p "Which Android device?" | cut -d : -f 1)
	getmount "$HOME -type d -maxdepth 3"
	simple-mtpfs --device "$chosen" "$mp"
	echo simple-mtpfs --device "$chosen" "$mp"
	}

asktype() { \
	case $(printf "USB\\nAndroid" | dmenu -i -p "Mount a USB drive or Android device?") in
		USB) mountusb ;;
		Android) mountandroid ;;
	esac
	}

if simple-mtpfs -l; then
	asktype
else
	mountusb
fi