docker.yml

name: Publish Docker image

on:
  # Allow manual runs
  workflow_dispatch:
  
  # Only run for push on the main branch or for tagged version
  push:
    branches:
      - master
    tags:
      - v*.*.*

env:
  REGISTRY: ghcr.io
  IMAGE_NAME: ${{ github.repository }}


permissions:
  packages: write


# define build arguments

jobs:
  build-image:
    strategy:
      fail-fast: false
      matrix:
        include:
          - arch: cuda
          - arch: rocm

    runs-on: [self-hosted, "${{ matrix.arch }}"]

    permissions:
      contents: read
      packages: write

    steps:
      - name: Show all images
        run: |
          docker image ls

      - name: Prune
        run: |
          # Prune all images older than 2 weeks
          # The images are still on github registry
          docker image prune -f -a --filter "until=336h"
          docker system prune -f

      - name: Check out the repo
        uses: actions/checkout@v3
      
      - name: Get Image Tag Name
        env:
          GITHUB_REF_NAME_ENV: ${{ github.ref_name }}
        run: |
          REGEX="(.*)v(.*)\.(.*)\.(.*)"
          IMAGE_TAG="nightly"
          if [[ "${GITHUB_REF_NAME_ENV}" =~ $REGEX ]]; then
              IMAGE_TAG="${GITHUB_REF_NAME##*/}"
          fi
          echo "IMAGE_TAG=$IMAGE_TAG" >> $GITHUB_ENV
    
      - name: Log in to the registry
        uses: docker/login-action@v2
        with:
          registry: ${{ env.REGISTRY }}
          username: ${{ github.actor }}
          password: ${{ secrets.GITHUB_TOKEN }}
      
      - name: Extract metadata (tags, labels) for the image
        id: meta
        uses: docker/metadata-action@v4
        with:
          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
          tags: |
            type=raw,value=${{ matrix.arch }}-${{ env.IMAGE_TAG }}

      - name: Build and push the image
        uses: docker/build-push-action@v3
        with:
          context: .
          push: true
          file: docker/Dockerfile-${{ matrix.arch }}
          tags: ${{ steps.meta.outputs.tags }}
          labels: ${{ steps.meta.outputs.labels }}
          build-args: |
            ARCH=${{ matrix.arch }}
            CONFIG=standard.yaml