From 14d7ce105a910b6e2cbbc94020ea5006e788b43b Mon Sep 17 00:00:00 2001
From: Sam Al-Sapti <sam@sapti.me>
Date: Sun, 30 Oct 2022 18:46:57 +0100
Subject: [PATCH] Add more options for USB unlocking

---
 scripts/.local/bin/usb-unlock.sh | 7 ++++++-
 waybar/.config/waybar/config     | 3 ++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/scripts/.local/bin/usb-unlock.sh b/scripts/.local/bin/usb-unlock.sh
index 256fbf0..eb392d4 100755
--- a/scripts/.local/bin/usb-unlock.sh
+++ b/scripts/.local/bin/usb-unlock.sh
@@ -1,3 +1,8 @@
 #!/usr/bin/env dash
 
-pkexec --disable-internal-agent usbctl temporary
+if usbctl check; then
+    MODE="$([ "$1" = "--permanent" ] && printf "unprotect" || printf "temporary")"
+    pkexec --disable-internal-agent usbctl "$MODE"
+else
+    pkexec --disable-internal-agent usbctl protect
+fi
diff --git a/waybar/.config/waybar/config b/waybar/.config/waybar/config
index c1cc2cb..ee44048 100644
--- a/waybar/.config/waybar/config
+++ b/waybar/.config/waybar/config
@@ -86,6 +86,7 @@
         "exec": "usb-status.sh",
         "return-type": "json",
         "interval": 1,
-        "on-click": "usb-unlock.sh"
+        "on-click": "usb-unlock.sh",
+        "on-click-right": "usb-unlock.sh --permanent"
     }
 }