diff --git a/system_files/desktop/shared/usr/share/ublue-os/just/80-bazzite.just b/system_files/desktop/shared/usr/share/ublue-os/just/80-bazzite.just
index 975de115..4caf345d 100644
--- a/system_files/desktop/shared/usr/share/ublue-os/just/80-bazzite.just
+++ b/system_files/desktop/shared/usr/share/ublue-os/just/80-bazzite.just
@@ -48,26 +48,49 @@ configure-waydroid:
deactivate
sudo rm -rf /tmp/waydroid_script
-# Install Sunshine
-install-sunshine:
- #!/usr/bin/bash
- systemctl enable sunshine-workaround.service
- ublue-update --wait
- rpm-ostree install -y https://github.com/LizardByte/Sunshine/releases/download/nightly-dev/sunshine-fedora-$(rpm -E %fedora)-amd64.rpm
- echo "Sunshine is installed!"
-
-# Remove Sunshine
-remove-sunshine:
- #!/usr/bin/bash
- systemctl disable sunshine-workaround.service
- ublue-update --wait
- rpm-ostree remove -y sunshine
- echo "Sunshine has been uninstalled."
-
-# Autostart Sunshine
-autostart-sunshine:
- #!/usr/bin/bash
- systemctl enable --user --now sunshine.service
+# Setup and configure Sunshine Game Streaming host
+setup-sunshine ACTION="":
+ #!/bin/bash
+ source /usr/lib/ujust/ujust.sh
+ SUNSHINE_STATE="$(rpm -qa sunshine)"
+ SERVICE_STATE="$(systemctl is-enabled --user sunshine.service)"
+ OPTION={{ ACTION }}
+ if [ "$SUNSHINE_STATE" == "" ]; then
+ SUNSHINE_STATE="${red}${b}Not Installed${n}"
+ else
+ SUNSHINE_STATE="${green}${b}Installed${n}"
+ fi
+ if [ "$SERVICE_STATE" == "enabled" ]; then
+ SERVICE_STATE="${green}${b}Enabled${n}"
+ else
+ SERVICE_STATE="${red}${b}Disabled${n}"
+ fi
+ if [ "$OPTION" == "help" ]; then
+ echo "Usage: ujust setup-sunshine