#!/bin/bash

# ==============================================
# ac-compose Cleanup Script (project-scoped)
# ==============================================
# Usage: ./cleanup.sh [--soft] [--hard] [--nuclear] [--dry-run] [--force] [--preserve-backups]
# Project: ac-compose

set -e

# Resolve project dir and compose
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_DIR="${SCRIPT_DIR}"
DEFAULT_COMPOSE_FILE="${PROJECT_DIR}/docker-compose.yml"
ENV_FILE="${PROJECT_DIR}/.env"
source "${PROJECT_DIR}/scripts/lib/compose_overrides.sh"
declare -a COMPOSE_FILE_ARGS=()

# Colors
RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'; BLUE='\033[0;34m'; MAGENTA='\033[0;35m'; NC='\033[0m'

print_status() {
  case "$1" in
    INFO)    echo -e "${BLUE}ℹ️  ${2}${NC}";;
    SUCCESS) echo -e "${GREEN}✅ ${2}${NC}";;
    WARNING) echo -e "${YELLOW}⚠️  ${2}${NC}";;
    ERROR)   echo -e "${RED}❌ ${2}${NC}";;
    DANGER)  echo -e "${RED}💀 ${2}${NC}";;
    HEADER)  echo -e "\n${MAGENTA}=== ${2} ===${NC}";;
  esac
}

usage(){
  cat <<EOF
Usage: $0 [CLEANUP_LEVEL] [OPTIONS]

CLEANUP LEVELS:
  --soft             Stop project containers (preserves data)
  --hard             Remove containers + networks (preserves volumes/images)
  --nuclear          Complete removal: containers, networks, volumes, images (DESTROYS DATA)

OPTIONS:
  --dry-run          Show actions without executing
  --force            Skip confirmation prompts
  --preserve-backups Keep backups when nuking storage (moves them aside and restores)
  -h, --help         Show this help
EOF
}

# Flags
CLEANUP_LEVEL=""
DRY_RUN=false
FORCE=false
PRESERVE_BACKUPS=false

while [[ $# -gt 0 ]]; do
  case "$1" in
    --soft|--hard|--nuclear) CLEANUP_LEVEL="${1#--}"; shift;;
    --dry-run) DRY_RUN=true; shift;;
    --force) FORCE=true; shift;;
    --preserve-backups) PRESERVE_BACKUPS=true; shift;;
    -h|--help) usage; exit 0;;
    *) echo "Unknown arg: $1"; usage; exit 1;;
  esac
done

execute_command() {
  local description="$1"; shift
  local cmd="$*"
  if $DRY_RUN; then
    print_status INFO "[DRY RUN] $description"
    echo "  $cmd"
  else
    print_status INFO "$description"
    eval "$cmd" || print_status WARNING "Command failed or no action needed"
  fi
}

confirm() {
  local msg="$1"
  if $FORCE; then
    print_status INFO "Force enabled; skipping confirmation"
    return 0
  fi
  echo -e "${YELLOW}⚠️  ${msg}${NC}"
  read -p "Are you sure? (yes/no): " ans
  [[ "$ans" =~ ^(yes|y|YES|Y)$ ]] || { print_status INFO "Cancelled"; exit 0; }
}

format_container_table() {
  local containers
  containers="$($COMPOSE_BASE ps -a --format 'table {{.Name}}\t{{.Status}}\t{{.Service}}\t{{.Image}}' 2>/dev/null || echo "")"
  if [ -n "$containers" ]; then
    echo "$containers" | head -1
    echo "$containers" | tail -n +2 | while IFS=$'\t' read -r name status service image; do
      local short_image="${image##*/}"  # Remove registry prefix
      short_image="${short_image%:*}"   # Remove tag
      printf "%-20s %-25s %-15s %s\n" "$name" "$status" "$service" "$short_image"
    done
  else
    echo "No containers found"
  fi
}

show_resources() {
  print_status HEADER "CURRENT PROJECT RESOURCES"
  echo -e "${BLUE}🐳 Containers:${NC}"
  format_container_table
  echo ""

  local network_name
  network_name="$(grep -oE '^NETWORK_NAME=.+$' "$ENV_FILE" 2>/dev/null | cut -d= -f2 | tr -d '\r' || echo 'azerothcore')"
  echo -e "${BLUE}🌐 Networks:${NC}"
  docker network ls --format 'table {{.Name}}\t{{.Driver}}\t{{.Scope}}' | head -1
  docker network ls --format '{{.Name}}\t{{.Driver}}\t{{.Scope}}' | grep -E "(^|\s)${network_name}($|\s)" || echo "No project networks found"
  echo ""

  echo -e "${BLUE}💾 Volumes:${NC}"
  docker volume ls --format 'table {{.Name}}\t{{.Driver}}' | head -1
  docker volume ls --format '{{.Name}}\t{{.Driver}}' | grep -E 'ac_|acore|azerothcore' || echo "No project volumes found"
}

