#!/usr/bin/env bash
# Only the four uniquely named Bonsai experiment artifacts are removed.
set -Eeuo pipefail
test "$(hostname)" = athena || exit 2
[[ ${1:-} == --all ]] || {
  printf '%s\n' 'Preview only. To remove: cleanup.sh --all' \
    'Container: mike-ai-bonsai2-ab' 'Image: mike-ai/bonsai2-ab:prism-b10685' \
    'Weights: /data/models/bonsai2-ab' 'Results: /data/benchmarks/bonsai2-ab' \
    'Staging: /opt/mike-ai/experiments/bonsai2-ab'; exit 0;
}
systemctl stop mike-ai-bonsai2-prepare.service 2>/dev/null || true
docker rm -f mike-ai-bonsai2-ab 2>/dev/null || true
docker image rm mike-ai/bonsai2-ab:prism-b10685 2>/dev/null || true
for description in \
  'PrismML-Eng/llama.cpp' \
  '/src/llama.cpp/build' \
  'ADD prism-cuda.tar.gz' \
  'useradd --system --uid 10013' \
  'nvidia/cuda:12.8.1-devel-ubuntu24.04'; do
  docker buildx prune --force --filter "description~=$description" >/dev/null 2>&1 || true
done
rm -rf -- /data/models/bonsai2-ab /data/benchmarks/bonsai2-ab
echo 'Bonsai experiment container, image, weights, results and staging removed.'
rm -rf -- /opt/mike-ai/experiments/bonsai2-ab