intermediate-docker.yml

name: Intermediate Docker

on:
  workflow_dispatch:
    inputs:
      GO_VERSION:
        description: "Go version"
        required: true
        type: choice
        options:
          - "1.20"
          - "1.21"
        default: "1.21"
      RUST_VERSION:
        description: "Rust toolchain version"
        required: true
        type: choice
        options:
          - nightly-2023-12-03
          - nightly-2022-12-10
        default: "nightly-2023-12-03"
      PYTHON_VERSION:
        description: "Python version"
        required: false
        type: choice
        options:
          - "3.10"
        default: "3.10"
      CUDA_VERSION:
        description: "Cuda version"
        required: false
        type: choice
        options:
          - "11.7.1"
          - "12.2.2"
        default: "11.7.1"
      BASE_IMAGE:
        description: "which intermediate image you want to update"
        required: true
        default: "go-alpine-builder"
        type: choice
        options:
          - cuda-go-rust-builder
          - go-rust-builder
          - go-alpine-builder
          - rust-builder
          - rust-alpine-builder
          - go-rust-alpine-builder
          - py-runner

defaults:
  run:
    working-directory: "build/dockerfiles/intermediate"

jobs:
  setup:
    runs-on: ubuntu-latest
    steps:
      - name: Checkout code
        uses: actions/checkout@v2
      - name: Set up QEMU
        uses: docker/setup-qemu-action@v2
      - name: Set up Docker Buildx
        uses: docker/setup-buildx-action@v2
      - name: Login to Docker Hub
        uses: docker/login-action@v2
        with:
          username: ${{ secrets.DOCKERHUB_USERNAME }}
          password: ${{ secrets.DOCKERHUB_TOKEN }}
      - name: set tag env
        run: |
          if [ ${{github.event.inputs.BASE_IMAGE}} == "cuda-go-rust-builder" ]; then
            echo "TAG=cuda-${{ github.event.inputs.CUDA_VERSION }}-go-${{ github.event.inputs.GO_VERSION }}-rust-${{ github.event.inputs.RUST_VERSION }}" >> $GITHUB_ENV
          elif [ ${{github.event.inputs.BASE_IMAGE}} == "go-rust-builder" ]; then
            echo "TAG=go-${{ github.event.inputs.GO_VERSION }}-rust-${{ github.event.inputs.RUST_VERSION }}" >> $GITHUB_ENV
          elif [ ${{github.event.inputs.BASE_IMAGE}} == "go-alpine-builder" ]; then
            echo "TAG=${{ github.event.inputs.GO_VERSION }}" >> $GITHUB_ENV
          elif [ ${{github.event.inputs.BASE_IMAGE}} == "rust-builder" ]; then
            echo "TAG=${{ github.event.inputs.RUST_VERSION }}" >> $GITHUB_ENV
          elif [ ${{github.event.inputs.BASE_IMAGE}} == "rust-alpine-builder" ]; then
            echo "TAG=${{ github.event.inputs.RUST_VERSION }}" >> $GITHUB_ENV
          elif [ ${{github.event.inputs.BASE_IMAGE}} == "go-rust-alpine-builder" ]; then
            echo "TAG=go-${{ github.event.inputs.GO_VERSION }}-rust-${{ github.event.inputs.RUST_VERSION }}" >> $GITHUB_ENV
          elif [ ${{github.event.inputs.BASE_IMAGE}} == "py-runner" ]; then
            echo "TAG=${{ github.event.inputs.PYTHON_VERSION }}" >> $GITHUB_ENV
          else
            echo "no BASE_IMAGE match"
          fi
  build-and-publish-intermediate-amd64:
    runs-on: ubuntu-latest
    needs: setup
    steps:
      - name: Build amd64 image
        id: build-amd64
        uses: docker/build-push-action@v5
        with:
          context: .
          push: true
          file: build/dockerfiles/intermediate/${{ github.event.inputs.BASE_IMAGE }}.Dockerfile
          platforms: linux/amd64
          tags: scrolltech/${{ github.event.inputs.BASE_IMAGE }}:${{ env.TAG }}
          env:
            PLATFORM: arm64
          build-args: |
            CUDA_VERSION=${{ github.event.inputs.CUDA_VERSION }}
            GO_VERSION=${{ github.event.inputs.GO_VERSION }}
            RUST_VERSION=${{ github.event.inputs.RUST_VERSION }}
            PYTHON_VERSION=${{ github.event.inputs.PYTHON_VERSION }}

  build-and-publish-intermediate-arm64:
    runs-on: ubuntu-latest
    needs: setup
    steps:
      - name: Build arm64 image
        id: build-arm64
        uses: docker/build-push-action@v5
        needs: setup
        env:
          PLATFORM: arm64
        with:
          context: .
          push: true
          file: build/dockerfiles/intermediate/${{ github.event.inputs.BASE_IMAGE }}.Dockerfile
          platforms: linux/arm64
          tags: scrolltech/${{ github.event.inputs.BASE_IMAGE }}:${{ env.TAG }}
          build-args: |
            CUDA_VERSION=${{ github.event.inputs.CUDA_VERSION }}
            GO_VERSION=${{ github.event.inputs.GO_VERSION }}
            RUST_VERSION=${{ github.event.inputs.RUST_VERSION }}
            PYTHON_VERSION=${{ github.event.inputs.PYTHON_VERSION }}