setup docker with multi-stage multi manifests

Author: Anmol1696

.github/workflows/docker.yml

@@ -12,9 +12,24 @@ env:
   IMAGE_NAME: constructive-io/docker/postgres-plus
   PG_VERSION: '17'
 
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}-docker-postgres-plus
+  cancel-in-progress: true
+
 jobs:
-  build:
-    runs-on: ubuntu-latest
+  build-postgres-plus:
+    strategy:
+      fail-fast: false
+      matrix:
+        include:
+          - platform: linux/amd64
+            arch: amd64
+            runner: ubuntu-latest
+          - platform: linux/arm64
+            arch: arm64
+            runner: ubuntu-24.04-arm
+    runs-on: ${{ matrix.runner }}
+
     permissions:
       contents: read
       packages: write
@@ -23,9 +38,6 @@ jobs:
       - name: Checkout
         uses: actions/checkout@v4
 
-      - name: Set up QEMU
-        uses: docker/setup-qemu-action@v3
-
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
 
@@ -49,15 +61,117 @@ jobs:
             type=semver,pattern={{version}},enable=${{ startsWith(github.ref, 'refs/tags/v') }}
             type=semver,pattern={{major}}.{{minor}},enable=${{ startsWith(github.ref, 'refs/tags/v') }}
 
-      - name: Build and push
+      - name: Build (no push, PR only, amd64)
+        if: github.event_name == 'pull_request' && matrix.platform == 'linux/amd64'
         uses: docker/build-push-action@v6
         with:
           context: .
-          platforms: linux/amd64,linux/arm64
-          push: ${{ github.event_name != 'pull_request' }}
+          file: ./Dockerfile
+          platforms: linux/amd64
+          push: false
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
           build-args: |
             PG_VERSION=${{ env.PG_VERSION }}
           cache-from: type=gha
           cache-to: type=gha,mode=max
+
+      - name: Build & push by digest
+        if: github.event_name != 'pull_request'
+        id: build
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          file: ./Dockerfile
+          platforms: ${{ matrix.platform }}
+          labels: ${{ steps.meta.outputs.labels }}
+          outputs: type=image,name=${{ env.REGISTRY }}/${{ env.IMAGE_NAME }},push-by-digest=true,push=true
+          build-args: |
+            PG_VERSION=${{ env.PG_VERSION }}
+          cache-from: type=gha
+          cache-to: type=gha,mode=max
+
+      - name: Export digest
+        if: github.event_name != 'pull_request'
+        run: |
+          mkdir -p "${{ runner.temp }}/digests"
+          digest="${{ steps.build.outputs.digest }}"
+          touch "${{ runner.temp }}/digests/${digest#sha256:}"
+
+      - name: Upload digest
+        if: github.event_name != 'pull_request'
+        uses: actions/upload-artifact@v4
+        with:
+          name: digests-${{ matrix.arch }}
+          path: ${{ runner.temp }}/digests/*
+
+  publish-postgres-plus-manifest:
+    if: github.event_name != 'pull_request'
+    runs-on: ubuntu-latest
+    needs: build-postgres-plus
+
+    permissions:
+      contents: read
+      packages: write
+
+    steps:
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Log in to Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ${{ env.REGISTRY }}
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Download digests
+        uses: actions/download-artifact@v4
+        with:
+          pattern: digests-*
+          path: ${{ runner.temp }}/digests
+          merge-multiple: true
+
+      - name: Extract metadata
+        id: meta
+        uses: docker/metadata-action@v5
+        with:
+          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
+          tags: |
+            type=raw,value=latest,enable=${{ github.ref == 'refs/heads/main' }}
+            type=raw,value=${{ env.PG_VERSION }},enable=${{ github.ref == 'refs/heads/main' }}
+            type=sha,format=short,prefix=
+            type=semver,pattern={{version}},enable=${{ startsWith(github.ref, 'refs/tags/v') }}
+            type=semver,pattern={{major}}.{{minor}},enable=${{ startsWith(github.ref, 'refs/tags/v') }}
+
+      - name: Create and push multi-arch manifests
+        run: |
+          set -euo pipefail
+
+          image="${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}"
+          digest_dir="${{ runner.temp }}/digests"
+
+          if [ ! -d "$digest_dir" ]; then
+            echo "No digests directory found at $digest_dir"
+            exit 1
+          fi
+
+          digests=""
+          for digest_file in "$digest_dir"/*; do
+            digest="$(basename "$digest_file")"
+            digests="$digests $image@sha256:$digest"
+          done
+
+          if [ -z "$digests" ]; then
+            echo "No digests found to create manifest"
+            exit 1
+          fi
+
+          echo "Creating manifests for tags:"
+          echo "${{ steps.meta.outputs.tags }}"
+
+          echo "${{ steps.meta.outputs.tags }}" | while read -r tag; do
+            [ -z "$tag" ] && continue
+            echo "Creating multi-arch manifest for $tag"
+            docker buildx imagetools create -t "$tag" $digests
+          done