ci(image): adjust build flow

.github/workflows/build-image.yml

@@ -1,12 +1,19 @@
-name: Build image manually
+name: Build image
 
 on:
+  push:
+    branches:
+      - main
   workflow_dispatch:
     inputs:
       docker_image_tag_name:
         type: string
         description: Docker image tag name (Optional)
 
+env:
+  ENGINE_IMAGE: ghcr.io/canner/wren-engine
+  IBIS_IMAGE: ghcr.io/canner/wren-engine-ibis
+
 jobs:
   prepare-tag:
     runs-on: ubuntu-latest
@@ -15,14 +22,21 @@ jobs:
       - name: Prepare tag name
         id: prepare_tag
         run: |
-          if [ -n "${{ github.event.inputs.docker_image_tag_name }}" ]; then
-            tag_name=${{ github.event.inputs.docker_image_tag_name }}
-          else
-            tag_name=$(echo ${{ github.ref_name }} | sed 's/[^a-zA-Z0-9]/-/g')-$(git log -1 --pretty=%h)
+          if [ "${{ github.event_name }}" = "push" ]; then
+            echo "type=sha" > tags.txt
+            echo "type=raw,value=nightly" > tags.txt
+          fi
+          if [ "${{ github.event_name }}" = "workflow_dispatch" ]; then
+            if [ -n "${{ github.event.inputs.docker_image_tag_name }}" ]; then
+              tag_name=${{ github.event.inputs.docker_image_tag_name }}
+            else
+              tag_name=$(echo ${{ github.ref_name }} | sed 's/[^a-zA-Z0-9]/-/g')-$(git log -1 --pretty=%h)
+            fi
+            echo "type=raw,value=$tag_name" > tags.txt
           fi
-          echo "tag_name=$tag_name" >> "$GITHUB_OUTPUT"
+          echo "tags=$(cat tags.txt)" >> $GITHUB_OUTPUT
     outputs:
-      tag_name: ${{ steps.prepare_tag.outputs.tag_name }}
+      tags: ${{ steps.prepare_tag.outputs.tags }}
   build-image:
     needs: prepare-tag
     runs-on: ubuntu-latest
@@ -41,9 +55,8 @@ jobs:
         id: meta
         uses: docker/metadata-action@v5
         with:
-          images: ghcr.io/canner/wren-engine
-          tags: |
-            type=raw,value=${{ needs.prepare-tag.outputs.tag_name }}
+          images: ${{ env.ENGINE_IMAGE }}
+          tags: ${{ needs.prepare-tag.outputs.tags }}
       - name: Login to GitHub Container Registry
         uses: docker/login-action@v3
         with:
@@ -69,67 +82,89 @@ jobs:
           build-args: |
             WREN_VERSION=${{ steps.prepare.outputs.WREN_VERSION }}
           push: true
-  build-ibis-amd64-image:
+  build-ibis-image:
     needs: prepare-tag
-    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        arch:
+          - runner: ubuntu-latest
+            platform: linux/amd64
+          - runner: linux_arm64_runner
+            platform: linux/arm64
+    runs-on: ${{ matrix.arch.runner }}
     steps:
+      - name: Prepare platform
+        run: |
+          platform=${{ matrix.arch.platform }}
+          echo "PLATFORM_PAIR=${platform//\//-}" >> $GITHUB_ENV
       - uses: actions/checkout@v4
       - name: Docker meta
         id: meta
         uses: docker/metadata-action@v5
         with:
-          images: ghcr.io/canner/wren-engine-ibis
-          tags: |
-            type=raw,value=${{ needs.prepare-tag.outputs.tag_name }}
+          images: ${{ env.IBIS_IMAGE }}
       - name: Login to GitHub Container Registry
         uses: docker/login-action@v3
         with:
           registry: ghcr.io
           username: ${{ github.actor }}
           password: ${{ secrets.GITHUB_TOKEN }}
-      - name: Set up QEMU
-        uses: docker/setup-qemu-action@v3
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
-      - name: Build and push
-        uses: docker/build-push-action@v5
+      - name: Build and push by digest
+        id: build
+        uses: docker/build-push-action@v6
         with:
+          platforms: ${{ matrix.arch.platform }}
+          labels: ${{ steps.meta.outputs.labels }}
           context: ./ibis-server
           build-contexts: |
             wren-core-py=./wren-core-py
             wren-core=./wren-core
-          platforms: linux/amd64
-          push: true
-          tags: ${{ steps.meta.outputs.tags }}
-  build-ibis-arm64-image:
-    needs: prepare-tag
+          outputs: type=image,name=${{ env.IBIS_IMAGE }},push-by-digest=true,name-canonical=true,push=true
+      - name: Export digest
+        run: |
+          mkdir -p /tmp/digests
+          digest="${{ steps.build.outputs.digest }}"
+          touch "/tmp/digests/${digest#sha256:}"
+      - name: Upload digest
+        uses: actions/upload-artifact@v4
+        with:
+          name: digests-${{ env.PLATFORM_PAIR }}
+          path: /tmp/digests/*
+          if-no-files-found: error
+          retention-days: 1
+  merge:
     runs-on: ubuntu-latest
+    needs: [ prepare-tag, build-ibis-image ]
     steps:
-      - uses: actions/checkout@v4
+      - name: Download digests
+        uses: actions/download-artifact@v4
+        with:
+          path: /tmp/digests
+          pattern: digests-*
+          merge-multiple: true
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
       - name: Docker meta
         id: meta
         uses: docker/metadata-action@v5
         with:
-          images: ghcr.io/canner/wren-engine-ibis
-          tags: |
-            type=raw,value=${{ needs.prepare-tag.outputs.tag_name }}
+          images: ${{ env.IBIS_IMAGE }}
+          tags: ${{ needs.prepare-tag.outputs.tags }}
       - name: Login to GitHub Container Registry
         uses: docker/login-action@v3
         with:
           registry: ghcr.io
           username: ${{ github.actor }}
           password: ${{ secrets.GITHUB_TOKEN }}
-      - name: Set up QEMU
-        uses: docker/setup-qemu-action@v3
-      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v3
-      - name: Build and push
-        uses: docker/build-push-action@v5
-        with:
-          context: ./ibis-server
-          build-contexts: |
-            wren-core-py=./wren-core-py
-            wren-core=./wren-core
-          platforms: linux/arm64
-          push: true
-          tags: ${{ steps.meta.outputs.tags }}
+      - name: Create manifest list and push
+        working-directory: /tmp/digests
+        run: |
+          docker buildx imagetools create $(jq -cr '.tags | map("-t " + .) | join(" ")' <<< "$DOCKER_METADATA_OUTPUT_JSON") \
+            $(printf '${{ env.IBIS_IMAGE }}@sha256:%s ' *) \
+            --tag ${{ steps.meta.outputs.tags }}
+      - name: Inspect image
+        run: |
+          docker buildx imagetools inspect ${{ steps.meta.outputs.tags }}