From 13998711acad7c538080bc8bb134e9ed141ace9e Mon Sep 17 00:00:00 2001
From: HikariKnight <2557889+HikariKnight@users.noreply.github.com>
Date: Fri, 12 Jan 2024 16:24:24 +0100
Subject: [PATCH] feat(just): combine several just commands into toggles or
 choices (#181)

* chore: merge install/remove-distrobox-git

* chore: merge enable/disable-updates

* chore: correct wording in description of distrobox-git

* chore(just): merge ublue/stock-gamepads and enable/disable-broadcom-wl

* chore(just): combine and merge all the nix recipes

* chore(just): remove empty lines that were used for readability while writing the just recipes

* chore(just): remove leftover spaces from before reorg

* chore(just): improve description wording for setup-distrobox-git and change brew-remove to remove-brew

* chore(just): format the message for configure-gamepads to look better

* chore(just): change gamepad description to be more clear

* chore(just): merge nvidia recipes into configure-nvidia

* chore(just): change fixscreenshare to fix-screenshare to conform with naming scheme

* chore(just): add some bold text for title or install state

* feat(just): add quick options for use in yafti

* chore(just): fix formating for just

* chore(just): add missing quick options for setup-nix

* chore(just): add aliases for recipes

* chore(just): add empty line between alias and recipe description

* chore(just): added missing empty line for upgrade alias

* chore: Always use #!/usr/bin/bash

* chore(just): fix conditions to be lowercase for enable/disable and install/remove

* chore(just): standardize choice recipe ARG as ACTION

* feat(just): add toggle-nvk, addresses #182

* chore(just): Add message about what to do after running enroll-secure-boot-key
---
 build/ublue-os-just/00-default.just   |  32 ++--
 build/ublue-os-just/10-update.just    |  59 ++++---
 build/ublue-os-just/20-clean.just     |   2 +-
 build/ublue-os-just/30-distrobox.just |   2 +-
 build/ublue-os-just/40-nvidia.just    | 107 +++++++-----
 build/ublue-os-just/50-akmods.just    |  84 ++++++---
 build/ublue-os-just/60-custom.just    |   2 +-
 build/ublue-os-just/70-nix.just       | 241 +++++++++++++++-----------
 8 files changed, 328 insertions(+), 201 deletions(-)

diff --git a/build/ublue-os-just/00-default.just b/build/ublue-os-just/00-default.just
index b29a59bc..45c561f0 100644
--- a/build/ublue-os-just/00-default.just
+++ b/build/ublue-os-just/00-default.just
@@ -9,7 +9,7 @@ bios:
 
 # Change the user's shell
 chsh new_shell:
-    #!/usr/bin/env bash
+    #!/usr/bin/bash
     set -euo pipefail
     if [ "{{ shell }}" = "{{ new_shell }}" ] ; then
       printf "Your shell is already set to %s.\n" "{{ new_shell }}"
@@ -24,7 +24,7 @@ chsh new_shell:
 
 # Regenerate GRUB config, useful in dual-boot scenarios where a second operating system isn't listed
 regenerate-grub:
-    #!/usr/bin/env bash
+    #!/usr/bin/bash
     if [ -d /sys/firmware/efi ]; then
       sudo grub2-mkconfig -o /etc/grub2-efi.cfg
     else
@@ -39,18 +39,26 @@ changelogs:
 enroll-secure-boot-key:
     echo 'Enter password "ublue-os" if prompted after your user password.'
     sudo mokutil --import /etc/pki/akmods/certs/akmods-ublue.der --timeout -1
+    echo 'When you reboot your computer, follow the instructions to start MOK util'
+    echo 'by pressing a key, then enroll the secure boot key and enter "ublue-os" as the password'
 
-# Upgrade Distrobox to the latest git version
-distrobox-git:
-    echo 'Installing latest git snapshot of Distrobox...'
-    curl -s https://raw.githubusercontent.com/89luca89/distrobox/main/install | sh -s -- --next --prefix ~/.local
-
-# Downgrades Distrobox to the Fedora version
-remove-distrobox-git:
-    echo 'Uninstalling latest git snapshot of Distrobox...'
-    curl -s https://raw.githubusercontent.com/89luca89/distrobox/main/uninstall | sh -s -- --prefix ~/.local
+# Install or uninstall distrobox-git
+setup-distrobox-git:
+    #!/usr/bin/bash
+    echo 'Which version of Distrobox do you want to use?'
+    OPTION=$(ugum choose "Latest distrobox-git snapshot" "uBlue provided version")
+    if [ "$OPTION" == "Latest distrobox-git snapshot" ]; then
+      echo 'Installing latest git snapshot of Distrobox...'
+      curl -s https://raw.githubusercontent.com/89luca89/distrobox/main/install | sh -s -- --next --prefix ~/.local
+    elif [ "$OPTION" == "uBlue provided version" ]; then
+      echo 'Uninstalling latest git snapshot of Distrobox...'
+      curl -s https://raw.githubusercontent.com/89luca89/distrobox/main/uninstall | sh -s -- --prefix ~/.local
+    else
+      echo 'Cancelling...'
+      exit 0
+    fi
 
 # Removes homebrew from system
-brew-remove:
+remove-brew:
     echo "Removing homebrew ..."
     /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/uninstall.sh)"
