[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 · Jul 1, 2022 · ee5709b · ee5709b
1 parent ec39199
commit ee5709b
Show file tree

Hide file tree

Showing 2 changed files with 134 additions and 1 deletion.
diff --git a/Jenkinsfile b/Jenkinsfile
diff --git a/ci/jenkins/Deploy.groovy.j2 b/ci/jenkins/Deploy.groovy.j2
@@ -86,6 +86,28 @@ def deploy() {
         }
       }
     }
+    if (env.BRANCH_NAME == 'main' && env.RETAG_STAGING_IMAGES == 'yes') {
+      withCredentials([string(
+        credentialsId: 'dockerhub-tlcpack-key',
+        variable: 'TLCPACK_TOKEN',
+        )]) {
+        {% for image in images %}
+        if ({{ image.name }}.contains("tlcpackstaging")) {
+          // Push image to tlcpack
+          def tag = {{ image.name }}.split(":")[1]
+          sh(
+            script: """
+              set -eux
+              CONTENT_TYPE="application/vnd.docker.distribution.manifest.v2+json"
+              MANIFEST=\$(curl -H "Accept: \${CONTENT_TYPE}" "https://hub.docker.com/v2/tlcpackstaging/{{ image.name }}/manifests/${tag}")
+              curl -X PUT -H "Authorization: Bearer \${TLCPACK_TOKEN}" -H "Content-Type: \${CONTENT_TYPE}" -d "\${MANIFEST}" "https://hub.docker.com/v2/tlcpack/{{ image.name.replace("_", "-") }}/manifests/${tag}"
+            """,
+            label: 'Tag tlcpackstaging image to tlcpack',
+          )
+        }
+        {% endfor %}
+      }
+    }
     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') }}) {