#!/bin/bash
# OAuth-DB CLI E2E 测试环境清理脚本
# 用途：停止并清理 Docker Compose E2E 环境

set -e

# 颜色定义
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

# 打印函数
print_info() {
    echo -e "${BLUE}ℹ${NC} $1"
}

print_success() {
    echo -e "${GREEN}✓${NC} $1"
}

print_warning() {
    echo -e "${YELLOW}⚠${NC} $1"
}

print_error() {
    echo -e "${RED}✗${NC} $1"
}

# 获取脚本所在目录
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
CLI_DIR="$(dirname "$SCRIPT_DIR")"

# 检查 docker-compose.debug.yml 是否存在
if [ ! -f "$CLI_DIR/docker-compose.debug.yml" ]; then
    print_error "docker-compose.debug.yml 不存在"
    exit 1
fi

print_info "停止 E2E 测试环境..."
cd "$CLI_DIR"

# 停止并删除容器、网络、卷
docker compose -f docker-compose.debug.yml down -v

print_success "E2E 测试环境已清理"

# 可选：清理编译缓存（节省磁盘空间）
if [ "$1" = "--clean-cache" ]; then
    print_info "清理编译缓存..."
    docker volume rm oauth-db-cli-target-cache 2>/dev/null || true
    docker volume rm oauth-db-cli-test-config 2>/dev/null || true
    print_success "编译缓存已清理"
fi

echo ""
print_info "重新启动环境："
print_info "  cd $CLI_DIR"
print_info "  ./tests/e2e_setup.sh"