diff --git a/build/ublue-os-just/10-update.just b/build/ublue-os-just/10-update.just
index cec39782..96e1f14e 100644
--- a/build/ublue-os-just/10-update.just
+++ b/build/ublue-os-just/10-update.just
@@ -1,8 +1,10 @@
 # vim: set ft=make :
 
+alias upgrade := update
+
 # Update system, flatpaks, and containers all at once
 update VERB_LEVEL="full":
-    #!/usr/bin/env bash
+    #!/usr/bin/bash
     if systemctl is-enabled ublue-update.timer | grep -q enabled;
     then
       echo "Starting the ublue-update service"
@@ -73,26 +75,39 @@ update-firmware:
     fwupdmgr get-updates
     fwupdmgr update
 
-# Disable all auto-update timers
-disable-updates:
-    #!/usr/bin/env bash
-    if systemctl is-enabled ublue-update.timer | grep -q enabled;
-    then
-      sudo systemctl disable ublue-update.timer
-    elif systemctl is-enabled flatpak-system-update.timer | grep -q enabled;
-    then
-      sudo systemctl disable --now flatpak-system-update.timer rpm-ostreed-automatic.timer
-      systemctl disable --now --user flatpak-user-update.timer
-    fi
+alias auto-update := toggle-updates
 
-# Enable all auto-update timers
-enable-updates:
-    #!/usr/bin/env bash
-    if systemctl is-enabled ublue-update.timer | grep -q disabled;
-    then
-       sudo systemctl enable ublue-update.timer
-    elif systemctl is-enabled flatpak-system-update.timer | grep -q disabled;
-    then
-      sudo systemctl enable --now flatpak-system-update.timer rpm-ostreed-automatic.timer
-      systemctl enable --now --user flatpak-user-update.timer
+# Turn automatic updates on or off
+toggle-updates ACTION="prompt":
+    #!/usr/bin/bash
+    CURRENT_STATE="Disabled"
+    if systemctl is-enabled ublue-update.timer | grep -q enabled; then
+      CURRENT_STATE="Enabled"
+    fi
+    bold=$(tput bold)
+    normal=$(tput sgr0)
+    OPTION={{ ACTION }}
+    if [ "$OPTION" == "prompt" ]; then
+      echo "Automatic updates are currently: ${bold}${CURRENT_STATE}${normal}"
+      echo "Enable or Disable automatic updates?"
+      OPTION=$(ugum choose Enable Disable)
+    elif [ "$OPTION" == "help" ]; then
+      echo "Usage: ujust toggle-updates <option>"
+      echo "  <option>: Specify the quick option - 'enable' or 'disable'"
+      echo "  Use 'enable' to Enable automatic updates."
+      echo "  Use 'disable' to Disable automatic updates."
+      exit 0
+    fi
+    if [ "${OPTION,,}" == "enable" ]; then
+      sudo systemctl enable ublue-update.timer
+      if systemctl is-enabled flatpak-system-update.timer | grep -q disabled; then
+        sudo systemctl enable --now flatpak-system-update.timer rpm-ostreed-automatic.timer
+        systemctl enable --now --user flatpak-user-update.timer
+      fi
+    elif [ "${OPTION,,}" == "disable" ]; then
+      sudo systemctl disable ublue-update.timer
+      if systemctl is-enabled flatpak-system-update.timer | grep -q enabled; then
+        sudo systemctl disable --now flatpak-system-update.timer rpm-ostreed-automatic.timer
+        systemctl disable --now --user flatpak-user-update.timer
+      fi
     fi
diff --git a/build/ublue-os-just/20-clean.just b/build/ublue-os-just/20-clean.just
index 74336c2f..1c771e10 100644
--- a/build/ublue-os-just/20-clean.just
+++ b/build/ublue-os-just/20-clean.just
@@ -2,7 +2,7 @@
 
 # Clean up old containers and flatpaks
 clean-system:
