name: Docker Build and Release Upload

on:
  push:
    branches:
      - torchaudio-builder

jobs:

  # Build ComfyUI-Rocm
  build-rocm:
    runs-on: alpine-linux
    steps:
      - name: checkout
        uses: actions/checkout@v4

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

      - name: Log in to DockerHub
        uses: docker/login-action@v3
        with:
          username: ${{ secrets.DOCKERHUB_USERNAME }}
          password: ${{ secrets.DOCKERHUB_TOKEN }}

      - name: Log in to docker.mistox.net
        uses: docker/login-action@v3
        with:
          registry: docker.mistox.net
          username: ${{ secrets.DOCKERHUB_USERNAME }}
          password: ${{ secrets.DOCKERHUB_TOKEN }}

      - name: build and push rocm-aarch64-base
        run: |
          docker buildx build \
          --platform=linux/arm64 \
          -t docker.mistox.net/rocm-aarch64-base:latest \
          -f rocm-base-aarch64-dockerfile \
          --push \
          .

      - name: build and push comfyui-rocm
        run: |
          docker buildx build \
          --platform=linux/arm64 \
          -t docker.mistox.net/comfyui-rocm-ampere:latest \
          -f comfyui-rocm-dockerfile \
          --push \
          .

      - name: build and push comfyui-cuda
        run: |
          docker buildx build \
          --platform=linux/arm64 \
          -t docker.mistox.net/comfyui-cuda-ampere:latest \
          -f comfyui-cuda-dockerfile \
          --push \
          .
      
      - name: build and push uvicorn
        run: |
          docker buildx build \
          --platform=linux/arm64 \
          -t docker.mistox.net/uvicorn-ampere:latest \
          -f uvicorn-dockerfile \
          --push \
          .