From fe3a2b870a702625c6dcb27647fc82a915c54d8e Mon Sep 17 00:00:00 2001
From: hgherzog <henryh@allenai.org>
Date: Wed, 16 Oct 2024 13:26:09 -0700
Subject: [PATCH] improve passing the image around

---
 .github/workflows/build_test.yaml | 22 ++++++++++++----------
 docker-compose.yaml               |  2 +-
 2 files changed, 13 insertions(+), 11 deletions(-)

diff --git a/.github/workflows/build_test.yaml b/.github/workflows/build_test.yaml
index 03cc2334..8a38398e 100644
--- a/.github/workflows/build_test.yaml
+++ b/.github/workflows/build_test.yaml
@@ -19,6 +19,8 @@ jobs:
     permissions:
       contents: read
       packages: write
+    outputs:
+        ghcr_docker_image: ${{ steps.image-names.outputs.ghcr_image_name }}
     steps:
       - name: Checkout repository
         uses: actions/checkout@v4
@@ -41,10 +43,6 @@ jobs:
             type=sha,format=short
             type=raw,value=latest,enable={{is_default_branch}}
 
-      - name: Extract image tag
-        id: extract-tag
-        run: echo "IMAGE_TAG=$(echo '${{ steps.meta.outputs.tags }}' | cut -d',' -f1)" >> $GITHUB_ENV
-
       - name: Build and push Docker image
         id: build-push
         uses: docker/build-push-action@v6
@@ -57,6 +55,15 @@ jobs:
             GIT_USERNAME=${{ secrets.GIT_USERNAME }}
             GIT_TOKEN=${{ secrets.GIT_TOKEN }}
 
+      - name: Store Image Names
+        # We need the docker image name downstream in test & deploy. This saves the full docker image names to outputs
+        id: image-names
+        run: |-
+          GHCR_IMAGE="${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}@${{ steps.build-push.outputs.digest }}"
+          GHCR_IMAGE=`echo ${GHCR_IMAGE} | tr '[:upper:]' '[:lower:]'` # docker requires that all image names be lowercase
+          echo "ghcr.io Docker image name is ${GHCR_IMAGE}"
+          echo "ghcr_image_name=\"${GHCR_IMAGE}\"" >> $GITHUB_OUTPUT
+
   test:
     needs: build
     runs-on: ubuntu-latest
@@ -79,17 +86,12 @@ jobs:
         run: |
           curl -H "Authorization: Bearer ${{ secrets.GITHUB_TOKEN }}" \
                  https://api.github.com/orgs/allenai/packages/container/rslearn_projects/versions
-      - name: Pull the latest image
-        run: |
-          docker pull ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:${{ env.IMAGE_TAG }}
 
       - name: Run tests with Docker Compose
         run: |
           docker-compose -f docker-compose.yaml up --exit-code-from tests
         env:
-          IMAGE_TAG: ${{ env.IMAGE_TAG }}
-          REGISTRY: ${{ env.REGISTRY }}
-          IMAGE_NAME: ${{ env.IMAGE_NAME }}
+          IMAGE_NAME: ${{ needs.build.outputs.ghcr_docker_image }}
 
       - name: Clean up
         if: always()
diff --git a/docker-compose.yaml b/docker-compose.yaml
index 68b13d6c..69a4a161 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -2,5 +2,5 @@ version: "3.9"
 
 services:
   tests:
-    image: ${REGISTRY}/${IMAGE_NAME}:${IMAGE_TAG}
+    image: ${IMAGE_NAME}
     command: python -m pytest tests/