#!/usr/bin/env bash
set -euo pipefail

command_name="${1:-build}"
requested_arch="${2:-${ALPINE_ARCH:-x86_64}}"
repo_root="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
. "${repo_root}/scripts/apk/package-lib.sh"

alpine_version="${ALPINE_VERSION:-3.23}"
repo_name="${ALPINE_REPO_NAME:-local}"
build_platform="${ALPINE_BUILD_PLATFORM:-linux/amd64}"
builder_uid="$(id -u)"
builder_gid="$(id -g)"

if [[ "${builder_uid}" == "0" ]]; then
  builder_uid=1000
fi
if [[ "${builder_gid}" == "0" ]]; then
  builder_gid=1000
fi

validate_arch() {
  case "$1" in
    x86_64|aarch64) ;;
    *) printf 'unsupported Alpine architecture: %s\n' "$1" >&2; return 2 ;;
  esac
}

run_for_arch() {
  local arch="$1"
  local subcommand="$2"
  local package_dir="$3"
  local image_name
  local container_package_dir

  validate_arch "${arch}"
  apk_validate_package_dir "${package_dir}"
  container_package_dir="$(apk_container_package_dir "${repo_root}" "${package_dir}")"
  image_name="${ALPINE_APK_BUILDER_IMAGE:-alpine-apk-builder:${arch}}"

  docker build \
    --platform "${build_platform}" \
    --build-arg "ALPINE_VERSION=${alpine_version}" \
    --build-arg "BUILDER_UID=${builder_uid}" \
    --build-arg "BUILDER_GID=${builder_gid}" \
    -f "${repo_root}/scripts/apk/Dockerfile" \
    -t "${image_name}" \
    "${repo_root}"

  docker_args=(--rm --platform "${build_platform}")
  if [[ -t 0 && -t 1 ]]; then
    docker_args+=(-it)
  fi

  docker run "${docker_args[@]}" \
    -e "ALPINE_ARCH=${arch}" \
    -e "CARCH=${arch}" \
    -e "ALPINE_REPO_NAME=${repo_name}" \
    -e "APKBUILD_DIR=${container_package_dir}" \
    -e "PACKAGER=${PACKAGER:-Joachim Schlöffel <me@joachim-schloeffel.com>}" \
    -v "${repo_root}:/work" \
    -v "${repo_root}/.cache/abuild:/home/builder/.abuild" \
    -v "${repo_root}/.cache/apk-distfiles:/var/cache/distfiles" \
    -v "${repo_root}/packages:/home/builder/packages" \
    "${image_name}" \
    "${subcommand}"
}

mkdir -p \
  "${repo_root}/.cache/abuild" \
  "${repo_root}/.cache/apk-distfiles" \
  "${repo_root}/packages"

case "${command_name}" in
  build-all)
    while IFS= read -r package_dir; do
      for arch in ${ALPINE_ARCHES:-x86_64 aarch64}; do
        if ! apk_package_supports_arch "${package_dir}" "${arch}"; then
          printf 'Skipping %s for unsupported arch %s\n' "$(apk_package_name "${package_dir}")" "${arch}"
          continue
        fi
        run_for_arch "${arch}" build "${package_dir}"
      done
    done < <(apk_package_dirs "${repo_root}")
    ;;
  build|checksum|lint)
    while IFS= read -r package_dir; do
      if [[ "${command_name}" == "build" ]] && ! apk_package_supports_arch "${package_dir}" "${requested_arch}"; then
        printf 'Skipping %s for unsupported arch %s\n' "$(apk_package_name "${package_dir}")" "${requested_arch}"
        continue
      fi
      run_for_arch "${requested_arch}" "${command_name}" "${package_dir}"
    done < <(apk_package_dirs "${repo_root}")
    ;;
  shell)
    package_count=0
    selected_package_dir=""
    while IFS= read -r package_dir; do
      package_count=$((package_count + 1))
      selected_package_dir="${package_dir}"
    done < <(apk_package_dirs "${repo_root}")

    if [[ "${package_count}" -ne 1 ]]; then
      printf 'apk:shell needs exactly one package; set ALPINE_PACKAGE=<name>\n' >&2
      exit 2
    fi
    run_for_arch "${requested_arch}" "${command_name}" "${selected_package_dir}"
    ;;
  *)
    printf 'unknown command: %s\n' "${command_name}" >&2
    printf 'usage: %s [build|build-all|checksum|lint|shell] [x86_64|aarch64]\n' "$0" >&2
    exit 2
    ;;
esac