-    #!/usr/bin/env bash
+    #!/usr/bin/bash
     podman system prune -a
     flatpak uninstall --unused
     rpm-ostree cleanup -bm
diff --git a/build/ublue-os-just/30-distrobox.just b/build/ublue-os-just/30-distrobox.just
index aae16860..a28d26f9 100644
--- a/build/ublue-os-just/30-distrobox.just
+++ b/build/ublue-os-just/30-distrobox.just
@@ -7,7 +7,7 @@ distrobox-arch:
 
 # Create a Bazzite-Arch container
 distrobox-bazzite:
-    #!/usr/bin/env bash
+    #!/usr/bin/bash
     echo 'Creating Bazzite distrobox ...'
     if command -v /usr/bin/mutter > /dev/null; then
       distrobox-create --unshare-netns --nvidia --image ghcr.io/ublue-os/bazzite-arch-gnome --name bazzite-arch -Y
diff --git a/build/ublue-os-just/40-nvidia.just b/build/ublue-os-just/40-nvidia.just
index 20beda85..ab1353a0 100644
--- a/build/ublue-os-just/40-nvidia.just
+++ b/build/ublue-os-just/40-nvidia.just
@@ -1,47 +1,74 @@
 # vim: set ft=make :
 
-# Set needed kernel arguments for Nvidia GPUs
-nvidia-set-kargs:
-    #!/usr/bin/env bash
-    if command -v nvidia-smi; then
-      rpm-ostree kargs \
-        --append-if-missing=rd.driver.blacklist=nouveau \
-        --append-if-missing=modprobe.blacklist=nouveau \
-        --append-if-missing=nvidia-drm.modeset=1 \
-        --delete-if-present=nomodeset
-    else
-      echo 'You do not appear to be on a Nvidia image, please refer to the README for your uBlue-OS image.'
-    fi
+alias nvidia := configure-nvidia
 
-# Test CUDA support for Nvidia GPUs
-nvidia-test-cuda:
-    #!/usr/bin/env bash
-    if lsmod | grep -wq "nvidia"; then
-      podman run \
-        --user 1000:1000 \
-        --security-opt=no-new-privileges \
-        --cap-drop=ALL \
-        --security-opt label=type:nvidia_container_t  \
-        --device=nvidia.com/gpu=all \
-        docker.io/nvidia/samples:vectoradd-cuda11.2.1
-    else
-      echo 'The Nvidia kernel module is not loaded. You may be using secure boot without the needed signing key, lacking the needed kargs, or may not be on a Nvidia image. See "just enroll-secure-boot-key" and "just nvidia-set-kargs".'
+# Configure the Nvidia driver
+configure-nvidia ACTION="prompt":
+    #!/usr/bin/bash
+    bold=$(tput bold)
+    normal=$(tput sgr0)
+    OPTION={{ ACTION }}
+    if [ "$OPTION" == "prompt" ]; then
+      echo "${bold}Configuring Nvidia drivers${normal}"
+      echo 'What would you like to do?'
+      OPTION=$(ugum choose "Set needed kernel arguments" "Test CUDA support" "Enable Nvidia VAAPI in Firefox Flatpak")
+    elif [ "$OPTION" == "help" ]; then
+      echo "Usage: ujust configure-nvidia <option>"
+      echo "  <option>: Specify the quick option - 'kargs', 'test-cuda' or 'firefox-vaapi'"
+      echo "  Use 'kargs' to Set needed kernel arguments"
+      echo "  Use 'test-cuda' to Test CUDA support"
+      echo "  Use 'firefox-vaapi' to Enable Nvidia VAAPI in Firefox Flatpak."
+      exit 0
+    fi
+    if [ "$OPTION" == "Set needed kernel arguments" ] || [ "${OPTION,,}" == "kargs" ]; then
+      if command -v nvidia-smi; then
+        rpm-ostree kargs \
+          --append-if-missing=rd.driver.blacklist=nouveau \
+          --append-if-missing=modprobe.blacklist=nouveau \
+          --append-if-missing=nvidia-drm.modeset=1 \
+          --delete-if-present=nomodeset
+      else
+        echo 'You do not appear to be on a Nvidia image, please refer to the README for your uBlue-OS image.'
+      fi
+    elif [ "$OPTION" == "Test CUDA support" ] || [ "${OPTION,,}" == "test-cuda" ]; then
+      if lsmod | grep -wq "nvidia"; then
+        podman run \
+          --user 1000:1000 \
+          --security-opt=no-new-privileges \
+          --cap-drop=ALL \
+          --security-opt label=type:nvidia_container_t  \
+          --device=nvidia.com/gpu=all \
+          docker.io/nvidia/samples:vectoradd-cuda11.2.1
+      else
+        echo 'The Nvidia kernel module is not loaded. You may be using secure boot without the needed signing key, lacking the needed kargs, or may not be on a Nvidia image. See "just enroll-secure-boot-key" and "just nvidia-set-kargs".'
+      fi
+    elif [ "$OPTION" == "Enable Nvidia VAAPI in Firefox Flatpak" ] || [ "${OPTION,,}" == "firefox-vaapi" ]; then
+      if lsmod | grep -wq "nvidia"; then
+        flatpak override \
+          --user \
+          --filesystem=host-os \
+          --env=LIBVA_DRIVER_NAME=nvidia \
+          --env=LIBVA_DRIVERS_PATH=/run/host/usr/lib64/dri \
+          --env=LIBVA_MESSAGING_LEVEL=1 \
+          --env=MOZ_DISABLE_RDD_SANDBOX=1 \
+          --env=NVD_BACKEND=direct \
+          --env=MOZ_ENABLE_WAYLAND=1 \
+          org.mozilla.firefox
+      else
+        echo 'The Nvidia kernel module is not loaded. You may be using secure boot without the needed signing key, lacking the needed kargs, or may not be on a Nvidia image. See "just enroll-secure-boot-key" and "just nvidia-set-kargs".'
+      fi
     fi
 
