From 36388bc8ada8ae6b7b9f0c71f2388d4f2bf44a2e Mon Sep 17 00:00:00 2001
From: HikariKnight <2557889+HikariKnight@users.noreply.github.com>
Date: Sun, 10 Mar 2024 06:17:44 +0100
Subject: [PATCH] feat(just): add option to add user to libvirt group (#865)

---
 .../usr/share/ublue-os/just/84-bazzite-virt.just     | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/system_files/desktop/shared/usr/share/ublue-os/just/84-bazzite-virt.just b/system_files/desktop/shared/usr/share/ublue-os/just/84-bazzite-virt.just
index 4c3e817c20..54cf06e299 100644
--- a/system_files/desktop/shared/usr/share/ublue-os/just/84-bazzite-virt.just
+++ b/system_files/desktop/shared/usr/share/ublue-os/just/84-bazzite-virt.just
@@ -23,6 +23,7 @@ setup-virtualization ACTION="":
       echo "  <option>: Specify the quick option to skip the prompt"
       echo "  Use 'enable' to select Enable Virtualization"
       echo "  Use 'disable' to select Disable Virtualization"
+      echo "  Use 'group' to select Add $USER to libvirt group"
       echo "  Use 'vfio-on' to select Enable VFIO drivers"
       echo "  Use 'vfio-off' to select Disable VFIO drivers"
       echo "  Use 'shm' to select Autocreate Looking-Glass shm"
@@ -31,7 +32,14 @@ setup-virtualization ACTION="":
       echo "${bold}Virtualization Setup${normal}"
       echo "NOTE: Enabling Virtualization will layer virt-manager and qemu"
       echo "      this will slow down system updates by a lot."
-      OPTION=$(Choose "Enable Virtualization" "Disable Virtualization" "Enable VFIO drivers" "Disable VFIO drivers" "Autocreate Looking-Glass shm")
+      OPTION=$(Choose \
+        "Enable Virtualization" \
+        "Disable Virtualization" \
+        "Add $USER to libvirt group" \
+        "Enable VFIO drivers" \
+        "Disable VFIO drivers" \
+        "Autocreate Looking-Glass shm" \
+      )
     fi
     if [[ "${OPTION,,}" =~ ^enable[[:space:]]virt ]]; then
       virt_test=$(rpm-ostree status | grep -A 4 "●" | grep "virt-manager")
@@ -121,4 +129,6 @@ setup-virtualization ACTION="":
       LOOKING_GLASS_TMP"
       echo "Adding SELinux context record for /dev/shm/looking-glass"
       sudo semanage fcontext -a -t svirt_tmpfs_t /dev/shm/looking-glass
+    elif [[ "${OPTION,,}" =~ group ]]; then
+      sudo usermod -aG libvirt $USER
     fi