[ci][docker] Tag tlcpackstaging images to tlcpack

See #11768, this PR changes the deploy workflow so that after a successful build with fallback images (see #11775), they get moved over to tlcpack automatically
apache · Aug 11, 2022 · 74c537b · 74c537b
1 parent ecfd969
commit 74c537b
Show file tree

Hide file tree

Showing 2 changed files with 156 additions and 1 deletion.
diff --git a/Jenkinsfile b/Jenkinsfile
diff --git a/ci/jenkins/Deploy.groovy.j2 b/ci/jenkins/Deploy.groovy.j2
@@ -90,6 +90,39 @@ def deploy() {
         }
       }
     }
+    if (env.BRANCH_NAME == 'main' && env.RETAG_STAGING_IMAGES == 'yes') {
+      withCredentials([string(
+        credentialsId: 'dockerhub-tlcpack-key',
+        variable: 'TLCPACK_TOKEN',
+        )]) {
+        try {
+          sh(
+            script: 'echo $TLCPACK_TOKEN | docker login --username octomldriazati --password-stdin',
+            label: 'Log in to Docker Hub'
+          )
+          {% for image in images %}
+          if ({{ image.name }}.contains("tlcpackstaging")) {
+            // Push image to tlcpack
+            def tag = {{ image.name }}.split(":")[1]
+            sh(
+              script: """
+                set -eux
+                docker pull tlcpackstaging/{{ image.name }}:${tag}
+                docker tag tlcpackstaging/{{ image.name }}:${tag} tlcpack/{{ image.name.replace("_", "-") }}:${tag}
+                docker push tlcpack/{{ image.name.replace("_", "-") }}:${tag}
+              """,
+              label: 'Tag tlcpackstaging/{{ image.name }} image to tlcpack',
+            )
+          }
+          {% endfor %}
+        } finally {
+          sh(
+            script: 'docker logout',
+            label: 'Clean up login credentials'
+          )
+        }
+      }
+    }
     if (env.BRANCH_NAME == 'main' && env.DEPLOY_DOCKER_IMAGES == 'yes' && rebuild_docker_images && upstream_revision != null) {
       node('CPU') {
         ws({{ m.per_exec_ws('tvm/deploy-docker') }}) {