mildly working usb mount script

usm

@@ -1,10 +1,27 @@
 #!/bin/sh
 # pick out the drive but not sda because thats the main drive
 choice=`lsblk -lp | \
-	grep 'disk $' | awk '{print $1, $4}' | grep -v 'sda' | \
-	dmenu -i -p 'Drive to mount'`
+	grep 'disk ' | \
+	awk '{print $1, $4}' | \
+	dmenu -i -p 'Drive to (un)mount'`
+if [ -z "$choice" ]; then exit; fi
+
+# Check if it's already mounted / something was even picked
 dev=`echo $choice | awk '{print $1}'`
-if [ -z "$dev" ];then exit 0;fi
-# Finally we can mount the thing
-mkdir -p "/mnt/`basename $dev`"
-mount "$dev" "$name"
+name="`basename $dev`"
+if [ -z "$dev" ]; then exit; fi
+if grep -qs "$dev" /proc/mounts;
+then
+	# unmount the drive 
+	umount "~/.mounts/$name"
+	rm -d "~/.mounts/$name"
+	# check if we still need the .mounts directory or nah
+	if [ "`ls ~/.mounts/`" ]
+	then
+		rm -d ~/.mounts/
+	fi
+else
+	# Finally we can mount the thing
+	mkdir -p "~/.mounts/$name"
+	mount "$dev" "~/.mounts/$name"
+fi