#!/bin/sh
sudo="sudo"
which doas && sudo="doas"

umo=0

while getopts "u" opt; do
    case "$opt" in
        u) umo=1
            ;;
        *) echo invalid arg
            exit
            ;;
    esac
done

shift $((OPTIND-1))
[ "${1:-}" = "--" ] && shift

if [ $umo = 1 ]; then
    nf=4
else
    nf=3
fi


parts="$(lsblk -rno path,size,type,mountpoint | grep -e part -e lvm | awk "NF==$nf {print \$1, \$2, \$3, \$4}")"

# choice="$(printf "$parts" | rofi -dmenu "beans" | sed 's/ .*//')"
choice="$(printf -- "%s" "$parts" | fzf | sed 's/ .*//')"

[ "$choice" = "" ] && exit 0

if [ $umo = 1 ]; then
    $sudo umount "$choice"
else
    $sudo mount "$choice" /mnts/flash
fi