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

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
PKG_ROOT="$ROOT_DIR"
ENV_FILE="$PKG_ROOT/compose/.env"
COMPOSE_FILE="$PKG_ROOT/compose/docker-compose.yml"

info(){ echo -e "\033[34m[UNINSTALL-GPU]\033[0m $*"; }

if [[ -f "$ENV_FILE" ]]; then
  info "stopping compose project"
  docker compose --env-file "$ENV_FILE" -f "$COMPOSE_FILE" down --remove-orphans || true
else
  info "compose/.env not found; attempting to remove container by name"
fi

# remove warmup container if still running
docker rm -f argus-net-warmup >/dev/null 2>&1 || true

# remove node container if present
docker rm -f argus-metric-gpu-node-swarm >/dev/null 2>&1 || true

info "uninstall completed"