name: Build and Push spark-vllm

on:
  push:
    branches:
      - main
  workflow_dispatch:

env:
  IMAGE_NAME: spark-vllm
  IMAGE_TAG: latest

jobs:
  docker:
    runs-on: dgx

    steps:
      - name: Checkout
        uses: actions/checkout@v4

      - name: Log in to Gitea Container Registry
        uses: docker/login-action@v3
        with:
          registry: ${{ github.server_url }}
          username: ${{ secrets.CI_USER }}
          password: ${{ secrets.CI_PASSWORD }}


      - name: Build image using upstream script
        env:
          DOCKER_DEFAULT_PLATFORM: linux/arm64
        run: |
          bash build-and-copy.sh -t ${IMAGE_NAME}:${IMAGE_TAG}

      - name: Tag and Push Docker Image
        shell: bash
        run: |
          VERSION=${GITEA_SHA::7}

          REGISTRY=${GITHUB_SERVER_URL#https://}

          TARGET_IMAGE=$REGISTRY/${{ github.repository_owner }}/${IMAGE_NAME}

          docker tag \
            ${IMAGE_NAME}:${IMAGE_TAG} \
            $TARGET_IMAGE:$VERSION

          docker tag \
            ${IMAGE_NAME}:${IMAGE_TAG} \
            $TARGET_IMAGE:latest

          docker push $TARGET_IMAGE:$VERSION
          docker push $TARGET_IMAGE:latest