# Load env for STORAGE_PATH etc.
STORAGE_PATH_DEFAULT="${PROJECT_DIR}/storage"
STORAGE_PATH_LOCAL_DEFAULT="${PROJECT_DIR}/local-storage"
if [ -f "$ENV_FILE" ]; then
  set -a; source "$ENV_FILE"; set +a
fi

compose_overrides::build_compose_args "$PROJECT_DIR" "$ENV_FILE" "$DEFAULT_COMPOSE_FILE" COMPOSE_FILE_ARGS
COMPOSE_FILE_ARGS_STR=""
for arg in "${COMPOSE_FILE_ARGS[@]}"; do
  COMPOSE_FILE_ARGS_STR+=" ${arg}"
done
COMPOSE_BASE="docker compose${COMPOSE_FILE_ARGS_STR}"
STORAGE_PATH="${STORAGE_PATH:-$STORAGE_PATH_DEFAULT}"
STORAGE_PATH_LOCAL="${STORAGE_PATH_LOCAL:-$STORAGE_PATH_LOCAL_DEFAULT}"
PROJECT_NAME="${COMPOSE_PROJECT_NAME:-ac-compose}"

sanitize_project_name(){
  local raw="$1"
  local sanitized
  sanitized="$(echo "$raw" | tr '[:upper:]' '[:lower:]')"
  sanitized="${sanitized// /-}"
  sanitized="$(echo "$sanitized" | tr -cd 'a-z0-9_-')"
  if [[ -z "$sanitized" ]]; then
    sanitized="acore-compose"
  elif [[ ! "$sanitized" =~ ^[a-z0-9] ]]; then
    sanitized="ac${sanitized}"
  fi
  echo "$sanitized"
}

PROJECT_IMAGE_PREFIX="$(sanitize_project_name "${COMPOSE_PROJECT_NAME:-acore-compose}")"

remove_storage_dir(){
  local path="$1"
  if [ -d "$path" ]; then
    rm -rf "$path" 2>/dev/null || sudo rm -rf "$path" 2>/dev/null || true
  fi
}

remove_project_volumes(){
  docker volume ls --format '{{.Name}}' \
    | grep -E "^${PROJECT_NAME}|^azerothcore" \
    | xargs -r docker volume rm >/dev/null 2>&1 || true
}

soft_cleanup() {
  print_status HEADER "SOFT CLEANUP - Stop runtime stack"
  confirm "This will stop all project containers (data preserved)."
  local profiles=(
    --profile services-standard
    --profile services-playerbots
    --profile services-modules
    --profile client-data
    --profile client-data-bots
    --profile modules
    --profile tools
    --profile db
  )
  execute_command "Stopping runtime profiles" $COMPOSE_BASE "${profiles[@]}" down
  print_status SUCCESS "Soft cleanup complete"
}

hard_cleanup() {
  print_status HEADER "HARD CLEANUP - Remove containers + networks"
  confirm "This will remove containers and networks (volumes/images preserved)."
  local profiles=(
    --profile services-standard
    --profile services-playerbots
    --profile services-modules
    --profile client-data
    --profile client-data-bots
    --profile modules
    --profile tools
    --profile db
  )
  execute_command "Removing containers and networks" $COMPOSE_BASE "${profiles[@]}" down --remove-orphans
  execute_command "Remove project volumes" remove_project_volumes
  # Remove straggler containers matching project name (defensive)
  execute_command "Remove stray project containers" "docker ps -a --format '{{.Names}}' | grep -E '^ac-' | xargs -r docker rm -f"
  # Remove project network if present and not automatically removed
  if [ -n "${NETWORK_NAME:-}" ]; then
    execute_command "Remove project network ${NETWORK_NAME}" "docker network rm ${NETWORK_NAME} 2>/dev/null || true"
  fi
  print_status SUCCESS "Hard cleanup complete"
}

