name: Docker Build and Release Upload

on:
  push:
    branches:
      - main

jobs:

  # Build ComfyUI-Cuda
  build-cuda:
    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: build and push comfyui-cuda
        run: |
          docker buildx build \
          --platform=linux/arm64 \
          -t mistox/comfyui-cuda-ampere:latest \
          -f comfyui-cuda-dockerfile \
          --push \
          .

  # 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: build and push comfyui-rocm
        run: |
          docker buildx build \
          --platform=linux/arm64 \
          -t mistox/comfyui-rocm-ampere:latest \
          -f comfyui-rocm-dockerfile \
          --push \
          .

  # Job 3: Uvicorn Build
  build-uvicorn:
    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: build and push uvicorn
        run: |
          docker buildx build \
          --platform=linux/arm64 \
          -t mistox/uvicorn-ampere:latest \
          -f uvicorn-dockerfile \
          --push \
          .