From 3108517ca4ce47a441cb28281e97b04cfdd3b5f4 Mon Sep 17 00:00:00 2001
From: bobslept <38557801+bobslept@users.noreply.github.com>
Date: Sat, 12 Aug 2023 15:52:54 +0200
Subject: [PATCH] feat(just): add devmode-on/off

---
 usr/share/ublue-os/just/custom.just | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/usr/share/ublue-os/just/custom.just b/usr/share/ublue-os/just/custom.just
index fe8568bc9c4..3fd2da20320 100644
--- a/usr/share/ublue-os/just/custom.just
+++ b/usr/share/ublue-os/just/custom.just
@@ -59,6 +59,32 @@ cockpit:
 code-profile:
   xdg-open https://vscode.dev/profile/github/c761b7738e9a7b02286d6d94cb2d1ecd
 
+# Rebase to a non developer bluefin image
+devmode-off:
+  #!/usr/bin/env bash
+  CURRENT_IMAGE=$(rpm-ostree status -b --json | jq -r '.deployments[0]."container-image-reference"')
+  if grep -q "bluefin-dx" <<< $CURRENT_IMAGE
+  then
+      echo "Rebasing to a non developer image"
+      NEW_IMAGE=$(echo $CURRENT_IMAGE | sed "s/bluefin-dx/bluefin/")
+      rpm-ostree rebase $NEW_IMAGE
+  else
+      echo "You are currently not on a developer image"
+  fi
+
+# Rebase to a developer bluefin image
+devmode-on:
+  #!/usr/bin/env bash
+  CURRENT_IMAGE=$(rpm-ostree status -b --json | jq -r '.deployments[0]."container-image-reference"')
+  if grep -q "bluefin-dx" <<< $CURRENT_IMAGE
+  then
+      echo "You are already on a developer image"
+  else
+      echo "Rebasing to a developer image"
+      NEW_IMAGE=$(echo $CURRENT_IMAGE | sed "s/bluefin/bluefin-dx/")
+      rpm-ostree rebase $NEW_IMAGE
+  fi
+
 distrobox-mlbox:
   echo 'Assembling pytorch-nvidia mlbox distrobox ...'
   distrobox assemble create --file /usr/share/ublue-os/distrobox/pytorch-nvidia.ini