#!/bin/sh

MNTPOINTS="/media/usb1\n/media/usb2\n/mnt"

MOUNTABLE=$(lsblk -lp | grep "^/dev/sd[b-z][1-9]" | awk '{print $1, "(" $4 ")"}')

#check if MOUTABLE is empty
if [ -z "$MOUNTABLE" ]; then
  echo "nothing to (un)mount" | dmenu
  exit 1
fi

MOU=$(printf "m\nu" | dmenu -p "mount or unmount?")

case $MOU in
  m)
  TOMOUNT=$(echo "$MOUNTABLE" | dmenu -i -p "which one?" | awk '{print $1}')
  if [ ! -z "$TOMOUNT" ]; then
    MNTPNT=$(printf "$MNTPOINTS" | dmenu -p "where?")
    echo "$TOMOUNT"
    echo "$MNTPNT"
    sudo mount $TOMOUNT $MNTPNT
    exit 0
  fi
  ;;
  u)
  MOUNTED=$(lsblk -lp | grep " /media/*" | awk '{print $1}')
  if [ -z "$MOUNTED" ]; then
    FAIL=$(echo "ok then..." | dmenu -p "nothing to unmount...")
    exit 0
  else
    TOUNMOUNT=$(echo "$MOUNTED" | dmenu -p "which one?")
    sudo umount $TOUNMOUNT
  fi
  ;;
esac