#!/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
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