#!/usr/bin/env sh
# Shrink the bundled ONNX Runtime native libraries down to a single Linux arch.
#
# The Maven `com.microsoft.onnxruntime:onnxruntime` jar ships native libs for every supported
# platform (macOS arm64, Windows x64, Linux x64/arm64 - ~42MB on 1.26.x; older 1.20.x also
# carried macOS x64 at ~88MB). A container image only ever loads one of them, so the rest is
# dead weight. Stripping to the target arch takes the jar down to ~8MB with no behaviour change
# (the model is still downloaded dynamically at runtime; only the unused native libs are removed).
#
# This is intentionally a Docker-build-only step: local/desktop builds keep every platform so
# cross-platform development still works.
#
# Usage: slim-onnxruntime.sh <target> [debian-arch]
#   <target> = a directory containing onnxruntime-*.jar (e.g. an extracted BOOT-INF/lib)
#            | a path to an onnxruntime-*.jar
#            | a Spring Boot fat jar that nests BOOT-INF/lib/onnxruntime-*.jar
#   [arch]   = amd64 | arm64 (defaults to the build host's dpkg arch)
#
# Safe no-op when onnxruntime is absent or `zip` is unavailable.
set -eu

target="${1:?usage: slim-onnxruntime.sh <target> [arch]}"
arch="${2:-$(dpkg --print-architecture 2>/dev/null || uname -m)}"
case "$arch" in
  amd64 | x86_64) keep="linux-x64" ;;
  arm64 | aarch64) keep="linux-aarch64" ;;
  *) keep="linux-x64" ;;
esac

if ! command -v zip >/dev/null 2>&1; then
  echo "[slim-onnxruntime] 'zip' not found - skipping (no slimming applied)"
  exit 0
fi

strip_ort_jar() { # $1 = path to an onnxruntime jar
  _jar="$1"
  _before=$(wc -c <"$_jar" 2>/dev/null || echo 0)
  for _p in osx-x64 osx-aarch64 win-x64 linux-x64 linux-aarch64; do
    [ "$_p" = "$keep" ] && continue
    zip -q -d "$_jar" "ai/onnxruntime/native/$_p/*" >/dev/null 2>&1 || true
  done
  _after=$(wc -c <"$_jar" 2>/dev/null || echo 0)
  echo "[slim-onnxruntime] kept '$keep'; $(basename "$_jar"): ${_before} -> ${_after} bytes"
}

if [ -d "$target" ]; then
  # Find the onnxruntime jar anywhere under the target dir so we are agnostic to the
  # Spring Boot layer layout (dependencies/lib vs dependencies/BOOT-INF/lib, etc.).
  found=$(find "$target" -name 'onnxruntime-*.jar' 2>/dev/null || true)
  if [ -n "$found" ]; then
    echo "$found" | while IFS= read -r ort; do
      [ -n "$ort" ] && strip_ort_jar "$ort"
    done
  else
    echo "[slim-onnxruntime] no onnxruntime jar under $target - nothing to do"
  fi
elif printf '%s' "$target" | grep -q 'onnxruntime-[^/]*\.jar$'; then
  strip_ort_jar "$target"
else
  # Spring Boot fat jar: extract the nested onnxruntime jar, slim it, put it back STORED
  # (Spring Boot requires nested jars to be uncompressed).
  if ! command -v unzip >/dev/null 2>&1; then
    echo "[slim-onnxruntime] 'unzip' needed for fat-jar mode - skipping"
    exit 0
  fi
  fat="$(cd "$(dirname "$target")" && pwd)/$(basename "$target")"
  nested=$(unzip -Z1 "$fat" 'BOOT-INF/lib/onnxruntime-*.jar' 2>/dev/null | head -n1 || true)
  if [ -z "$nested" ]; then
    echo "[slim-onnxruntime] no nested onnxruntime jar in $(basename "$fat") - nothing to do"
    exit 0
  fi
  tmp=$(mktemp -d)
  (cd "$tmp" && unzip -qo "$fat" "$nested")
  strip_ort_jar "$tmp/$nested"
  (cd "$tmp" && zip -q -0 -X "$fat" "$nested")
  rm -rf "$tmp"
  echo "[slim-onnxruntime] updated nested $nested inside $(basename "$fat")"
fi