nuclear_cleanup() {
  print_status HEADER "NUCLEAR CLEANUP - COMPLETE REMOVAL"
  print_status DANGER "THIS WILL DESTROY ALL PROJECT DATA"
  confirm "Proceed with complete removal?"

  # Down with volumes
  local profiles=(
    --profile services-standard
    --profile services-playerbots
    --profile services-modules
    --profile client-data
    --profile client-data-bots
    --profile modules
    --profile tools
    --profile db
  )
  execute_command "Removing containers, networks and volumes" $COMPOSE_BASE "${profiles[@]}" down --volumes --remove-orphans
  execute_command "Remove leftover volumes" remove_project_volumes

  # Remove project images (server/tool images typical to this project)
  execute_command "Remove acore images" "docker images --format '{{.Repository}}:{{.Tag}}' | grep -E '^acore/' | xargs -r docker rmi"
  execute_command "Remove local project images" "docker images --format '{{.Repository}}:{{.Tag}}' | grep -E '^${PROJECT_IMAGE_PREFIX}:' | xargs -r docker rmi"
  execute_command "Remove legacy playerbots images" "docker images --format '{{.Repository}}:{{.Tag}}' | grep -E '^uprightbass360/azerothcore-wotlk-playerbots' | xargs -r docker rmi"
  execute_command "Remove tool images" "docker images --format '{{.Repository}}:{{.Tag}}' | grep -E 'phpmyadmin|uprightbass360/keira3' | xargs -r docker rmi"

  # Storage cleanup (preserve backups if requested)
  if $PRESERVE_BACKUPS; then
    print_status INFO "Preserving backups under ${STORAGE_PATH}/backups"
    TMP_PRESERVE="${PROJECT_DIR}/.preserve-backups"
    if [ -d "${STORAGE_PATH}/backups" ]; then
      execute_command "Staging backups" "mkdir -p '${TMP_PRESERVE}' && cp -a '${STORAGE_PATH}/backups' '${TMP_PRESERVE}/'"
    fi
    execute_command "Removing main storage" "remove_storage_dir '${STORAGE_PATH}'"
    execute_command "Removing local storage" "remove_storage_dir '${STORAGE_PATH_LOCAL}'"
    if [ -d "${TMP_PRESERVE}/backups" ]; then
      execute_command "Restoring backups" "mkdir -p '${STORAGE_PATH}' && mv '${TMP_PRESERVE}/backups' '${STORAGE_PATH}/backups' && rm -rf '${TMP_PRESERVE}'"
      print_status SUCCESS "Backups preserved at ${STORAGE_PATH}/backups"
    fi
  else
    execute_command "Removing main storage and backups" "remove_storage_dir '${STORAGE_PATH}'; remove_storage_dir '${PROJECT_DIR}/backups'"
    execute_command "Removing local storage" "remove_storage_dir '${STORAGE_PATH_LOCAL}'"
  fi

  # Optional system prune for project context
  execute_command "Docker system prune (dangling)" "docker system prune -af --volumes"
  print_status SUCCESS "Nuclear cleanup completed"
}

show_summary() {
  local lvl="$1"
  print_status HEADER "CLEANUP SUMMARY"
  case "$lvl" in
    soft)
      echo -e "${GREEN}✅ Containers: Stopped${NC}"; echo -e "${BLUE}ℹ️  Networks/Volumes/Images: Preserved${NC}";;
    hard)
      echo -e "${GREEN}✅ Containers/Networks: Removed${NC}"; echo -e "${BLUE}ℹ️  Volumes/Images: Preserved${NC}";;
    nuclear)
      echo -e "${RED}💀 Containers/Networks/Volumes/Images: DESTROYED${NC}";;
  esac
}

main(){
  print_status HEADER "ac-compose CLEANUP"

  if ! command -v docker >/dev/null 2>&1; then
    print_status ERROR "Docker not found"
    exit 1
  fi
  if [ ! -f "$DEFAULT_COMPOSE_FILE" ]; then
    print_status ERROR "Compose file not found at $DEFAULT_COMPOSE_FILE"
    exit 1
  fi
  if [ -z "$CLEANUP_LEVEL" ]; then
    usage; exit 1
  fi

  show_resources

  case "$CLEANUP_LEVEL" in
    soft)   soft_cleanup;;
    hard)   hard_cleanup;;
    nuclear) nuclear_cleanup;;
    *) usage; exit 1;;
  esac

  show_summary "$CLEANUP_LEVEL"
  print_status SUCCESS "🧹 Cleanup completed"
}

main "$@"