From b343db9ad572825aca3f8460ff75a4de06689c4c Mon Sep 17 00:00:00 2001
From: Ian Rodney <ian.rodney@gmail.com>
Date: Wed, 18 Nov 2020 12:14:44 -0800
Subject: [PATCH] [docker] Modify script to allow for arbitrary name changes
 (#12092)

---
 docker/fix-docker-latest.sh | 70 ++++++++++++++++++++-----------------
 1 file changed, 38 insertions(+), 32 deletions(-)

diff --git a/docker/fix-docker-latest.sh b/docker/fix-docker-latest.sh
index fdf547865..dbfd34874 100755
--- a/docker/fix-docker-latest.sh
+++ b/docker/fix-docker-latest.sh
@@ -1,18 +1,24 @@
 #!/bin/bash
 # This script is not for normal use and is used in the event that CI (or a user) overwrites the latest tag.
+set -x
 
 IMAGE="1.0.0"
+DEST="latest"
 
 while [[ $# -gt 0 ]]
 do
 key="$1"
 case $key in
-    --latest-tag)
+    --source-tag)
     shift
     IMAGE=$1
     ;;
+    --dest-tag)
+    shift
+    DEST=$1
+    ;;
     *)
-    echo "Usage: fix-docker-latest.sh --latest-tag <TAG>"
+    echo "Usage: fix-docker-latest.sh --source-tag <TAG> --dest-tag <LATEST>"
     exit 1
 esac
 shift
@@ -21,46 +27,46 @@ done
 echo "You must be logged into a user with push priviledges to do this."
 
 docker pull rayproject/ray:"$IMAGE"
-docker tag rayproject/ray:"$IMAGE" rayproject/ray:latest
-docker tag rayproject/ray:"$IMAGE" rayproject/ray:latest-cpu
+docker tag rayproject/ray:"$IMAGE" rayproject/ray:"$DEST"
+docker tag rayproject/ray:"$IMAGE" rayproject/ray:"$DEST"-cpu
 docker pull rayproject/ray:"$IMAGE"-gpu
-docker tag rayproject/ray:"$IMAGE"-gpu rayproject/ray:latest-gpu
-docker push rayproject/ray:latest
-docker push rayproject/ray:latest-cpu
-docker push rayproject/ray:latest-gpu
+docker tag rayproject/ray:"$IMAGE"-gpu rayproject/ray:"$DEST"-gpu
+docker push rayproject/ray:"$DEST"
+docker push rayproject/ray:"$DEST"-cpu
+docker push rayproject/ray:"$DEST"-gpu
 
 docker pull rayproject/ray-deps:"$IMAGE"
-docker tag rayproject/ray-deps:"$IMAGE" rayproject/ray-deps:latest
-docker tag rayproject/ray-deps:"$IMAGE" rayproject/ray-deps:latest-cpu
+docker tag rayproject/ray-deps:"$IMAGE" rayproject/ray-deps:"$DEST"
+docker tag rayproject/ray-deps:"$IMAGE" rayproject/ray-deps:"$DEST"-cpu
 docker pull rayproject/ray-deps:"$IMAGE"-gpu
-docker tag rayproject/ray-deps:"$IMAGE"-gpu rayproject/ray-deps:latest-gpu
-docker push rayproject/ray-deps:latest
-docker push rayproject/ray-deps:latest-cpu
-docker push rayproject/ray-deps:latest-gpu
+docker tag rayproject/ray-deps:"$IMAGE"-gpu rayproject/ray-deps:"$DEST"-gpu
+docker push rayproject/ray-deps:"$DEST"
+docker push rayproject/ray-deps:"$DEST"-cpu
+docker push rayproject/ray-deps:"$DEST"-gpu
 
 docker pull rayproject/base-deps:"$IMAGE"
-docker tag rayproject/base-deps:"$IMAGE" rayproject/base-deps:latest
-docker tag rayproject/base-deps:"$IMAGE" rayproject/base-deps:latest-cpu
+docker tag rayproject/base-deps:"$IMAGE" rayproject/base-deps:"$DEST"
+docker tag rayproject/base-deps:"$IMAGE" rayproject/base-deps:"$DEST"-cpu
 docker pull rayproject/base-deps:"$IMAGE"-gpu
-docker tag rayproject/base-deps:"$IMAGE"-gpu rayproject/base-deps:latest-gpu
-docker push rayproject/base-deps:latest
-docker push rayproject/base-deps:latest-cpu
-docker push rayproject/base-deps:latest-gpu
+docker tag rayproject/base-deps:"$IMAGE"-gpu rayproject/base-deps:"$DEST"-gpu
+docker push rayproject/base-deps:"$DEST"
+docker push rayproject/base-deps:"$DEST"-cpu
+docker push rayproject/base-deps:"$DEST"-gpu
 
 docker pull rayproject/ray-ml:"$IMAGE"
-docker tag rayproject/ray-ml:"$IMAGE" rayproject/ray-ml:latest
-docker tag rayproject/ray-ml:"$IMAGE" rayproject/ray-ml:latest-cpu
+docker tag rayproject/ray-ml:"$IMAGE" rayproject/ray-ml:"$DEST"
+docker tag rayproject/ray-ml:"$IMAGE" rayproject/ray-ml:"$DEST"-cpu
 docker pull rayproject/ray-ml:"$IMAGE"-gpu
-docker tag rayproject/ray-ml:"$IMAGE"-gpu rayproject/ray-ml:latest-gpu
-docker push rayproject/ray-ml:latest
-docker push rayproject/ray-ml:latest-cpu
-docker push rayproject/ray-ml:latest-gpu
+docker tag rayproject/ray-ml:"$IMAGE"-gpu rayproject/ray-ml:"$DEST"-gpu
+docker push rayproject/ray-ml:"$DEST"
+docker push rayproject/ray-ml:"$DEST"-cpu
+docker push rayproject/ray-ml:"$DEST"-gpu
 
 docker pull rayproject/autoscaler:"$IMAGE"
-docker tag rayproject/autoscaler:"$IMAGE" rayproject/autoscaler:latest
-docker tag rayproject/autoscaler:"$IMAGE" rayproject/autoscaler:latest-cpu
+docker tag rayproject/autoscaler:"$IMAGE" rayproject/autoscaler:"$DEST"
+docker tag rayproject/autoscaler:"$IMAGE" rayproject/autoscaler:"$DEST"-cpu
 docker pull rayproject/autoscaler:"$IMAGE"-gpu
-docker tag rayproject/autoscaler:"$IMAGE"-gpu rayproject/autoscaler:latest-gpu
-docker push rayproject/autoscaler:latest
-docker push rayproject/autoscaler:latest-cpu
-docker push rayproject/autoscaler:latest-gpu
+docker tag rayproject/autoscaler:"$IMAGE"-gpu rayproject/autoscaler:"$DEST"-gpu
+docker push rayproject/autoscaler:"$DEST"
+docker push rayproject/autoscaler:"$DEST"-cpu
+docker push rayproject/autoscaler:"$DEST"-gpu