poweroff external storage devices with bemenu

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