-# Enable VAAPI in Firefox Flatpak for Nvidia GPUs
-nvidia-setup-firefox-vaapi:
-    #!/usr/bin/env bash
-    if lsmod | grep -wq "nvidia"; then
-      flatpak override \
-        --user \
-        --filesystem=host-os \
-        --env=LIBVA_DRIVER_NAME=nvidia \
-        --env=LIBVA_DRIVERS_PATH=/run/host/usr/lib64/dri \
-        --env=LIBVA_MESSAGING_LEVEL=1 \
-        --env=MOZ_DISABLE_RDD_SANDBOX=1 \
-        --env=NVD_BACKEND=direct \
-        --env=MOZ_ENABLE_WAYLAND=1 \
-        org.mozilla.firefox
+# Switch between Nvidia image and NVK
+toggle-nvk:
+    #!/usr/bin/bash
+    CURRENT_IMAGE=$(rpm-ostree status -b --json | jq -r '.deployments[0]."container-image-reference"' | sed -E 's/^.+\/(.+:.+)$/\1/')
+    CURRENT_URI=$(rpm-ostree status -b --json | jq -r '.deployments[0]."container-image-reference"' | sed -E 's/^(.+\/).+:.+$/\1/')
+    NEW_IMAGE=$CURRENT_IMAGE
+    if [[ "$CURRENT_IMAGE" =~ "nvidia" ]]; then
+      NEW_IMAGE=${CURRENT_IMAGE/-nvidia/}
     else
-      echo 'The Nvidia kernel module is not loaded. You may be using secure boot without the needed signing key, lacking the needed kargs, or may not be on a Nvidia image. See "just enroll-secure-boot-key" and "just nvidia-set-kargs".'
+      NEW_IMAGE=${CURRENT_IMAGE/:/-nvidia:}
     fi
+    echo "Rebasing to ${NEW_IMAGE}"
+    rpm-ostree rebase ${CURRENT_URI}${NEW_IMAGE}
diff --git a/build/ublue-os-just/50-akmods.just b/build/ublue-os-just/50-akmods.just
index f29e861f..950a195e 100644
--- a/build/ublue-os-just/50-akmods.just
+++ b/build/ublue-os-just/50-akmods.just
@@ -1,29 +1,65 @@
 # vim: set ft=make :
 
-# Use stock gamepad drivers
-use-stock-gamepads:
-    #!/usr/bin/env bash
-    sudo bash -c '> /etc/modprobe.d/xone.conf'
-    sudo bash -c '> /etc/modprobe.d/xpadneo.conf'
-    sudo bash -c '> /etc/modprobe.d/xpad-noone-blacklist.conf'
-    sudo bash -c '> /etc/modules-load.d/xpad-noone.conf'
+alias gamepads := configure-gamepads
 
