build-image.yml

name: Build and push images

on:
  workflow_dispatch:
  release:
    types: [published]

env:
  PYTHON_VERSION: 3.9
  REGISTRY: ghcr.io
  BASE_IMAGE: emma-heriot-watt/base:latest
  BUILDER_IMAGE: emma-heriot-watt/builder:latest
  REPOSITORY: emma-heriot-watt/perception

jobs:
  build_image:
    name: Build and push image
    runs-on: ubuntu-latest
    strategy:
      matrix:
        torch_version_suffix: ["", "+cu113"]

    permissions:
      contents: read
      packages: write

    steps:
      - name: Define image suffix
        id: image-tag
        env:
          TORCH_VERSION_SUFFIX: ${{ matrix.torch_version_suffix }}
        run: |
          IMAGE_TAG=$(echo "$TORCH_VERSION_SUFFIX" | tr + -)
          echo "tag=$IMAGE_TAG" >> "$GITHUB_OUTPUT"

      - name: Check out the repository
        uses: actions/checkout@v4

      - name: Log in to the Container registry
        uses: docker/login-action@v3
        with:
          registry: ${{ env.REGISTRY }}
          username: ${{ github.actor }}
          password: ${{ secrets.GITHUB_TOKEN }}

      - name: Extract metadata (tags, labels) for Docker
        id: meta
        uses: docker/metadata-action@v5
        with:
          images: ${{ env.REGISTRY }}/${{ env.REPOSITORY }}
          tags: |
            type=pep440,enable=true,priority=900,suffix=${{ steps.image-tag.outputs.tag }}

      - name: Build and push image
        uses: docker/build-push-action@v5
        with:
          context: .
          push: true
          tags: ${{ steps.meta.outputs.tags }}
          labels: ${{ steps.meta.outputs.labels }}
          build-args: |
            BASE_IMAGE_NAME=${{ env.REGISTRY }}/${{ env.BASE_IMAGE }}
            BUILDER_IMAGE_NAME=${{ env.REGISTRY }}/${{ env.BUILDER_IMAGE }}
            TORCH_VERSION_SUFFIX=${{ matrix.torch_version_suffix }}