From 7be0ee44a5cea13172f3a61417ae5e035577a7ad Mon Sep 17 00:00:00 2001
From: Adam Fidel <adam@fidel.id>
Date: Fri, 31 Jan 2025 20:06:07 -0600
Subject: [PATCH] fix(just): toggle-nvk now supports nvidia-open

---
 build/ublue-os-just/40-nvidia.just | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/build/ublue-os-just/40-nvidia.just b/build/ublue-os-just/40-nvidia.just
index 5527ad52..25a0b9ab 100644
--- a/build/ublue-os-just/40-nvidia.just
+++ b/build/ublue-os-just/40-nvidia.just
@@ -64,10 +64,12 @@ toggle-nvk:
     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
+    if [[ "$CURRENT_IMAGE" =~ "nvidia-open" ]]; then
+      NEW_IMAGE=${CURRENT_IMAGE/-nvidia-open/}
+    elif [[ "$CURRENT_IMAGE" =~ "nvidia" ]]; then
       NEW_IMAGE=${CURRENT_IMAGE/-nvidia/}
     else
-      NEW_IMAGE=${CURRENT_IMAGE/:/-nvidia:}
+      NEW_IMAGE=${CURRENT_IMAGE/:/-nvidia-open:}
     fi
     echo "Rebasing to ${NEW_IMAGE}"
     rpm-ostree rebase ${CURRENT_URI}${NEW_IMAGE}