#!/usr/bin/env bash

set -euo pipefail

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

if [[ -z "${GRADLE_USER_HOME:-}" ]]; then
  if [[ -d "$HOME/.gradle" && -w "$HOME/.gradle" ]]; then
    export GRADLE_USER_HOME="$HOME/.gradle"
  else
    export GRADLE_USER_HOME="$ROOT_DIR/.gradle-local"
  fi
fi

mkdir -p "$GRADLE_USER_HOME"

if [[ -d "$HOME/.local/jdk" ]]; then
  export JAVA_HOME="$HOME/.local/jdk"
fi

if [[ -d "$HOME/android-sdk" ]]; then
  export ANDROID_SDK_ROOT="$HOME/android-sdk"
fi

if [[ -n "${JAVA_HOME:-}" ]]; then
  export PATH="$JAVA_HOME/bin:$PATH"
fi

if [[ -n "${ANDROID_SDK_ROOT:-}" ]]; then
  export PATH="$ANDROID_SDK_ROOT/cmdline-tools/latest/bin:$ANDROID_SDK_ROOT/platform-tools:$PATH"
fi

if [[ -d "$HOME/.local/gradle/gradle-8.7/bin" ]]; then
  export PATH="$HOME/.local/gradle/gradle-8.7/bin:$PATH"
fi

unset HTTP_PROXY HTTPS_PROXY ALL_PROXY http_proxy https_proxy all_proxy

GRADLE_BIN="gradle"
if [[ -x "$ROOT_DIR/gradlew" ]]; then
  GRADLE_BIN="$ROOT_DIR/gradlew"
fi

ATTEMPTS="${ATTEMPTS:-5}"
SLEEP_SECONDS="${SLEEP_SECONDS:-15}"
TASK="${1:-assembleDebug}"
LOCAL_CHAQUOPY_REPO="${LOCAL_CHAQUOPY_REPO:-$ROOT_DIR/.m2-chaquopy}"
CHAQUOPY_MAVEN_BASE="${CHAQUOPY_MAVEN_BASE:-https://repo.maven.apache.org/maven2}"

prefetch_artifact() {
  local relative_path="$1"
  local destination="$LOCAL_CHAQUOPY_REPO/$relative_path"

  if [[ -f "$destination" ]]; then
    return 0
  fi

  mkdir -p "$(dirname "$destination")"
  echo "Prefetching $relative_path"
  curl \
    --fail \
    --location \
    --retry 8 \
    --retry-all-errors \
    --continue-at - \
    --connect-timeout 15 \
    --speed-limit 1024 \
    --speed-time 20 \
    --max-time 90 \
    --output "$destination" \
    "$CHAQUOPY_MAVEN_BASE/$relative_path"
}

prefetch_chaquopy_runtime() {
  local artifacts=(
    "com/chaquo/python/runtime/chaquopy_java/17.0.0/chaquopy_java-17.0.0.pom"
    "com/chaquo/python/runtime/chaquopy_java/17.0.0/chaquopy_java-17.0.0.jar"
    "com/chaquo/python/runtime/libchaquopy_java/17.0.0/libchaquopy_java-17.0.0.pom"
    "com/chaquo/python/runtime/libchaquopy_java/17.0.0/libchaquopy_java-17.0.0-3.12-arm64-v8a.so"
    "com/chaquo/python/runtime/libchaquopy_java/17.0.0/libchaquopy_java-17.0.0-3.12-x86_64.so"
    "com/chaquo/python/target/3.12.12-0/target-3.12.12-0.pom"
    "com/chaquo/python/target/3.12.12-0/target-3.12.12-0-arm64-v8a.zip"
    "com/chaquo/python/target/3.12.12-0/target-3.12.12-0-x86_64.zip"
  )

  for artifact in "${artifacts[@]}"; do
    prefetch_artifact "$artifact"
  done
}

prefetch_chaquopy_runtime

for attempt in $(seq 1 "$ATTEMPTS"); do
  echo "==> Android build attempt $attempt/$ATTEMPTS ($TASK)"
  if "$GRADLE_BIN" --no-daemon --console=plain "$TASK"; then
    exit 0
  fi

  if [[ "$attempt" -lt "$ATTEMPTS" ]]; then
    echo "Build failed, retrying in ${SLEEP_SECONDS}s..."
    sleep "$SLEEP_SECONDS"
  fi
done

echo "Android build failed after $ATTEMPTS attempts."
exit 1