name: Docker Image CI

on:
  push:
    branches: [ '**' ]
    tags: ['*']
  pull_request:
    branches: [ "main" ]

jobs:

  build:

    runs-on: ubuntu-latest

    strategy:
      matrix:
        config:
          - tag: ""
            dockerfile: "./Dockerfile.default"
            platforms: "linux/amd64,linux/arm64"
            build_base: "ubuntu:22.04"
            runtime_base: "ubuntu:22.04"


          - tag: "-openblas"
            dockerfile: "./Dockerfile.default"
            platforms: "linux/amd64,linux/arm64" 
            build_base: "ubuntu:22.04"
            runtime_base: "ubuntu:22.04"
            extra_deps: "libopenblas-dev"
            cmake_args: "-DGGML_OPENBLAS=On"

            
          - tag: "-cuda11-7"
            dockerfile: "./Dockerfile.default"
            platforms: "linux/amd64"
            build_base: "nvidia/cuda:11.7.1-devel-ubuntu22.04"
            runtime_base: "nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu22.04"
            cmake_args: "-DGGML_CUBLAS=ON -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc" 
            
          - tag: "-cuda12-0"
            dockerfile: "./Dockerfile.default"
            platforms: "linux/amd64"
            build_base: "nvidia/cuda:12.0.0-devel-ubuntu22.04"
            runtime_base: "nvidia/cuda:12.0.0-runtime-ubuntu22.04"
            cmake_args: "-DGGML_CUBLAS=ON -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc" 
            
          - tag: "-cuda12-2"
            dockerfile: "./Dockerfile.default"
            platforms: "linux/amd64"
            build_base: "nvidia/cuda:12.2.0-devel-ubuntu22.04"
            runtime_base: "nvidia/cuda:12.2.0-runtime-ubuntu22.04"
            cmake_args: "-DGGML_CUBLAS=ON -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc" 
          
          - tag: "-clblast"
            dockerfile: "./Dockerfile.default"
            platforms: "linux/amd64"
            build_base: "ubuntu:22.04"
            runtime_base: "ubuntu:22.04"
            extra_deps: libclblast-dev
            cmake_args: DGGML_CLBLAST=On
            

    steps:
    
    - name: Checkout
      uses: actions/checkout@v3
      with:
        submodules: true

    # Add support for more platforms with QEMU (optional)
    # https://github.com/docker/setup-qemu-action
    - name: Set up QEMU
      uses: docker/setup-qemu-action@v2

    - name: Set up Docker BuildX
      uses: docker/setup-buildx-action@v2
      with:
        platforms: linux/amd64,linux/arm64,linux/arm68/v8

    - name: Login to GH ContainerHub
      uses: docker/login-action@v2
      with:
        registry: ghcr.io
        username: ravenscroftj
        password: ${{ secrets.GH_TOKEN }}

    - name: Build and push incremental
      uses: docker/build-push-action@v4.1.1
      if: (!startsWith(github.ref, 'refs/tags/'))
      with:
        file: ${{matrix.config.dockerfile}}
        push: true
        tags: ghcr.io/ravenscroftj/turbopilot:nightly${{matrix.config.tag}}-${{ github.sha }}
        context: ${{github.workspace}}
        platforms: ${{matrix.config.platforms}}
        build-args: |
          EXTRA_DEPS=${{matrix.config.extra_deps}}
          CMAKE_ARGS=${{matrix.config.cmake_args}}
          BUILD_BASE=${{matrix.config.build_base}}
          RUNTIME_BASE=${{matrix.config.runtime_base}}


    - name: Build and push release (Main Latest Build)
      uses: docker/build-push-action@v4
      if: startsWith(github.ref, 'refs/tags/') && matrix.config.tag == ''
      with:
        file: ${{matrix.config.dockerfile}}
        push: true
        tags: ghcr.io/ravenscroftj/turbopilot:${{ github.ref_name }}, ghcr.io/ravenscroftj/turbopilot:latest
        context: ${{github.workspace}}
        platforms: ${{matrix.config.platforms}} 
        build-args: ${{matrix.config.build_args}}


    - name: Build and push release (Accelerated Builds)
      uses: docker/build-push-action@v4
      if: startsWith(github.ref, 'refs/tags/') && matrix.config.tag != ''
      with:
        file: ${{matrix.config.dockerfile}}
        push: true
        tags: ghcr.io/ravenscroftj/turbopilot:${{ github.ref_name }}${{matrix.config.tag}}
        context: ${{github.workspace}}
        platforms: ${{matrix.config.platforms}}
        build-args: ${{matrix.config.build_args}}