-# Use uBlue gamepad drivers (Default)
-use-ublue-gamepads:
-    #!/usr/bin/env bash
-    sudo rm -f /etc/modprobe.d/xone.conf
-    sudo rm -f /etc/modprobe.d/xpadneo.conf
-    sudo rm -f /etc/modprobe.d/xpad-noone-blacklist.conf
-    sudo rm -f /etc/modules-load.d/xpad-noone.conf
+# Configure gamepad drivers
+configure-gamepads ACTION="prompt":
+    #!/usr/bin/bash
+    bold=$(tput bold)
+    normal=$(tput sgr0)
+    OPTION={{ ACTION }}
+    if [ "$OPTION" == "prompt" ]; then
+      echo "${bold}Configuring gamepad drivers${normal}"
+      echo 'Which drivers do you want to use?'
+      echo 'Note: Enabling xone and xpadneo can break some 3rd party controllers'
+      OPTION=$(ugum choose "Default Fedora drivers" "Enable xone and xpadneo")
+    elif [ "$OPTION" == "help" ]; then
+      echo "Usage: ujust configure-gamepads <option>"
+      echo "  <option>: Specify the quick option - 'stock' or 'ublue'"
+      echo "  Use 'stock' to select Default Fedora Drivers"
+      echo "  Use 'ublue' to select Enable xone and xpadneo"
+      exit 0
+    fi
+    if [ "$OPTION" == "Enable xone and xpadneo" ] || [ "$OPTION" == "ublue" ]; then
+      sudo rm -f /etc/modprobe.d/xone.conf
+      sudo rm -f /etc/modprobe.d/xpadneo.conf
+      sudo rm -f /etc/modprobe.d/xpad-noone-blacklist.conf
+      sudo rm -f /etc/modules-load.d/xpad-noone.conf
+      echo "${bold}Enabled${normal} xone and xpadneo, please reboot for changes to take effect"
+    elif [ "$OPTION" == "Default Fedora drivers" ] || [ "$OPTION" == "stock" ]; then
+      sudo bash -c '> /etc/modprobe.d/xone.conf'
+      sudo bash -c '> /etc/modprobe.d/xpadneo.conf'
+      sudo bash -c '> /etc/modprobe.d/xpad-noone-blacklist.conf'
+      sudo bash -c '> /etc/modules-load.d/xpad-noone.conf'
+      echo "${bold}Disabled${normal} xone and xpadneo, please reboot for changes to take effect"
+    fi
 
-# Disable Broadcom WL driver (Default - WL breaks numerous other Wi-Fi adapters)
-disable-broadcom-wl:
-    #!/usr/bin/env bash
-    sudo bash -c '> /etc/modules-load.d/broadcom-wl-blacklist.conf'
-    sudo bash -c 'echo "blacklist wl" > /etc/modules-load.d/default-disable-broadcom-wl.conf'
+alias broadcom-wl := configure-broadcom-wl
 
-# Use Broadcom WL driver
-use-broadcom-wl:
-    #!/usr/bin/env bash
-    sudo rm -f /etc/modprobe.d/broadcom-wl-blacklist.conf
-    sudo rm -f /etc/modprobe.d/default-disable-broadcom-wl.conf
+# Configure Broadcom WL driver (Enabling WL breaks numerous other Wi-Fi adapters)
+configure-broadcom-wl ACTION="prompt":
+    #!/usr/bin/bash
+    bold=$(tput bold)
+    normal=$(tput sgr0)
+    OPTION={{ ACTION }}
+    if [ "$OPTION" == "prompt" ]; then
+      echo "${bold}Configuring Broadcom Wi-Fi${normal}"
+      echo 'Enabling Broadcom WL driver will break numerous other Wi-Fi adapters.'
+      echo 'Enable or Disable Broadcom Wl?'
+      OPTION=$(ugum choose Enable Disable)
+    elif [ "$OPTION" == "help" ]; then
+      echo "Usage: ujust configure-broadcom-wl <option>"
+      echo "  <option>: Specify the quick option - 'enable' or 'disable'"
+      echo "  Use 'enable' to select Enable"
+      echo "  Use 'disable' to select Disable"
+      exit 0
+    fi
+    if [ "${OPTION,,}" == "enable" ]; then
+      sudo rm -f /etc/modprobe.d/broadcom-wl-blacklist.conf
+      sudo rm -f /etc/modprobe.d/default-disable-broadcom-wl.conf
+      echo "${bold}Enabled${normal} Broadcom Wireless, please reboot for changes to take effect"
+    elif [ "${OPTION,,}" == "disable" ]; then
+      sudo bash -c '> /etc/modules-load.d/broadcom-wl-blacklist.conf'
+      sudo bash -c 'echo "blacklist wl" > /etc/modules-load.d/default-disable-broadcom-wl.conf'
+      echo "${bold}Disabled${normal} Broadcom Wireless, please reboot for changes to take effect"
+    fi
diff --git a/build/ublue-os-just/60-custom.just b/build/ublue-os-just/60-custom.just
index 065a980a..e52d2773 100644
--- a/build/ublue-os-just/60-custom.just
+++ b/build/ublue-os-just/60-custom.just
@@ -1,5 +1,5 @@
 # vim: set ft=make :
 
 # Enable xwaylandvideobridge
