add dmount, an alternative to dmenumount that can unlock encrypted storage

1 files changed, 104 insertions, 0 deletions

diff --git a/dmount b/dmount
new file mode 100755
index 0000000..8b0b2b7
--- /dev/null
+++ b/dmount
@@ -0,0 +1,104 @@
+#!/bin/sh
+
+printhelp () {
+  printf "tmp\n"
+}
+
+show () {
+  printf "$@\n"
+  notify-send "dmount" "$@"
+}
+
+error_msg () {
+  show "$@"
+  exit 1
+}
+
+while getopts "xwhd:" o; do case "${o}" in
+  x) DMENU="dmenu" ;;
+  w) DMENU="bemenu" ;;
+  d) ARGS="$ARGS $OPTARG" ;;
+  h)
+    printhelp
+    exit
+    ;;
+  *)
+    printhelp
+    exit 1;;
+esac done
+
+if [ -z "$DMENU" ]; then
+  if [ -n "$WAYLAND_DISPLAY" ]; then
+    DMENU="bemenu"
+  elif [ -n "$DISPLAY" ]; then
+    DMENU="dmenu"
+  else
+    error_msg "Can't tell if Wayland or Xorg; is there a display server?"
+  fi
+fi
+
+case "$DMENU" in
+  dmenu) PASSARG="-P" ;;
+  bemenu) PASSARG="-x" ;;
+esac
+
+check_crypto () {
+  [ "$(lsblk -oFSTYPE "$1" | wc -l)" -gt 2 ] && printf "y"
+}
+
+is_crypto () {
+  lsblk -oFSTYPE "$1" | grep -q "crypto_LUKS" && printf "y"
+}
+
+unlock_crypto () {
+  NAME="$(cat /dev/null | $DMENU -p "Name" $ARGS)"
+  PASSWORD="$(cat /dev/null | $DMENU -p "Password" $PASSARG $ARGS)"
+  echo "$PASSWORD" | sudo cryptsetup open "$1" "$NAME" || FAILED="y"
+  unset PASSWORD
+  [ -n "$FAILED" ] && \
+    error_msg "Failed to decrypt partition $1!"
+  show "Decrypted $1 successfully."
+
+}
+
+get_available () { lsblk -oPATH,FSTYPE,MOUNTPOINT | tail -n+2 | while read i; do
+    [ -z "$(echo "$i" | awk '{print $2}')" ] && continue
+    [ -n "$(echo "$i" | awk '{print $3}')" ] && continue
+    #echo "${i%% *}"
+    #echo "$(lsblk -oMOUNTPOINT "${i%% *}" | tail -n+2 | grep -v -E '^\s*$')"
+    [ -n "$(echo "$i" | grep -q "crypto_LUKS" && check_crypto "${i%% *}")" ] \
+      && continue
+    echo "$i"
+  done
+}
+
+getmount() { \
+  [ -z "$chosen" ] && exit 1
+  # shellcheck disable=SC2086
+  mp="$(find $1 2>/dev/null | $DMENU -l 10 $ARGS -p "Type in mount point.")" || error_msg "Canceled; no mount point chosen."
+  test -z "$mp" && \
+    exit 1
+  if [ ! -d "$mp" ]; then
+    mkdiryn=$(printf "No\\nYes" | $DMENU -l 10 $ARGS -p "$mp does not exist. Create it?") || \
+      exit 1
+    [ "$mkdiryn" = "Yes" ] && (mkdir -p "$mp" || doas mkdir -p "$mp")
+  fi
+}
+
+mount_device () {
+  getmount "/mnt /media /mount /home -maxdepth 3 -type d -empty $alreadymounted"
+  doas mount "$1" "$mp" && show "Mounted successfully\n"
+}
+
+DEVICE="$(get_available | $DMENU -p "Devices" $ARGS -l 15)"
+[ -z "$DEVICE" ] && exit 1
+chosen="${DEVICE%% *}"
+
+if [ -n "$(is_crypto "$chosen")" ]; then
+  unlock_crypto "$chosen"
+  unset DEVICE chosen DMENU ARGS PASSARG
+  dmount
+fi
+
+mount_device "$chosen"
+unset DEVICE chosen DMENU ARGS PASSARG