From 582a378c2af5d656809a282dc2e28c5f69b35b4b Mon Sep 17 00:00:00 2001
From: Adam Fidel <ledif@users.noreply.github.com>
Date: Sat, 1 Mar 2025 23:49:59 -0600
Subject: [PATCH] feat: expand recipe to rollback broken NVIDIA images (#250)

* feat: expand recipe to rollback broken NVIDIA images

* fix just lint error

* improve comments
---
 Justfile | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/Justfile b/Justfile
index 697a0229..6b5a0b38 100644
--- a/Justfile
+++ b/Justfile
@@ -871,16 +871,20 @@ tag-images image_name="" default_tag="" tags="":
     # Show Images
     ${PODMAN} images
 
-# TODO: generalize this
-# Used on Jan 28 2025 to fix NVIDIA regression
-#   > just retag-stable-daily-nvidia-on-ghcr stable-daily-41.20250126.3 0
+# # Examples:
+#   > just retag-nvidia-on-ghcr stable-daily stable-daily-41.20250126.3 0
+#   > just retag-nvidia-on-ghcr latest latest-41.20250228.1 0
 #
-# Need to generate a PAT with package write access (https://github.com/settings/tokens)
-# Set $GITHUB_USERNAME and $GITHUB_PAT variables
+# working_tag: The tag of the most recent known good image (e.g., stable-daily-41.20250126.3)
+# stream:      One of latest, stable-daily, stable or gts
+# dry_run:     Only print the skopeo commands instead of running them
+#
+# First generate a PAT with package write access (https://github.com/settings/tokens)
+# and set $GITHUB_USERNAME and $GITHUB_PAT environment variables
 
 # Retag images on GHCR
 [group('Admin')]
-retag-stable-daily-nvidia-on-ghcr working_tag="" dry_run="1":
+retag-nvidia-on-ghcr working_tag="" stream="" dry_run="1":
     #!/bin/bash
     set -euxo pipefail
     skopeo="echo === skopeo"
@@ -889,5 +893,5 @@ retag-stable-daily-nvidia-on-ghcr working_tag="" dry_run="1":
         skopeo="skopeo"
     fi
     for image in aurora-nvidia-open aurora-nvidia aurora-dx-nvidia aurora-dx-nvidia-open; do
-      $skopeo copy docker://ghcr.io/ublue-os/${image}:{{ working_tag }} docker://ghcr.io/ublue-os/${image}:stable-daily
+      $skopeo copy docker://ghcr.io/ublue-os/${image}:{{ working_tag }} docker://ghcr.io/ublue-os/${image}:{{ stream }}
     done