From 9926cf2202ae39dd2cc3e8d3347e2b3b00e86cbb Mon Sep 17 00:00:00 2001
From: Oliver Davies <oliver@oliverdavies.dev>
Date: Thu, 13 Mar 2025 19:59:41 +0000
Subject: [PATCH] Add new scripts

---
 bin/displayselect | 85 +++++++++++++++++++++++++++++++++++++++++++++++
 bin/mounter       | 36 ++++++++++++++++++++
 bin/unmounter     | 26 +++++++++++++++
 3 files changed, 147 insertions(+)
 create mode 100755 bin/displayselect
 create mode 100755 bin/mounter
 create mode 100755 bin/unmounter

diff --git a/bin/displayselect b/bin/displayselect
new file mode 100755
index 00000000..a4ebb202
--- /dev/null
+++ b/bin/displayselect
@@ -0,0 +1,85 @@
+#!/bin/sh
+
+# A UI for detecting and selecting all displays. Probes xrandr for connected
+# displays and lets user select one to use. User may also select "manual
+# selection" which opens arandr.
+
+twoscreen() { # If multi-monitor is selected and there are two screens.
+  mirror=$(printf "no\\nyes" | dmenu -i -p "Mirror displays?")
+
+  # Mirror displays using native resolution of external display and a scaled
+  # version for the internal display
+  if [ "$mirror" = "yes" ]; then
+    external=$(echo "$screens" | dmenu -i -p "Optimize resolution for:")
+    internal=$(echo "$screens" | grep -v "$external")
+
+    res_external=$(xrandr --query | sed -n "/^$external/,/\+/p" | \
+      tail -n 1 | awk '{print $1}')
+
+    res_internal=$(xrandr --query | sed -n "/^$internal/,/\+/p" | \
+      tail -n 1 | awk '{print $1}')
+
+    res_ext_x=$(echo "$res_external" | sed 's/x.*//')
+    res_ext_y=$(echo "$res_external" | sed 's/.*x//')
+    res_int_x=$(echo "$res_internal" | sed 's/x.*//')
+    res_int_y=$(echo "$res_internal" | sed 's/.*x//')
+
+    scale_x=$(echo "$res_ext_x / $res_int_x" | bc -l)
+    scale_y=$(echo "$res_ext_y / $res_int_y" | bc -l)
+
+    xrandr --output "$external" --auto --scale 1.0x1.0 \
+      --output "$internal" --auto --same-as "$external" \
+      --scale "$scale_x"x"$scale_y"
+  else
+
+    primary=$(echo "$screens" | dmenu -i -p "Select primary display:")
+    secondary=$(echo "$screens" | grep -v ^"$primary"$)
+    direction=$(printf "left\\nright" | dmenu -i -p "What side of $primary should $secondary be on?")
+
+    xrandr --output "$primary" --auto --scale 1.0x1.0 --output "$secondary" --"$direction"-of "$primary" --auto --scale 1.0x1.0
+  fi
+}
+
+morescreen() { # If multi-monitor is selected and there are more than two screens.
+  primary=$(echo "$screens" | dmenu -i -p "Select primary display:")
+  secondary=$(echo "$screens" | grep -v ^"$primary"$ | dmenu -i -p "Select secondary display:")
+  direction=$(printf "left\\nright" | dmenu -i -p "What side of $primary should $secondary be on?")
+  tertiary=$(echo "$screens" | grep -v ^"$primary"$ | grep -v ^"$secondary"$ | dmenu -i -p "Select third display:")
+
+  xrandr --output "$primary" --auto --output "$secondary" --"$direction"-of "$primary" --auto --output "$tertiary" --"$(printf "left\\nright" | grep -v "$direction")"-of "$primary" --auto
+}
+
+multimon() { # Multi-monitor handler.
+  case "$(echo "$screens" | wc -l)" in
+    2) twoscreen ;;
+    *) morescreen ;;
+  esac ;}
+
+onescreen() { # If only one output available or chosen.
+  xrandr --output "$1" --auto --scale 1.0x1.0 $(echo "$allposs" | grep -v "\b$1" | awk '{print "--output", $1, "--off"}' | paste -sd ' ' -)
+}
+
+postrun() { # Stuff to run to clean up.
+  { killall dunst ; setsid -f dunst ;} >/dev/null 2>&1 # Restart dunst to ensure proper location on screen
+}
+
+# Get all possible displays
+allposs=$(xrandr -q | grep "connected")
+
+# Get all connected screens.
+screens=$(echo "$allposs" | awk '/ connected/ {print $1}')
+
+# If there's only one screen
+[ "$(echo "$screens" | wc -l)" -lt 2 ] &&
+  { onescreen "$screens"; postrun; notify-send "Only one screen detected." "Using it in its optimal settings...";  exit ;}
+
+# Get user choice including multi-monitor and manual selection:
+chosen=$(printf "%s\\nmulti-monitor\\nmanual selection" "$screens" | dmenu -i -p "Select display arangement:") &&
+
+case "$chosen" in
+  "manual selection") arandr ; exit ;;
+  "multi-monitor") multimon ;;
+  *) onescreen "$chosen" ;;
+esac
+
+postrun
diff --git a/bin/mounter b/bin/mounter
new file mode 100755
index 00000000..60b828eb
--- /dev/null
+++ b/bin/mounter
@@ -0,0 +1,36 @@
+#!/usr/bin/env bash
+
+set -o errexit
+set -o nounset
+set -o pipefail
+
+lsblk_output="$(lsblk -rpo "uuid,name,type,size,label,mountpoint,mountpoints,fstype" | grep "part" | grep -Ev "/(boot|nix/store)")"
+partition_names="$(echo "$lsblk_output" | awk '{ printf "%s (%s)\n", $2, $4 }' )";
+
+selected=$(echo "$partition_names" | dmenu -p "Select drive:" | awk '{ print $1 }')
+test -n "$selected"
+mount_device="$selected"
+
+if sudo cryptsetup isLuks "$selected"; then
+  luks_name="usb"
+
+  ${TERMINAL:-st} -n floatterm -g 60x1 -e sudo cryptsetup open "$selected" "$luks_name"
+
+  mount_device="/dev/mapper/$luks_name"
+
+  test -b "/dev/mapper/$luks_name"
+fi
+
+mount_point=$(echo -e "/mnt\n/media\n/run/media/$USER" | dmenu -p "Select mount point:")
+test -n "$mount_point"
+
+sudo mkdir -p "$mount_point"
+
+if sudo mount "$mount_device" "$mount_point"; then
+  notify-send "Device mounted" "Mounted $selected at $mount_point."
+else
+  notify-send "Mount failed" "Failed to mount $selected."
+
+  exit 1
+fi
+
diff --git a/bin/unmounter b/bin/unmounter
new file mode 100755
index 00000000..cc3d5d8b
--- /dev/null
+++ b/bin/unmounter
@@ -0,0 +1,26 @@
+#!/usr/bin/env bash
+
+set -o errexit
+set -o nounset
+set -o pipefail
+
+lsblk_output="$(lsblk -nrpo "name,type,size,mountpoint")"
+mounted_drives="$(echo "$lsblk_output" | awk '($2=="part"||$2="crypt")&&$4!~/\/boot|\/home$|SWAP/&&length($4)>1{printf "%s (%s)\n",$4,$3}')"
+
+all_unmountable="$(echo "$mounted_drives" | sed "/^$/d;s/ *$//")"
+test -n "$all_unmountable"
+
+selected="$(echo "$all_unmountable" | dmenu -i -p "Unmount which drive?")"
+selected="${selected%% *}"
+test -n "$selected"
+
+sudo -A umount -l "/${selected#*/}"
+notify-send "Device unmounted" "$selected has been unmounted."
+
+# Close the selected drive if decrypted.
+cryptid="$(echo "$lsblk_output" | grep "/${selected#*/}$")"
+cryptid="${cryptid%% *}"
+test -b /dev/mapper/"${cryptid##*/}"
+sudo -A cryptsetup close "$cryptid"
+
+notify-send "Device dencryption closed" "Drive is now securely locked again."