name: Publish Docker Images

on:
  push:
    tags:
      - "v*"
  workflow_dispatch:

permissions:
  contents: read
  packages: write

concurrency:
  group: docker-publish-${{ github.ref }}
  cancel-in-progress: true

env:
  REGISTRY: ghcr.io
  IMAGE_NAME: tatertotterson/microwakeword

jobs:
  docker:
    name: Docker image
    runs-on: ubuntu-latest
    steps:
      - name: Check out repository
        uses: actions/checkout@v4

      - name: Set up Docker Buildx
        uses: docker/setup-buildx-action@v3

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

      - name: Docker metadata
        id: meta
        uses: docker/metadata-action@v5
        with:
          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
          flavor: |
            latest=false
          tags: |
            type=raw,value=latest
            type=ref,event=tag

      - name: Build and push image
        uses: docker/build-push-action@v6
        with:
          context: .
          file: dockerfile
          platforms: linux/amd64
          push: true
          tags: ${{ steps.meta.outputs.tags }}
          labels: ${{ steps.meta.outputs.labels }}
          cache-from: type=gha,scope=mww-trainer-nvidia-docker
          cache-to: type=gha,mode=max,scope=mww-trainer-nvidia-docker

      - name: Docker metadata (Blackwell)
        id: meta-blackwell
        uses: docker/metadata-action@v5
        with:
          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
          flavor: |
            latest=false
          tags: |
            type=raw,value=blackwell
            type=ref,event=tag,suffix=-blackwell

      - name: Build and push Blackwell image
        uses: docker/build-push-action@v6
        with:
          context: .
          file: dockerfile.blackwell
          platforms: linux/amd64
          push: true
          tags: ${{ steps.meta-blackwell.outputs.tags }}
          labels: ${{ steps.meta-blackwell.outputs.labels }}
          cache-from: type=gha,scope=mww-trainer-nvidia-docker-blackwell
          cache-to: type=gha,mode=max,scope=mww-trainer-nvidia-docker-blackwell