1 files changed, 97 insertions, 0 deletions
diff --git a/qmenu/qmenu-dm b/qmenu/qmenu-dm
new file mode 100644
index 0000000..4f95247
--- /dev/null
+++ b/qmenu/qmenu-dm
@@ -0,0 +1,97 @@
+#!/bin/sh
+
+get_qube_label() {
+
+qube_label=$(grep -s "^$(qvm-prefs "$qube" label)=" "$HOME"/.config/qmenu.conf | cut -d= -f2)
+
+[ -z "$qube_label" ] && qube_label='#000000'
+}
+
+
+case $1 in
+
+	--all)
+
+		qube_list=$(qvm-ls --no-spinner --running -O name,label | grep -v '^dom0 \|^sys-usb' | sed '1d; s/ red/ 1\@ /g; s/ orange/ 2\@ /g; s/ yellow/ 3\@ /g; s/ green/ 4\@ /g; s/ gray/ 5\@ /g; s/ blue/ 6\@ /g; s/ purple/ 7\@ /g; s/ black/ 8\@ /g' | sort -k2,2 | sed 's/ 1\@ / red/g; s/ 2\@ / orange/g; s/ 3\@ / yellow/g; s/ 4\@ / green/g; s/ 5\@ / gray/g; s/ 6\@ / blue/g; s/ 7\@ / purple/g; s/ 8\@ / black/g');;
+
+	--focused)
+
+		qube=$(xprop -id "$(xdotool getwindowfocus)" _QUBES_VMNAME | cut -f2 -d\")
+
+		[ "$qube" = '_QUBES_VMNAME:  not found.' ] && exit 2;;
+
+	*)
+
+		printf "Usage: $0 [OPTION]\nList and manage your connected devices via dmenu.\n\n --all\n --focused\n\n"
+
+		[ "$1" = --help ] && exit 0; exit 2
+esac
+
+while :; do
+
+	device=$(echo "$(qvm-block ls | sed -e 's/^/|block| /' & qvm-device mic ls | sed -e 's/^/|mic| /' & qvm-usb ls | sed -e 's/^/|usb| /')" | sort -r -k1,2 | dmenu-unlinked -l 32) || exit 1
+
+	device_type=$(echo "$device" | cut -d\| -f2)
+
+	if [ -n "$device" ] && [ "$(echo "$device" | wc -l)" -eq 1 ]; then
+
+		device_id=$(echo "$device" | cut -f2 -d\ )
+
+		if [ "$device_type" = block ]; then
+
+			device_name=$(echo "$device" | awk '{print $3,$4}')
+			holding_qube=$(echo "$device" | awk '{print $5}')
+		else
+			device_name=$(echo "$device" | awk '{print $3}')
+			holding_qube=$(echo "$device" | awk '{print $4}')
+		fi
+
+		if [ -z "$holding_qube" ]; then
+
+			if [ "$1" = --all ]; then
+
+				qube=$(echo "$qube_list" | dmenu-unlinked -p "attach to:" -l 32 | cut -f1 -d\ )
+			fi
+
+			if [ -n "$qube" ]; then
+
+				get_qube_label
+
+				if [ "$(printf 'No\nYes' | dmenu-unlinked -sb "$qube_label" -i -p "Attach '$device_name' to $qube?")" = Yes ]; then
+
+					qvm-device "$device_type" attach -q "$qube" "$device_id" && exit 0
+
+					echo Quit... | dmenu-unlinked -p "Error: Failed to attach device!" > /dev/null 2>&1
+
+					exit 2
+				fi
+			fi
+
+		elif [ "$1" = --all ] || [ "$holding_qube" = "$qube" ]; then
+
+			get_qube_label
+
+			if [ "$(printf 'No\nYes' | dmenu-unlinked -sb "$qube_label" -i -p "Detach '$device_name' from $holding_qube?")" = Yes ]; then
+
+				qvm-device "$device_type" detach -q "$holding_qube" "$device_id" && exit 0
+
+				echo Quit... | dmenu-unlinked -p "Error: Failed to detach device!" > /dev/null 2>&1
+
+				exit 2
+			fi
+		else
+			get_qube_label
+
+			if [ "$(printf 'No\nYes' | dmenu-unlinked -sb "$qube_label" -i -p "Detach '$device_name' from $holding_qube and attach it to $qube?")" = Yes ]; then
+
+				qvm-device "$device_type" detach -q "$holding_qube" "$device_id" &&
+
+					qvm-device "$device_type" attach -q "$qube" "$device_id" && exit 0
+
+				echo Quit... | dmenu-unlinked -p "Error: Failed to handle device!" > /dev/null 2>1&
+
+				exit 2
+			fi
+		fi
+	fi
+done