#!/usr/bin/env bash

build() {
  cd "$(
    cd "$(dirname "$0")" > /dev/null 2>&1
    pwd -P
  )" || exit 9

  local image_name="${1:-zinit}"
  local tag="${2:-latest}"
  local zsh_version="${3}"
  shift 3

  local dockerfile="../docker/Dockerfile"

  if [[ -n $zsh_version ]]; then
    tag="zsh${zsh_version}-${tag}"
  fi

  echo -e "\e[34mBuilding image: ${image_name}\e[0m" >&2

  local -a args
  [[ -n $NO_CACHE ]] && args+=(--no-cache "$@")

  if docker build \
    --build-arg "PUSERNAME=$(id -u -n)" \
    --build-arg "PUID=$(id -u)" \
    --build-arg "PGID=$(id -g)" \
    --build-arg "TERM=${TERM:-xterm-256color}" \
    --build-arg "ZINIT_ZSH_VERSION=${zsh_version}" \
    --file "$dockerfile" \
    --tag "${image_name}:${tag}" \
    "${args[@]}" \
    "$(realpath ..)"; then
    {
      echo -e "\e[34mTo use this image for zunit tests run: \e[0m"
      echo -e "\e[34mexport CONTAINER_IMAGE=\"${image_name}\" CONTAINER_TAG=\"${tag}\"\e[0m"
      echo -e "\e[34mzunit run --verbose\e[0m"
    } >&2
  else
    echo -e "\e[31m❌ Container failed to build.\e[0m" >&2
    return 1
  fi
}

if [[ ${BASH_SOURCE[0]} == "${0}" ]]; then
  BUILD_ZSH_VERSION="${BUILD_ZSH_VERSION:-}"
  CONTAINER_IMAGE="${CONTAINER_IMAGE:-ghcr.io/zdharma-continuum/zinit}"
  CONTAINER_TAG="${CONTAINER_TAG:-latest}"
  NO_CACHE="${NO_CACHE:-}"

  while [[ -n $* ]]; do
    case "$1" in
      --image | -i)
        CONTAINER_IMAGE="$2"
        shift 2
        ;;
      --no-cache | -N)
        NO_CACHE=1
        shift
        ;;
      --zsh-version | -zv | --zv)
        BUILD_ZSH_VERSION="${2}"
        shift 2
        ;;
      *)
        break
        ;;
    esac
  done

  build "${CONTAINER_IMAGE}" "${CONTAINER_TAG}" "${BUILD_ZSH_VERSION}" "$@"
fi

# Local Variables:
# mode: Shell-Script
# sh-indentation: 2
# indent-tabs-mode: nil
# sh-basic-offset: 2
# End:
# vim: ft=bash sw=2 ts=2 et