-fixscreenshare:
+fix-screenshare:
     cp /usr/share/applications/org.kde.xwaylandvideobridge.desktop $HOME/.config/autostart/
diff --git a/build/ublue-os-just/70-nix.just b/build/ublue-os-just/70-nix.just
index 3bbe2052..198b958f 100644
--- a/build/ublue-os-just/70-nix.just
+++ b/build/ublue-os-just/70-nix.just
@@ -1,119 +1,160 @@
 # vim: set ft=make :
 
-# Install Nix with the Determinate Nix Installer
-nix-install:
-    #!/usr/bin/env bash
-    curl --proto '=https' --tlsv1.2 -sSf -L https://install.determinate.systems/nix | sudo bash -s -- install --no-confirm
-    sudo rm -f /etc/systemd/system/nix-daemon.service
-    sudo rm -f /etc/systemd/system/nix-daemon.socket
-    sudo cp /nix/var/nix/profiles/default/lib/systemd/system/nix-daemon.service /etc/systemd/system/nix-daemon.service
-    sudo cp /nix/var/nix/profiles/default/lib/systemd/system/nix-daemon.socket /etc/systemd/system/nix-daemon.socket
+alias nix := setup-nix
 
-# Uninstall Nix
-nix-remove:
-    sudo /nix/nix-installer uninstall
-    sudo rm -f /etc/systemd/system/nix-daemon.service
-    sudo rm -f /etc/systemd/system/nix-daemon.socket
-
-# Adds the nix bin path to the sudoers config
-nix-sudo:
-    #!/usr/bin/env bash
-    sudo echo "Adding sudo path variables for nix"
-    sudo rm -f /etc/sudoers.d/nix-sudo-env
-    SUDOPATHVARIABLE5=$(sudo printenv PATH)
-    sudo tee /etc/sudoers.d/nix-sudo-env <<EOF
-    Defaults  secure_path = /nix/var/nix/profiles/default/bin:/nix/var/nix/profiles/default/sbin:$SUDOPATHVARIABLE5
+# Install, uninstall or configure nix
+setup-nix ACTION="prompt":
+    #!/usr/bin/bash
+    CURRENT_STATE="Not Installed"
+    if [ -f "/etc/systemd/system/nix-daemon.socket" ]; then
+      CURRENT_STATE="Installed"
+    fi
+    bold=$(tput bold)
+    normal=$(tput sgr0)
+    OPTION={{ ACTION }}
+    if [ "$OPTION" == "prompt" ]; then
+      echo "Nix is currently: ${bold}${CURRENT_STATE}${normal}"
+      echo 'What would you like to do with Nix?'
+      OPTION=$(ugum choose \
+          Install \
+          Uninstall \
+          "Update nix user profile" \
+          "Update all nix profiles" \
+          "Re-copy nix systemd units to system" \
+          "Add nix path to sudo" \
+          "Remove nix path from sudo" \
+        )
+    elif [ "$OPTION" == "help" ]; then
+      echo "Usage: ujust setup-nix <option>"
+      echo "  <option>: Specify the quick option - 'install', 'remove', etc"
+      echo "  Use 'install' to select Install nix"
+      echo "  Use 'remove' to select Uninstall nix"
+      echo "  Use 'upgrade' to Upgrade nix user profile"
+      echo "  Use 'upgrade-all' to Upgrade all nix profiles"
+      echo "  Use 'recopy' to Re-copy nix systemd units to system"
+      echo "  Use 'sudo' to Add nix path to sudo"
+      echo "  Use 'remove-sudo' to Remove nix path from sudo."
+      exit 0
+    fi
+    if [ "${OPTION,,}" == "install" ]; then
+      curl --proto '=https' --tlsv1.2 -sSf -L https://install.determinate.systems/nix | sudo bash -s -- install --no-confirm
+      sudo rm -f /etc/systemd/system/nix-daemon.service
+      sudo rm -f /etc/systemd/system/nix-daemon.socket
+      sudo cp /nix/var/nix/profiles/default/lib/systemd/system/nix-daemon.service /etc/systemd/system/nix-daemon.service
+      sudo cp /nix/var/nix/profiles/default/lib/systemd/system/nix-daemon.socket /etc/systemd/system/nix-daemon.socket
+    elif [ "${OPTION,,}" == "uninstall" ] || [ "${OPTION,,}" == "remove" ]; then
+      sudo /nix/nix-installer uninstall
+      sudo rm -f /etc/systemd/system/nix-daemon.service
+      sudo rm -f /etc/systemd/system/nix-daemon.socket
+    elif [ "$OPTION" == "Update nix user profile" ] || [ "${OPTION,,}" == "upgrade" ]; then
+      nix profile upgrade '.*'
+    elif [ "$OPTION" == "Update all nix profiles" ] || [ "${OPTION,,}" == "upgrade-all" ]; then
+      nix profile upgrade '.*'
+      sudo nix profile upgrade '.*'
+    elif [ "$OPTION" == "Re-copy nix systemd units to system" ] || [ "${OPTION,,}" == "recopy" ]; then
+      sudo rm -f /etc/systemd/system/nix-daemon.service
+      sudo rm -f /etc/systemd/system/nix-daemon.socket
+      sudo cp /nix/var/nix/profiles/default/lib/systemd/system/nix-daemon.service /etc/systemd/system/nix-daemon.service
+      sudo cp /nix/var/nix/profiles/default/lib/systemd/system/nix-daemon.socket /etc/systemd/system/nix-daemon.socket
+    elif [ "$OPTION" == "Add nix path to sudo" ] || [ "${OPTION,,}" == "sudo" ]; then
+      sudo echo "Adding sudo path variables for nix"
+      sudo rm -f /etc/sudoers.d/nix-sudo-env
+      SUDOPATHVARIABLE5=$(sudo printenv PATH)
+      sudo tee /etc/sudoers.d/nix-sudo-env <<EOF
+      Defaults  secure_path = /nix/var/nix/profiles/default/bin:/nix/var/nix/profiles/default/sbin:$SUDOPATHVARIABLE5
     EOF
