#!/usr/bin/env bash
# SPDX-License-Identifier: Apache-2.0
# Download the public sample data that diffusion-planner-p150 may NOT redistribute (license unstated or non-commercial)
# into ${DIFFUSION_PLANNER_SAMPLES:-$HOME/.cache/tt_diffusion_planner/samples}, sha256-checked. Tests and benchmarks read it from there.
# Nothing downloaded here is executed; parse it with numpy / sqlite3 only.
set -euo pipefail
DEST="${DIFFUSION_PLANNER_SAMPLES:-$HOME/.cache/tt_diffusion_planner/samples}"
mkdir -p "$DEST"

fetch() {  # fetch <url> <sha256> <file name>
  local url="$1" sum="$2" name="$3"
  if [ -f "$DEST/$name" ] && echo "$sum  $DEST/$name" | sha256sum -c --status; then
    echo "ok      $name"; return
  fi
  curl -fL --retry 3 -o "$DEST/$name.part" "$url"
  echo "$sum  $DEST/$name.part" | sha256sum -c --status || { echo "sha256 mismatch: $name" >&2; exit 1; }
  mv "$DEST/$name.part" "$DEST/$name"; echo "fetched $name"
}

# Example (Autoware demo rosbag; pinned in autoware/ansible/roles/demo_artifacts/tasks/main.yaml:48-53):
# fetch https://autoware-files.s3.us-west-2.amazonaws.com/recordings/bags/demos/sample-rosbag.zip \
#       5f9d36353393b3d249212153c19049822b1298db56512aa045b4f7f6fc37cf88 sample-rosbag.zip
# nothing to fetch: both shipped samples are redistributable (Apache-2.0)
echo "samples in $DEST"