#!/bin/bash

# 情绪博物馆管理脚本
# 提供服务管理、监控、备份等功能

set -e

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

log_info() {
    echo -e "${GREEN}[INFO]${NC} $1"
}

log_warn() {
    echo -e "${YELLOW}[WARN]${NC} $1"
}

log_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

log_step() {
    echo -e "${BLUE}[STEP]${NC} $1"
}

# 显示帮助信息
show_help() {
    echo "情绪博物馆管理脚本"
    echo ""
    echo "用法: $0 [命令] [选项]"
    echo ""
    echo "命令:"
    echo "  start         启动所有服务"
    echo "  stop          停止所有服务"
    echo "  restart       重启所有服务"
    echo "  status        查看服务状态"
    echo "  logs          查看服务日志"
    echo "  backup        备份数据"
    echo "  restore       恢复数据"
    echo "  update        更新服务"
    echo "  clean         清理资源"
    echo "  monitor       监控服务"
    echo "  health        健康检查"
    echo ""
    echo "选项:"
    echo "  -f, --follow  跟踪日志输出"
    echo "  -s, --service 指定服务名称"
    echo "  -h, --help    显示帮助信息"
    echo ""
    echo "示例:"
    echo "  $0 start                    # 启动所有服务"
    echo "  $0 logs -f                  # 跟踪所有服务日志"
    echo "  $0 logs -s gateway          # 查看网关服务日志"
    echo "  $0 restart -s ai-service    # 重启AI服务"
    echo ""
}

# 启动服务
start_services() {
    log_step "启动服务..."
    
    if [ -f "docker-compose.prod.yml" ]; then
        docker-compose -f docker-compose.prod.yml up -d
    else
        docker-compose up -d
    fi
    
    log_info "服务启动完成"
    sleep 5
    show_status
}

# 停止服务
stop_services() {
    log_step "停止服务..."
    
    if [ -f "docker-compose.prod.yml" ]; then
        docker-compose -f docker-compose.prod.yml down
    else
        docker-compose down
    fi
    
    log_info "服务停止完成"
}

# 重启服务
restart_services() {
    local service_name=${1:-}
    
    if [ -n "$service_name" ]; then
        log_step "重启服务: $service_name"
        docker-compose restart "$service_name"
    else
        log_step "重启所有服务..."
        stop_services
        sleep 3
        start_services
    fi
}

# 查看服务状态
show_status() {
    log_step "服务状态:"
    echo ""
    docker-compose ps
    echo ""
    
    # 显示资源使用情况
    log_step "资源使用情况:"
    docker stats --no-stream --format "table {{.Container}}\t{{.CPUPerc}}\t{{.MemUsage}}\t{{.NetIO}}\t{{.BlockIO}}"
}

# 查看日志
show_logs() {
    local follow_flag=""
    local service_name=""
    
    # 解析参数
    while [[ $# -gt 0 ]]; do
        case $1 in
            -f|--follow)
                follow_flag="-f"
                shift
                ;;
            -s|--service)
                service_name="$2"
                shift 2
                ;;
            *)
                service_name="$1"
                shift
                ;;
        esac
    done
    
    if [ -n "$service_name" ]; then
        log_info "查看服务日志: $service_name"
        docker-compose logs $follow_flag "$service_name"
    else
        log_info "查看所有服务日志"
        docker-compose logs $follow_flag
    fi
}

# 备份数据
backup_data() {
    local backup_dir="backups/$(date +%Y%m%d_%H%M%S)"
    
    log_step "开始数据备份..."
    mkdir -p "$backup_dir"
    
    # 备份MySQL数据
    log_info "备份MySQL数据..."
    docker-compose exec -T mysql mysqldump -u root -p123456 --all-databases > "$backup_dir/mysql_backup.sql"
    
    # 备份Redis数据
    log_info "备份Redis数据..."
    docker-compose exec -T redis redis-cli BGSAVE
    docker cp $(docker-compose ps -q redis):/data/dump.rdb "$backup_dir/redis_backup.rdb"
    
    # 备份配置文件
    log_info "备份配置文件..."
    cp -r deploy "$backup_dir/"
    cp docker-compose*.yml "$backup_dir/"
    cp .env "$backup_dir/" 2>/dev/null || true
    
    # 压缩备份
    tar -czf "$backup_dir.tar.gz" -C backups "$(basename $backup_dir)"
    rm -rf "$backup_dir"
    
    log_info "备份完成: $backup_dir.tar.gz"
}

