From 06c2883bb157443568ca09e1a8feef9de55fe454 Mon Sep 17 00:00:00 2001
From: qoijjj <129108030+qoijjj@users.noreply.github.com>
Date: Mon, 19 Aug 2024 18:21:50 -0700
Subject: [PATCH] fix: improve usbguard just command

---
 .../usr/share/ublue-os/just/60-custom.just      | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/files/system/usr/share/ublue-os/just/60-custom.just b/files/system/usr/share/ublue-os/just/60-custom.just
index ffb24fe..6a867a4 100644
--- a/files/system/usr/share/ublue-os/just/60-custom.just
+++ b/files/system/usr/share/ublue-os/just/60-custom.just
@@ -161,14 +161,17 @@ override-reset-module mod_name:
 
 # Setup USBGuard
 setup-usbguard:
-    #!/usr/bin/pkexec /usr/bin/bash
+    #!/usr/bin/bash
     echo "Notice: This will generate a policy based on your existing connected USB devices."
-    sudo mkdir -p /var/log/usbguard
-    sudo mkdir -p /etc/usbguard
-    sudo chmod 755 /etc/usbguard
-    sudo sh -c 'usbguard generate-policy > /etc/usbguard/rules.conf'
-    sudo systemctl enable --now usbguard.service
-    sudo usbguard add-user $(whoami)
+    ACTIVE_USERNAME=$(whoami)
+    pkexec sh -c '
+        mkdir -p /var/log/usbguard
+        mkdir -p /etc/usbguard
+        chmod 755 /etc/usbguard
+        usbguard generate-policy > /etc/usbguard/rules.conf
+        systemctl enable --now usbguard.service
+        usbguard add-user $1
+    ' -- $ACTIVE_USERNAME
     systemctl enable --user --now usbguard-notifier.service
 
 # Rerun Yafti