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

MODE=${1:-preview}
MODEL_ROOT=${MODEL_ROOT:-/data/models/experiments/qwen38-ab-20260907}
RESULT_DIR=${RESULT_DIR:-/data/benchmarks/qwen38-ab-20260907}
NAME=mike-ai-llama-qwen38-ab

QWOPUS="$MODEL_ROOT/qwopus/Qwopus3.8-27B-Flash-MTP-IQ4_XS.gguf"
BARTOWSKI="$MODEL_ROOT/bartowski/Qwen3.8-27B-IQ4_XS.gguf"

case "$MODE" in
  preview)
    echo "Dry run only. Exact owned targets:"
    printf '  container: %s\n  results:   %s\n  model:     %s\n  model:     %s\n' \
      "$NAME" "$RESULT_DIR" "$QWOPUS" "$BARTOWSKI"
    echo "Use --results, --models or --all to delete these exact targets."
    exit 0
    ;;
  --results|--models|--all) ;;
  *)
    echo "Usage: $0 [--results|--models|--all]" >&2
    exit 2
    ;;
esac

docker rm -f "$NAME" >/dev/null 2>&1 || true

if [[ $MODE == --results || $MODE == --all ]]; then
  if [[ -d $RESULT_DIR ]]; then
    find "$RESULT_DIR" -maxdepth 1 -type f -name '*.json' -delete
    rmdir "$RESULT_DIR" 2>/dev/null || true
  fi
fi

if [[ $MODE == --models || $MODE == --all ]]; then
  rm -f -- "$QWOPUS" "$QWOPUS.part" "$BARTOWSKI" "$BARTOWSKI.part"
  rmdir "$MODEL_ROOT/qwopus" "$MODEL_ROOT/bartowski" 2>/dev/null || true
  rmdir "$MODEL_ROOT" 2>/dev/null || true
fi

echo "Cleanup complete for mode $MODE. No production model path was addressed."