# 恢复数据
restore_data() {
    local backup_file="$1"
    
    if [ -z "$backup_file" ]; then
        log_error "请指定备份文件"
        echo "用法: $0 restore <backup_file.tar.gz>"
        exit 1
    fi
    
    if [ ! -f "$backup_file" ]; then
        log_error "备份文件不存在: $backup_file"
        exit 1
    fi
    
    log_step "开始数据恢复..."
    log_warn "此操作将覆盖现有数据，请确认后继续"
    read -p "是否继续? (y/N): " -n 1 -r
    echo
    
    if [[ ! $REPLY =~ ^[Yy]$ ]]; then
        log_info "恢复操作已取消"
        exit 0
    fi
    
    # 解压备份文件
    local restore_dir="restore_$(date +%Y%m%d_%H%M%S)"
    mkdir -p "$restore_dir"
    tar -xzf "$backup_file" -C "$restore_dir"
    
    # 恢复MySQL数据
    log_info "恢复MySQL数据..."
    docker-compose exec -T mysql mysql -u root -p123456 < "$restore_dir"/*/mysql_backup.sql
    
    # 恢复Redis数据
    log_info "恢复Redis数据..."
    docker-compose stop redis
    docker cp "$restore_dir"/*/redis_backup.rdb $(docker-compose ps -q redis):/data/dump.rdb
    docker-compose start redis
    
    # 清理临时文件
    rm -rf "$restore_dir"
    
    log_info "数据恢复完成"
}

# 更新服务
update_services() {
    log_step "更新服务..."
    
    # 拉取最新代码
    if [ -d ".git" ]; then
        log_info "拉取最新代码..."
        git pull
    fi
    
    # 重新构建镜像
    log_info "重新构建镜像..."
    docker-compose build --no-cache
    
    # 重启服务
    log_info "重启服务..."
    restart_services
    
    log_info "服务更新完成"
}

# 清理资源
clean_resources() {
    log_step "清理Docker资源..."
    
    log_warn "此操作将清理未使用的Docker资源"
    read -p "是否继续? (y/N): " -n 1 -r
    echo
    
    if [[ $REPLY =~ ^[Yy]$ ]]; then
        # 清理未使用的镜像
        docker image prune -f
        
        # 清理未使用的容器
        docker container prune -f
        
        # 清理未使用的网络
        docker network prune -f
        
        # 清理未使用的卷（谨慎使用）
        # docker volume prune -f
        
        log_info "资源清理完成"
    else
        log_info "清理操作已取消"
    fi
}

# 监控服务
monitor_services() {
    log_step "服务监控面板"
    echo ""
    
    while true; do
        clear
        echo "=== 情绪博物馆服务监控 ==="
        echo "时间: $(date)"
        echo ""
        
        # 显示服务状态
        echo "📊 服务状态:"
        docker-compose ps
        echo ""
        
        # 显示资源使用
        echo "💻 资源使用:"
        docker stats --no-stream --format "table {{.Container}}\t{{.CPUPerc}}\t{{.MemUsage}}\t{{.NetIO}}"
        echo ""
        
        # 显示磁盘使用
        echo "💾 磁盘使用:"
        df -h | grep -E "(Filesystem|/dev/)"
        echo ""
        
        echo "按 Ctrl+C 退出监控"
        sleep 5
    done
}

# 健康检查
health_check() {
    log_step "执行健康检查..."
    
    local all_healthy=true
    
    # 检查MySQL
    if docker-compose exec -T mysql mysqladmin ping -h localhost -u root -p123456 &> /dev/null; then
        log_info "✅ MySQL服务正常"
    else
        log_error "❌ MySQL服务异常"
        all_healthy=false
    fi
    
    # 检查Redis
    if docker-compose exec -T redis redis-cli ping | grep -q PONG; then
        log_info "✅ Redis服务正常"
    else
        log_error "❌ Redis服务异常"
        all_healthy=false
    fi
    
    # 检查Nacos
    if curl -s http://localhost:8848/nacos/v1/ns/operator/metrics &> /dev/null; then
        log_info "✅ Nacos服务正常"
    else
        log_error "❌ Nacos服务异常"
        all_healthy=false
    fi
    
    # 检查网关
    if curl -s http://localhost:9000/actuator/health &> /dev/null; then
        log_info "✅ 网关服务正常"
    else
        log_error "❌ 网关服务异常"
        all_healthy=false
    fi
    
    # 检查AI服务
    if curl -s http://localhost:9002/actuator/health &> /dev/null; then
        log_info "✅ AI服务正常"
    else
        log_error "❌ AI服务异常"
        all_healthy=false
    fi
    
    # 检查前端
    if curl -s http://localhost:80/health &> /dev/null; then
        log_info "✅ 前端服务正常"
    else
        log_error "❌ 前端服务异常"
        all_healthy=false
    fi
    
    if $all_healthy; then
        log_info "🎉 所有服务健康检查通过"
    else
        log_warn "⚠️  部分服务存在问题，请检查日志"
    fi
}

# 主函数
main() {
    case "${1:-}" in
        "start")
            start_services
            ;;
        "stop")
            stop_services
            ;;
        "restart")
            shift
            restart_services "$@"
            ;;
        "status")
            show_status
            ;;
        "logs")
            shift
            show_logs "$@"
            ;;
        "backup")
            backup_data
            ;;
        "restore")
            restore_data "$2"
            ;;
        "update")
            update_services
            ;;
        "clean")
            clean_resources
            ;;
        "monitor")
            monitor_services
            ;;
        "health")
            health_check
            ;;
        "-h"|"--help"|"help")
            show_help
            ;;
        *)
            show_help
            ;;
    esac
}

main "$@"