From 4663df657638e4f3a4cd9618df857bdb2348ea38 Mon Sep 17 00:00:00 2001
From: Luke Smith <luke@lukesmith.xyz>
Date: Tue, 28 Aug 2018 23:39:56 -0400
Subject: [PATCH] android mtp compatibility added to (u)mount scripts

---
 .scripts/dmenumount  | 53 +++++++++++++++++++++++++++++++-------------
 .scripts/dmenuumount | 31 +++++++++++++++++++++-----
 2 files changed, 64 insertions(+), 20 deletions(-)

diff --git a/.scripts/dmenumount b/.scripts/dmenumount
index 91683e6..33060c2 100755
--- a/.scripts/dmenumount
+++ b/.scripts/dmenumount
@@ -4,20 +4,43 @@
 # 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
-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 "Mount which drive?" | awk '{print $1}')"
-[ -z "$chosen" ] && exit 1
-sudo -A mount "$chosen" && exit 0
-# You may want to change the line below for more suggestions for mounting.
-# I.e. you can increase the depth of the search, or add directories.
-# This will increase the load time briefly though.
-mp="$(find /mnt /media /mount /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"
+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
-sudo -A mount "$chosen" "$mp" && pgrep -x dunst && notify-send "$chosen mounted to $mp."
diff --git a/.scripts/dmenuumount b/.scripts/dmenuumount
index 0114fc4..f70217f 100755
--- a/.scripts/dmenuumount
+++ b/.scripts/dmenuumount
@@ -3,8 +3,29 @@
 # Provides you with mounted partitions, select one to unmount.
 # Drives mounted at /, /boot and /home will not be options to unmount.
 
-drives=$(lsblk -nrpo "name,type,size,mountpoint" | awk '$2=="part"&&$4!~/\/boot|\/home$|SWAP/&&length($4)>1{printf "%s (%s) on %s\n",$1,$3,$4}')
-[ -z "$drives" ] && exit
-chosen=$(echo "$drives" | dmenu -i -p "Unmount which drive?" | awk '{print $1}')
-[ -z "$chosen" ] && exit
-sudo -A umount "$chosen" && pgrep -x dunst && notify-send "$chosen unmounted."
+unmountusb() {
+	drives=$(lsblk -nrpo "name,type,size,mountpoint" | awk '$2=="part"&&$4!~/\/boot|\/home$|SWAP/&&length($4)>1{printf "%s (%s)\n",$4,$3}')
+	[ -z "$drives" ] && exit
+	chosen=$(echo "$drives" | dmenu -i -p "Unmount which drive?" | awk '{print $1}')
+	[ -z "$chosen" ] && exit
+	sudo -A umount "$chosen" && pgrep -x dunst && notify-send "$chosen unmounted."
+	}
+
+unmountandroid() { \
+	chosen=$(awk '/simple-mtpfs/ {print $2}' /etc/mtab | dmenu -i -p "Unmount which device?")
+	[ -z "$chosen" ] && exit
+	fusermount -u "$chosen" && pgrep -x dunst && notify-send "$chosen unmounted."
+	}
+
+asktype() { \
+	case $(printf "USB\\nAndroid" | dmenu -i -p "Unmount a USB drive or Android device?") in
+		USB) unmountusb ;;
+		Android) unmountandroid ;;
+	esac
+	}
+
+if grep simple-mtpfs /etc/mtab; then
+	asktype
+else
+	unmountusb && rmdiryn=$(printf "No\\nYes" | dmenu -i -p "Remove directory \`$chosen\`?") && [ "$rmdiryn" = "Yes" ] && (rmdir "$chosen" || sudo -A rmdir "$chosen")
+fi