#!/bin/sh

######################################################################
# @author       : swytch (adapted from Luke Smith - lukesmith.xyz)
# @file         : dmenumount
# @license      : MIT
# @created      : Wednesday May 20, 2020 18:05:31 CEST
#
# @description  : dmenu prompt to mount unmounted drives
######################################################################


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() { \
	chosen="$(echo "$usbdrives" | dmenu -i -p "Mount which drive?" | awk '{print $1}')"
	alreadymounted=$(lsblk -nrpo "name,type,mountpoint" \
		| awk '$2=="part"&&$3!~/\/efi|\/home$|SWAP/&&length($3)>1\{printf "-not \\( -path *%s -prune \\) \\ \n",$3}')
	getmount "/mnt /media /mount /home -maxdepth 2 -type d $alreadymounted"
	partitiontype="$(lsblk -no "fstype" "$chosen")"
	case "$partitiontype" in
		"vfat") sudo -A mount -t vfat "$chosen" "$mp" -o rw,umask=0000;;
		*) sudo -A mount "$chosen" "$mp"; user="$(whoami)";
		sudo -A chown "$user":"$user" "$mp";;
	esac
	notify-send " USB mounting" "$chosen mounted to $mp."
}

usbdrives="$(lsblk -rpo "name,type,size,mountpoint" | awk '$2=="part"&&$4==""{printf "%s (%s)\n",$1,$3}')"

if [ -z "$usbdrives" ]; then
	notify-send -u "critical" -t 3000 " USB ERROR" "No drive to mount."
else
	notify-send "USB drive(s) detected."
	mountusb
fi