[ci][docker] Tag tlcpackstaging images to tlcpack

See apache#11768, this PR changes the deploy workflow so that after a successful build with fallback images (see apache#11775), they get moved over to tlcpack automatically
driazati · Jun 22, 2022 · e7fb0c2 · e7fb0c2
1 parent 1e0e954
commit e7fb0c2
Show file tree

Hide file tree

Showing 2 changed files with 124 additions and 1 deletion.
diff --git a/Jenkinsfile b/Jenkinsfile
diff --git a/jenkins/Deploy.groovy.j2 b/jenkins/Deploy.groovy.j2
@@ -86,6 +86,23 @@ def deploy() {
         }
       }
     }
+    if (env.BRANCH_NAME == 'main' && env.RETAG_STAGING_IMAGES == 'yes') {
+      {% 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 "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') }}) {