name: Build and Publish Docker Images

on:
  push:
    branches: [main]

jobs:
  build-cpu:
    runs-on: ubuntu-latest
    steps:
      - name: Checkout repository
        uses: actions/checkout@v4

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

      - name: Log in to Gitea Container Registry
        uses: docker/login-action@v3
        with:
          registry: git.toomuchtaco.net
          username: ${{ gitea.actor }}
          password: ${{ secrets.PACKING_TOKEN }}

      - name: Build and push CPU image
        uses: docker/build-push-action@v6
        with:
          context: .
          file: Dockerfile
          push: true
          tags: |
            git.toomuchtaco.net/${{ gitea.repository }}:latest
            git.toomuchtaco.net/${{ gitea.repository }}:cpu

  build-cuda:
    runs-on: ubuntu-latest
    steps:
      - name: Checkout repository
        uses: actions/checkout@v4

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

      - name: Log in to Gitea Container Registry
        uses: docker/login-action@v3
        with:
          registry: git.toomuchtaco.net
          username: ${{ gitea.actor }}
          password: ${{ secrets.PACKING_TOKEN }}

      - name: Build and push CUDA image
        uses: docker/build-push-action@v6
        with:
          context: .
          file: Dockerfile.cuda
          push: true
          tags: git.toomuchtaco.net/${{ gitea.repository }}:cuda

  build-rocm:
    runs-on: ubuntu-latest
    steps:
      - name: Checkout repository
        uses: actions/checkout@v4

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

      - name: Log in to Gitea Container Registry
        uses: docker/login-action@v3
        with:
          registry: git.toomuchtaco.net
          username: ${{ gitea.actor }}
          password: ${{ secrets.PACKING_TOKEN }}

      - name: Build and push ROCm image
        uses: docker/build-push-action@v6
        with:
          context: .
          file: Dockerfile.rocm
          push: true
          tags: git.toomuchtaco.net/${{ gitea.repository }}:rocm