#!/usr/bin/env bash

DOCKER_BUILD_LIB_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

if ! declare -F run_logged >/dev/null 2>&1; then
  source "$DOCKER_BUILD_LIB_DIR/docker-e2e-logs.sh"
fi
if ! declare -F docker_e2e_timeout_cmd >/dev/null 2>&1; then
  source "$DOCKER_BUILD_LIB_DIR/docker-e2e-container.sh"
fi

docker_build_on_missing_enabled() {
  case "${OPENCLAW_DOCKER_BUILD_ON_MISSING:-}" in
    1 | true | TRUE | yes | YES)
      return 0
      ;;
    0 | false | FALSE | no | NO)
      return 1
      ;;
  esac

  [ "${OPENCLAW_TESTBOX:-0}" = "1" ]
}

docker_build_command() {
  local build_cmd=(docker build)
  if [ "${OPENCLAW_DOCKER_BUILD_USE_BUILDX:-0}" = "1" ] || docker_build_args_need_buildx "$@"; then
    build_cmd=(docker buildx build --load)
    if [ -n "${OPENCLAW_DOCKER_BUILD_CACHE_FROM:-}" ]; then
      build_cmd+=(--cache-from "${OPENCLAW_DOCKER_BUILD_CACHE_FROM}")
    fi
    if [ -n "${OPENCLAW_DOCKER_BUILD_CACHE_TO:-}" ]; then
      build_cmd+=(--cache-to "${OPENCLAW_DOCKER_BUILD_CACHE_TO}")
    fi
  fi

  printf '%s\0' env DOCKER_BUILDKIT=1 "${build_cmd[@]}" "$@"
}

docker_build_args_need_buildx() {
  for arg in "$@"; do
    case "$arg" in
      --build-context | --build-context=*)
        return 0
        ;;
    esac
  done
  return 1
}

docker_build_transient_failure() {
  local log_file="$1"
  grep -Eqi \
    'frontend grpc server closed unexpectedly|failed to dial gRPC|no active session|buildkit.*connection.*closed|rpc error: code = Unavailable' \
    "$log_file"
}

docker_build_retry_count() {
  local configured="${OPENCLAW_DOCKER_BUILD_RETRIES:-2}"
  if [[ "$configured" =~ ^[0-9]+$ ]]; then
    echo "$configured"
    return 0
  fi
  echo 2
}

docker_build_timeout_required() {
  case "${OPENCLAW_DOCKER_BUILD_REQUIRE_TIMEOUT:-0}" in
    1 | true | TRUE | yes | YES)
      return 0
      ;;
  esac
  return 1
}

docker_build_heartbeat_seconds() {
  local configured="${OPENCLAW_DOCKER_BUILD_HEARTBEAT_SECONDS:-30}"
  if [[ "$configured" =~ ^[0-9]+$ ]] && [ "$configured" -ge 1 ]; then
    echo "$configured"
    return
  fi
  echo 30
}

docker_build_run_command() {
  local timeout_value="$1"
  shift

  if docker_e2e_timeout_bin >/dev/null 2>&1 || docker_build_timeout_required; then
    docker_e2e_timeout_cmd "$timeout_value" "$@"
    return
  fi

  "$@"
}

docker_build_run_logged() {
  local label="$1"
  local timeout_value="$2"
  local log_file="$3"
  shift 3
  local heartbeat_seconds
  heartbeat_seconds="$(docker_build_heartbeat_seconds)"
  local started_at="$SECONDS"
  local next_heartbeat=$heartbeat_seconds
  local build_status=0

  docker_build_run_command "$timeout_value" "$@" >"$log_file" 2>&1 &
  local build_pid="$!"
  while kill -0 "$build_pid" 2>/dev/null; do
    /bin/sleep 1
    local elapsed_seconds=$((SECONDS - started_at))
    if [ "$elapsed_seconds" -ge "$next_heartbeat" ] && kill -0 "$build_pid" 2>/dev/null; then
      local log_bytes="0"
      if [ -f "$log_file" ]; then
        log_bytes="$(wc -c <"$log_file" 2>/dev/null || echo 0)"
        log_bytes="${log_bytes//[[:space:]]/}"
      fi
      echo "Docker build $label still running (${elapsed_seconds}s elapsed, ${log_bytes} log bytes captured)..."
      next_heartbeat=$((elapsed_seconds + heartbeat_seconds))
    fi
  done

  wait "$build_pid" || build_status="$?"
  return "$build_status"
}

docker_build_with_retries() {
  local label="$1"
  shift
  local retries
  retries="$(docker_build_retry_count)"
  local attempt=1
  local max_attempts=$((retries + 1))
  local log_file
  local command=()
  while IFS= read -r -d '' part; do
    command+=("$part")
  done < <(docker_build_command "$@")

  local timeout_value="${OPENCLAW_DOCKER_BUILD_TIMEOUT:-3600s}"
  while true; do
    log_file="$(docker_e2e_run_log "$label")"
    if docker_build_run_logged "$label" "$timeout_value" "$log_file" "${command[@]}"; then
      rm -f "$log_file"
      return 0
    fi

    if [ "$attempt" -ge "$max_attempts" ] || ! docker_build_transient_failure "$log_file"; then
      docker_e2e_print_log "$log_file"
      rm -f "$log_file"
      return 1
    fi

    echo "Docker build failed with a transient BuildKit transport error; retrying ($attempt/$retries)..." >&2
    docker_e2e_print_log "$log_file"
    rm -f "$log_file"
    attempt=$((attempt + 1))
    /bin/sleep "$attempt"
  done
}

docker_build_exec() {
  docker_build_with_retries docker-build "$@"
}

docker_build_run() {
  local label="$1"
  shift

  OPENCLAW_DOCKER_BUILD_REQUIRE_TIMEOUT="${OPENCLAW_DOCKER_BUILD_REQUIRE_TIMEOUT:-1}" \
    docker_build_with_retries "$label" "$@"
}