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

if [[ -z "${REGISTRY_HOST:-}" ]]; then
  echo "--- REGISTRY_HOST is not set; skipping Jetson engine artifact pull"
  exit 0
fi

ENGINE_REPOSITORY="${JETSON_ENGINE_REPOSITORY:-$REGISTRY_HOST/azaion/detections-jetson-engine}"
BRANCH="${CI_COMMIT_BRANCH:-local}"
default_tag="$(printf '%s' "$BRANCH" | sed 's#^refs/heads/##; s#^refs/tags/##; s#[^A-Za-z0-9_.-]#-#g; s#^-*##; s#-*$##')"
default_tag="${default_tag:-local}"
ENGINE_TAG="${JETSON_ENGINE_TAG:-$default_tag}"
TARGET_DIR="${JETSON_ENGINE_TARGET_DIR:-fixtures/models}"
image="$ENGINE_REPOSITORY:$ENGINE_TAG"

echo "--- Pulling Jetson engine artifact image: $image"
if ! docker pull "$image"; then
  echo "--- Jetson engine artifact image not found; smoke will use ONNX fallback"
  exit 0
fi

cid="$(docker create "$image")"
trap 'docker rm -f "$cid" >/dev/null 2>&1 || true' EXIT

mkdir -p "$TARGET_DIR"
docker cp "$cid:/models/." "$TARGET_DIR/"

echo "--- Installed Jetson engine files:"
find "$TARGET_DIR" -maxdepth 1 -type f -name 'azaion*.engine' -print -exec ls -lh {} \;