-    sudo echo "Finished adding sudo variables for nix"
+      sudo echo "Finished adding sudo variables for nix"
+    elif [ "$OPTION" == "Remove nix path from sudo" ] || [ "${OPTION,,}" == "remove-sudo" ]; then
+      sudo rm -f /etc/sudoers.d/nix-sudo-env
+    fi
 
-# Removes nix bin path from sudoers config
-nix-sudo-undo:
-    sudo rm -f /etc/sudoers.d/nix-sudo-env
+alias fleek := setup-fleek
 
-# Install Fleek (https://getfleek.dev/)
-fleek-install:
-    #!/usr/bin/env bash
-    if [[ -d "/nix" ]]; then
+# Install, uninstall or configure Fleek (https://getfleek.dev/)
+setup-fleek ACTION="prompt":
+    #!/usr/bin/bash
+    if [ ! -f "/etc/systemd/system/nix-daemon.socket" ]; then
+      echo "Please install nix first by running ujust setup-nix"
+      exit 0
+    fi
+    OPTION={{ ACTION }}
+    if [ "$OPTION" == "prompt" ]; then
+      echo 'What would you like to do with Fleek?'
+      OPTION=$(ugum choose "Install Fleek" "Uninstall Fleek" "Integrate Nix/Fleek applications with system theme")
+    elif [ "$OPTION" == "help" ]; then
+      echo "Usage: ujust setup-fleek <option>"
+      echo "  <option>: Specify the quick option - 'install', 'remove', or 'integrate'"
+      echo "  Use 'install' to select install"
+      echo "  Use 'remove' to select uninstall"
+      echo "  Use 'integrate' to integrate nix applications with system theme."
+      exit 0
+    fi
+    if [ "$OPTION" == "Install Fleek" ] || [ "${OPTION,,} == "install" ]; then
       mkdir -p $HOME/.config/nix
       echo "experimental-features = nix-command flakes" >> $HOME/.config/nix/nix.conf
       curl -fsSL https://getfleek.dev/installer | env FORCE=1 bash
-    else
-      echo "Install nix with ujust nix-install before running ujust fleek-install."
-    fi
-
-# Removes Fleek
-fleek-remove:
-    #!/usr/bin/env bash
-    if [[ -x "/var/usrlocal/bin/fleek" ]]; then
-      sudo rm -f /var/usrlocal/bin/fleek
-    else
-      echo "Fleek is not installed or has been already removed"
-    fi
-
-# Integrate Nix/Fleek applications with system theme
-fleek-themes-install:
-    #!/usr/bin/env bash
-    if [[ -x "/var/usrlocal/bin/fleek" ]]; then
-      if command -v mutter; then
-        fleek add adw-gtk3 --apply
+    elif [ "$OPTION" == "Uninstall Fleek" ] || [ "${OPTION,,} == "remove" ]; then
+      if [[ -x "/var/usrlocal/bin/fleek" ]]; then
+        sudo rm -f /var/usrlocal/bin/fleek
+      else
+        echo "Fleek is not installed or has been already removed"
+      fi
+    elif [ "$OPTION" == "Integrate Nix/Fleek applications with system theme" ] || [ "${OPTION,,} == "integrate" ]; then
+      if [[ -x "/var/usrlocal/bin/fleek" ]]; then
+        if command -v mutter; then
+          fleek add adw-gtk3 --apply
+        else
+          fleek add breeze-icons
+          fleek add libsForQt5.breeze-qt5
+          fleek apply
+        fi
       else
-        fleek add breeze-icons
-        fleek add libsForQt5.breeze-qt5
-        fleek apply
+        echo "Install fleek with ujust fleek-install before running ujust fleek-themes-install."
       fi
-    else
-      echo "Install fleek with ujust fleek-install before running ujust fleek-themes-install."
     fi
 
-# Installs Devbox
-devbox-install:
-    #!/usr/bin/env bash
-    if [[ -d "/nix" ]]; then
-      curl -fsSL https://get.jetpack.io/devbox | bash
-    else
-      echo "Install nix with ujust nix-install before running ujust devbox-install."
-    fi
+alias devbox := setup-devbox
 
-# Removes Devbox
-devbox-remove:
-    #!/usr/bin/env bash
-    if [[ -x "/var/usrlocal/bin/devbox" ]]; then
-      sudo rm -f /var/usrlocal/bin/devbox
-    else
-      echo "Devbox is not installed or has been already removed"
+# Install or uninstall devbox
+setup-devbox ACTION="prompt":
+    #!/usr/bin/bash
+    if [ ! -f "/etc/systemd/system/nix-daemon.socket" ]; then
+      echo "Please install nix first by running ujust setup-nix"
+      exit 0
     fi
-
-# Installs nixGL
-nixgl-install:
-    #!/usr/bin/env bash
-    if [[ -d "/nix" ]]; then
-      nix profile install github:guibou/nixGL --impure
-    else
-      echo "NixGL requires nix to install."
+    CURRENT_STATE="Not Installed"
+    if [ -f "/var/usrlocal/bin/devbox" ]; then
+      CURRENT_STATE="Installed"
     fi
-
-# Upgrades user nix profile
-nix-update-user:
-    #!/usr/bin/env bash
-    if [[ -d "/nix" ]]; then
-      nix profile upgrade '.*'
-    else
-      echo "Nix is not installed"
+    bold=$(tput bold)
+    normal=$(tput sgr0)
+    OPTION={{ ACTION }}
+    if [ "$OPTION" == "prompt" ]; then
+      echo "Devbox is currently: ${bold}${CURRENT_STATE}${normal}"
+      echo 'What would you like to do with devbox?'
+      OPTION=$(ugum choose Install Uninstall)
+    elif [ "$OPTION" == "help" ]; then
+      echo "Usage: ujust toggle-updates <option>"
+      echo "  <option>: Specify the quick option - 'install' or 'remove'"
+      echo "  Use 'install' to Install devbox"
+      echo "  Use 'remove' to Remove devbox."
+      exit 0
     fi
-
-# Upgrades both user and root nix profiles
-nix-update-all:
-    #!/usr/bin/env bash
-    if [[ -d "/nix" ]]; then
-      nix profile upgrade '.*'
-      sudo nix profile upgrade '.*'
-    else
-      echo "Nix is not installed"
+    if [ "${OPTION,,}" == "install" ]; then
+      curl -fsSL https://get.jetpack.io/devbox | bash
+    elif [ "${OPTION,,}" == "uninstall" ] || [ "${OPTION,,}" == "remove" ]; then
+      if [[ -x "/var/usrlocal/bin/devbox" ]]; then
+        sudo rm -f /var/usrlocal/bin/devbox
+      else
+        echo "Devbox is not installed or has been already removed"
+      fi
     fi
 
-# Deletes and recoppies nix systemd units to ensure that they are the same
-nix-systemd-copy:
-    sudo rm -f /etc/systemd/system/nix-daemon.service
-    sudo rm -f /etc/systemd/system/nix-daemon.socket
-    sudo cp /nix/var/nix/profiles/default/lib/systemd/system/nix-daemon.service /etc/systemd/system/nix-daemon.service
-    sudo cp /nix/var/nix/profiles/default/lib/systemd/system/nix-daemon.socket /etc/systemd/system/nix-daemon.socket
+# Installs nixGL
+install-nixgl:
+    #!/usr/bin/bash
+    if [ ! -f "/etc/systemd/system/nix-daemon.socket" ]; then
+      echo "Please install nix first by running ujust setup-nix"
+      exit 0
+    fi
+    nix profile install github:guibou/nixGL --impure