#!/bin/bash

# Builds extra Livebook images.
#
# Images are pushed to GHCR, so you need to make sure that the docker
# client is authenticated.

set -ex
cd "$(dirname "$0")/.."

. versions

livebook_version="$(cat mix.exs | grep '@version "' | grep -o '".*"' | tr -d '"')"

if [ -z "$livebook_version" ] || [[ "$livebook_version" == *"dev"* ]]; then
  echo "No releasable Livebook version detected"
  exit 1
fi

if [[ "$livebook_version" == *"-"* ]]; then
  stable=false
else
  stable=true
fi

cuda_tag_list=("cuda11.8" "cuda12.1")
cuda_version_list=("11-8" "12-1")

for idx in "${!cuda_tag_list[@]}"; do
  cuda_tag="${cuda_tag_list[idx]}"
  cuda_version="${cuda_version_list[idx]}"

  base_image="ghcr.io/livebook-dev/utils:elixir-$elixir-erlang-$otp-$cuda_tag"

  if docker manifest inspect $base_image > /dev/null; then
    echo "Using base image: $base_image"
  else
    echo "Building base image: $base_image"

    docker buildx build --push --platform linux/amd64,linux/arm64 \
      -t $base_image \
      --build-arg ELIXIR_VERSION=$elixir \
      --build-arg ERLANG_VERSION=$otp \
      --build-arg UBUNTU_VERSION=$ubuntu \
      --build-arg CUDA_VERSION=$cuda_version \
      -f docker/base/elixir-cuda.dockerfile \
      docker/base
  fi

  image="ghcr.io/livebook-dev/livebook:$livebook_version-$cuda_tag"

  if docker manifest inspect $image > /dev/null; then
    echo "Skipping image, since it already exists: $image"
  else
    echo "Building image: $image"

    extra_args=""

    if [[ "$stable" == "true" ]]; then
      extra_args="-t ghcr.io/livebook-dev/livebook:latest-$cuda_tag"
    fi

    docker buildx build --push --platform linux/amd64,linux/arm64 \
      --build-arg BASE_IMAGE="$base_image" \
      -t $image \
      $extra_args \
      .
  fi
done