sleap_cuda_production.yml

name: Build and Push sleap-cuda (Production Workflow)

# Run on push to main branch after testing is complete
on:
  push:
    branches:
      - main
    paths:
      - sleap_cuda/** # Only run on changes to sleap_cuda
      - .github/workflows/sleap_cuda_production.yml # Only run on changes to this workflow

jobs:
  build:
    runs-on: ubuntu-latest # Only build on Ubuntu for now since Docker is not available on macOS runners
    strategy:
      matrix:
        platform: [linux/amd64] # Only build amd64 for now
      max-parallel: 2  # Build both architectures in parallel (if more than one)
    outputs:
      git_sha: ${{ steps.get_sha.outputs.sha }}
      sanitized_platform: ${{ steps.sanitize_platform.outputs.sanitized_platform }}
    steps:
      - name: Checkout code
        # https://github.com/actions/checkout
        uses: actions/checkout@v4

      - name: Get Git SHA
        id: get_sha
        run: echo "sha=$(git rev-parse HEAD)" >> $GITHUB_OUTPUT

      - name: Debug Git SHA
        run: echo "Git SHA ${{ steps.get_sha.outputs.sha }}"

      # Generate a sanitized platform string with slashes replaced by dashes
      - name: Sanitize platform name
        id: sanitize_platform
        run: |
          sanitized_platform="${{ matrix.platform }}" # Copy platform value
          sanitized_platform="${sanitized_platform/\//-}" # Replace / with -
          echo "sanitized_platform=$sanitized_platform" >> $GITHUB_OUTPUT

      - name: Set up Docker Buildx
        # https://github.com/docker/setup-buildx-action
        uses: docker/setup-buildx-action@v3
        with:
          driver: docker-container # Use a container driver for Buildx (default)

      - name: Log in to Docker Hub
        # https://github.com/docker/login-action
        uses: docker/login-action@v3
        with:
          username: ${{ vars.DOCKERHUB_USERNAME }}
          password: ${{ secrets.DOCKERHUB_TOKEN }}

      - name: Build and push Docker image
        # https://github.com/docker/build-push-action
        uses: docker/build-push-action@v6
        with:
          context: ./sleap_cuda # Build context wrt the root of the repository
          file: ./sleap_cuda/Dockerfile # Path to Dockerfile wrt the root of the repository
          platforms: ${{ matrix.platform }}
          push: true # Push the image to Docker Hub
          # Tags for the production images, including the "latest" tag
          tags: |
            ${{ vars.DOCKERHUB_USERNAME }}/sleap-cuda:latest
            ${{ vars.DOCKERHUB_USERNAME }}/sleap-cuda:${{ steps.sanitize_platform.outputs.sanitized_platform }}
            ${{ vars.DOCKERHUB_USERNAME }}/sleap-cuda:${{ steps.sanitize_platform.outputs.sanitized_platform }}-nvidia-cuda-11.3.1-cudnn8-runtime-ubuntu20.04
            ${{ vars.DOCKERHUB_USERNAME }}/sleap-cuda:${{ steps.sanitize_platform.outputs.sanitized_platform }}-sleap-1.3.4
            ${{ vars.DOCKERHUB_USERNAME }}/sleap-cuda:${{ steps.sanitize_platform.outputs.sanitized_platform }}-${{ steps.get_sha.outputs.sha }}