diff options
| -rwxr-xr-x | dotfiles/.local/bin/menublk | 176 |
1 files changed, 101 insertions, 75 deletions
diff --git a/dotfiles/.local/bin/menublk b/dotfiles/.local/bin/menublk index 3020422..3c507ac 100755 --- a/dotfiles/.local/bin/menublk +++ b/dotfiles/.local/bin/menublk @@ -3,6 +3,8 @@ # Script for mounting/unmounting drives (including encrypted ones) # For encrypted drives, it works only if there's just one partition inside the # crypt_LUKS volume +# It now also allows to power off devices so that you can safely disconnect +# them. It lists only USB devices. # Yaroslav de la Peña Smirnov @@ -11,96 +13,120 @@ readarray -t blkdevs <<< "$(lsblk -r -o NAME,TYPE,FSTYPE,MOUNTPOINT | tail -n +2)" get_mountable() { - for blkdev in "${blkdevs[@]}"; do - name="$(cut -d " " -f1 <<< "${blkdev}")" - btype="$(cut -d " " -f2 <<< "${blkdev}")" - fstype="$(cut -d " " -f3 <<< "${blkdev}")" - mountpoint="$(cut -d " " -f4 <<< "${blkdev}")" - if [ "$btype" = "part" ]; then - if [ -z "$mountpoint" ]; then - if [ "$fstype" = "crypto_LUKS" ]; then - if [ "$(lsblk -r -o FSTYPE "/dev/$name" | tail -n 1)" = "crypto_LUKS" ]; then - echo "$name" - fi - else - echo "$name" - fi - fi + for blkdev in "${blkdevs[@]}"; do + name="$(cut -d " " -f1 <<< "${blkdev}")" + btype="$(cut -d " " -f2 <<< "${blkdev}")" + fstype="$(cut -d " " -f3 <<< "${blkdev}")" + mountpoint="$(cut -d " " -f4 <<< "${blkdev}")" + if [ "$btype" = "part" ]; then + if [ -z "$mountpoint" ]; then + if [ "$fstype" = "crypto_LUKS" ]; then + if [ "$(lsblk -r -o FSTYPE "/dev/$name" | tail -n 1)" = "crypto_LUKS" ]; then + echo "$name" + fi + else + echo "$name" fi - done + fi + fi + done } get_unmountable() { - for blkdev in "${blkdevs[@]}"; do - name="$(cut -d " " -f1 <<< "${blkdev}")" - btype="$(cut -d " " -f2 <<< "${blkdev}")" - fstype="$(cut -d " " -f3 <<< "${blkdev}")" - mountpoint="$(cut -d " " -f4 <<< "${blkdev}")" - if [ "$btype" = "part" ]; then - if [ -z "$mountpoint" ] && [ "$fstype" = "crypto_LUKS" ]; then - mountpoint="$(lsblk -r -o MOUNTPOINT "/dev/$name" | tail -n 1)" - fi - if [ -n "$mountpoint" ] && [ "$mountpoint" != "/" ] && [ "$mountpoint" != "/boot" ]; then - echo "$name" - fi - fi - done + for blkdev in "${blkdevs[@]}"; do + name="$(cut -d " " -f1 <<< "${blkdev}")" + btype="$(cut -d " " -f2 <<< "${blkdev}")" + fstype="$(cut -d " " -f3 <<< "${blkdev}")" + mountpoint="$(cut -d " " -f4 <<< "${blkdev}")" + if [ "$btype" = "part" ]; then + if [ -z "$mountpoint" ] && [ "$fstype" = "crypto_LUKS" ]; then + mountpoint="$(lsblk -r -o MOUNTPOINT "/dev/$name" | tail -n 1)" + fi + if [ -n "$mountpoint" ] && [ "$mountpoint" != "/" ] && [ "$mountpoint" != "/boot" ] + then + echo "$name" + fi + fi + done } mount_dev() { - fstype="$(lsblk -r -o FSTYPE "/dev/$1" | tail -n 1)" - part="/dev/$1" - if [ "$fstype" = "crypto_LUKS" ]; then - password="$(echo "" | bemenu -p "Enter passphrase for encrypted drive:" --tf="$BEMENU_TF" --tb="$BEMENU_NB" --ff="$BEMENU_NB" --fb="$BEMENU_NB" --fn="$BEMENU_FN" --nb="$BEMENU_NB" --nf="$BEMENU_NF" --hf="$BEMENU_HF" --hb="$BEMENU_HB")" - if [ -z "$password" ]; then - exit 0 - fi - printf "%s" "$password" > "$HOME/.cache/diskey" - part="$(udisksctl unlock --key-file="$HOME/.cache/diskey" -b "/dev/$1" | cut -d " " -f4 )" - part="${part%?}" - rm -f "$HOME/.cache/diskey" - fi - message="$(udisksctl mount -b "$part")" - if [ $? -eq 0 ]; then - notify-send "💽Drive mounting" "$message" - else - notify-send "💽Drive mounting" "An error occurred, could not mount device $1" + fstype="$(lsblk -r -o FSTYPE "/dev/$1" | tail -n 1)" + part="/dev/$1" + if [ "$fstype" = "crypto_LUKS" ]; then + password="$(echo "" | bemenu -p "Enter passphrase for encrypted drive:" --tf="$BEMENU_TF" --tb="$BEMENU_NB" --ff="$BEMENU_NB" --fb="$BEMENU_NB" --fn="$BEMENU_FN" --nb="$BEMENU_NB" --nf="$BEMENU_NF" --hf="$BEMENU_HF" --hb="$BEMENU_HB")" + if [ -z "$password" ]; then + exit 0 fi + printf "%s" "$password" > "$HOME/.cache/diskey" + part="$(udisksctl unlock --key-file="$HOME/.cache/diskey" -b "/dev/$1" | cut -d " " -f4 )" + part="${part%?}" + rm -f "$HOME/.cache/diskey" + fi + message="$(udisksctl mount -b "$part")" + if [ $? -eq 0 ]; then + notify-send "💽Drive mounting" "$message" + else + notify-send "💽Drive mounting" "An error occurred, could not mount device $1" + fi } unmount_dev() { - fstype="$(lsblk -r -o FSTYPE "/dev/$1" | sed "2q;d")" - if [ "$fstype" = "crypto_LUKS" ]; then - volume="/dev/mapper/$(lsblk -r -o NAME "/dev/$1" | tail -n 1)" - if udisksctl unmount -b "$volume" && udisksctl lock -b "/dev/$1"; then - notify-send "💽Drive manager" "Device $1 has been unmounted and locked" - else - notify-send "💽Drive manager" "An error occurred, could not unmount device $1" - fi + fstype="$(lsblk -r -o FSTYPE "/dev/$1" | sed "2q;d")" + if [ "$fstype" = "crypto_LUKS" ]; then + volume="/dev/mapper/$(lsblk -r -o NAME "/dev/$1" | tail -n 1)" + if udisksctl unmount -b "$volume" && udisksctl lock -b "/dev/$1"; then + notify-send "💽Drive manager" "Device $1 has been unmounted and locked" else - if udisksctl unmount -b "/dev/$1"; then - notify-send "💽Drive manager" "Device $1 has been unmounted" - else - notify-send "💽Drive manager" "An error occurred, could not unmount device $1" - fi + notify-send "💽Drive manager" "An error occurred, could not unmount device $1" fi + else + if udisksctl unmount -b "/dev/$1"; then + notify-send "💽Drive manager" "Device $1 has been unmounted" + else + notify-send "💽Drive manager" "An error occurred, could not unmount device $1" + fi + fi +} + +get_usb_storage() { + for dev in /sys/block/*; do + if udevadm info --query=property --path=$dev | grep -q ^ID_BUS=usb + then + echo "$(cut -d "/" -f4 <<< ${dev})" + fi + done +} + +poweroff_dev() { + if udisksctl power-off -b "/dev/$1"; then + notify-send "💽Drive manager" "Device $1 has been powered off" + else + notify-send "💽Drive manager" "An error occurred, could not power off device $1. Make sure that it's not in use." + fi } -action="$(printf "mount\nunmount\n" | bemenu -p "What to do?" --tf="$BEMENU_TF" --tb="$BEMENU_NB" --fb="$BEMENU_NB" --fn="$BEMENU_FN" --nb="$BEMENU_NB" --nf="$BEMENU_NF" --hf="$BEMENU_HF" --hb="$BEMENU_HB")" +action="$(printf "mount\nunmount\npoweroff\n" | bemenu -p "What to do?" --tf="$BEMENU_TF" --tb="$BEMENU_NB" --fb="$BEMENU_NB" --fn="$BEMENU_FN" --nb="$BEMENU_NB" --nf="$BEMENU_NF" --hf="$BEMENU_HF" --hb="$BEMENU_HB")" case "$action" in - mount) - blkdev="$(get_mountable | bemenu -p "Choose device to mount:" --tf="$BEMENU_TF" --tb="$BEMENU_NB" --fb="$BEMENU_NB" --fn="$BEMENU_FN" --nb="$BEMENU_NB" --nf="$BEMENU_NF" --hf="$BEMENU_HF" --hb="$BEMENU_HB")" - if [ -z "$blkdev" ]; then - exit 0 - fi - mount_dev "$blkdev";; - unmount) - blkdev="$(get_unmountable | bemenu -p "Choose device to umount:" --tf="$BEMENU_TF" --tb="$BEMENU_NB" --fb="$BEMENU_NB" --fn="$BEMENU_FN" --nb="$BEMENU_NB" --nf="$BEMENU_NF" --hf="$BEMENU_HF" --hb="$BEMENU_HB")" - if [ -z "$blkdev" ]; then - exit 0 - fi - unmount_dev "$blkdev";; - *) - notify-send "💽Drive manager" "That's not an option";; + mount) + blkdev="$(get_mountable | bemenu -p "Choose device to mount:" --tf="$BEMENU_TF" --tb="$BEMENU_NB" --fb="$BEMENU_NB" --fn="$BEMENU_FN" --nb="$BEMENU_NB" --nf="$BEMENU_NF" --hf="$BEMENU_HF" --hb="$BEMENU_HB")" + if [ -z "$blkdev" ]; then + exit 0 + fi + mount_dev "$blkdev";; + unmount) + blkdev="$(get_unmountable | bemenu -p "Choose device to umount:" --tf="$BEMENU_TF" --tb="$BEMENU_NB" --fb="$BEMENU_NB" --fn="$BEMENU_FN" --nb="$BEMENU_NB" --nf="$BEMENU_NF" --hf="$BEMENU_HF" --hb="$BEMENU_HB")" + if [ -z "$blkdev" ]; then + exit 0 + fi + unmount_dev "$blkdev";; + poweroff) + blkdev="$(get_usb_storage | bemenu -p "Choose device to power off:" --tf="$BEMENU_TF" --tb="$BEMENU_NB" --fb="$BEMENU_NB" --fn="$BEMENU_FN" --nb="$BEMENU_NB" --nf="$BEMENU_NF" --hf="$BEMENU_HF" --hb="$BEMENU_HB")" + if [ -z "$blkdev" ]; then + exit 0 + fi + poweroff_dev "$blkdev";; + *) + exit 0;; esac |