#!/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"
}

# 配置变量
COMPOSE_FILE="docker-compose.custom.yml"
FRONTEND_DIR="/data/www/emotion-museum"
BACKEND_DIR="/data/builds"
LOG_DIR="/data/logs/emotion-museum"

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

# 启动服务
start_services() {
    log_step "启动服务..."
    
    # 按顺序启动服务
    log_info "启动基础服务..."
    docker-compose -f "$COMPOSE_FILE" up -d mysql redis nacos
    
    log_info "等待基础服务启动..."
    sleep 30
    
    log_info "启动应用服务..."
    docker-compose -f "$COMPOSE_FILE" up -d emotion-gateway emotion-ai emotion-user
    
    log_info "等待应用服务启动..."
    sleep 20
    
    log_info "启动Nginx..."
    docker-compose -f "$COMPOSE_FILE" up -d nginx
    
    log_info "服务启动完成"
    sleep 5
    show_status
}

# 停止服务
stop_services() {
    log_step "停止服务..."
    docker-compose -f "$COMPOSE_FILE" down
    log_info "服务停止完成"
}

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

# 查看服务状态
show_status() {
    log_step "服务状态:"
    echo ""
    docker-compose -f "$COMPOSE_FILE" ps
    echo ""
    
    # 显示资源使用情况
    log_step "资源使用情况:"
    docker stats --no-stream --format "table {{.Container}}\t{{.CPUPerc}}\t{{.MemUsage}}\t{{.NetIO}}\t{{.BlockIO}}"
    echo ""
    
    # 显示目录信息
    log_step "部署目录信息:"
    echo "前端文件: $FRONTEND_DIR ($(du -sh $FRONTEND_DIR 2>/dev/null | cut -f1 || echo 'N/A'))"
    echo "后端JAR:  $BACKEND_DIR ($(ls -la $BACKEND_DIR/*.jar 2>/dev/null | wc -l || echo '0') 个文件)"
    echo "日志目录: $LOG_DIR ($(du -sh $LOG_DIR 2>/dev/null | cut -f1 || echo 'N/A'))"
}

# 查看日志
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 -f "$COMPOSE_FILE" logs $follow_flag "$service_name"
    else
        log_info "查看所有服务日志"
        docker-compose -f "$COMPOSE_FILE" logs $follow_flag
    fi
}

# 健康检查
health_check() {
    log_step "执行健康检查..."
    
    local all_healthy=true
    
    # 检查MySQL
    if docker-compose -f "$COMPOSE_FILE" 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 -f "$COMPOSE_FILE" 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:9001/actuator/health &> /dev/null; then
        log_info "✅ 用户服务正常"
    else
        log_error "❌ 用户服务异常"
        all_healthy=false
    fi
    
    # 检查Nginx
    if curl -s http://localhost/nginx-health &> /dev/null; then
        log_info "✅ Nginx服务正常"
    else
        log_error "❌ Nginx服务异常"
        all_healthy=false
    fi
    
    # 检查前端文件
    if [ -f "$FRONTEND_DIR/index.html" ]; then
        log_info "✅ 前端文件正常"
    else
        log_error "❌ 前端文件异常"
        all_healthy=false
    fi
    
    # 检查后端JAR文件
    local jar_count=$(ls -1 $BACKEND_DIR/*.jar 2>/dev/null | wc -l)
    if [ "$jar_count" -eq 3 ]; then
        log_info "✅ 后端JAR文件正常 ($jar_count 个)"
    else
        log_error "❌ 后端JAR文件异常 (期望3个，实际$jar_count个)"
        all_healthy=false
    fi
    
    if $all_healthy; then
        log_info "🎉 所有服务健康检查通过"
    else
        log_warn "⚠️  部分服务存在问题，请检查日志"
    fi
}

# 更新服务
update_services() {
    log_step "更新服务..."
    
    # 重新构建前端
    if [ -d "web" ]; then
        log_info "重新构建前端..."
        cd web
        npm run build
        cd ..
        
        # 部署前端
        log_info "部署前端文件..."
        sudo rm -rf "$FRONTEND_DIR"/*
        sudo cp -r web/dist/* "$FRONTEND_DIR/"
        sudo chown -R www-data:www-data "$FRONTEND_DIR"
    fi
    
    # 重新构建后端
    if [ -d "backend" ]; then
        log_info "重新构建后端..."
        cd backend
        mvn clean package -DskipTests
        cd ..
        
        # 部署后端JAR
        log_info "部署后端JAR文件..."
        local services=("emotion-gateway" "emotion-ai" "emotion-user")
        for service in "${services[@]}"; do
            local source_jar="backend/${service}/target/${service}-1.0.0.jar"
            local target_jar="$BACKEND_DIR/${service}.jar"
            if [ -f "$source_jar" ]; then
                sudo cp "$source_jar" "$target_jar"
                sudo chown $USER:$USER "$target_jar"
            fi
        done
    fi
    
    # 重启服务
    log_info "重启服务..."
    restart_services
    
    log_info "服务更新完成"
}

# 备份数据
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 -f "$COMPOSE_FILE" exec -T mysql mysqldump -u root -p123456 --all-databases > "$backup_dir/mysql_backup.sql"
    
    # 备份Redis数据
    log_info "备份Redis数据..."
    docker-compose -f "$COMPOSE_FILE" exec -T redis redis-cli BGSAVE
    docker cp $(docker-compose -f "$COMPOSE_FILE" ps -q redis):/data/dump.rdb "$backup_dir/redis_backup.rdb"
    
    # 备份前端文件
    log_info "备份前端文件..."
    sudo tar -czf "$backup_dir/frontend_backup.tar.gz" -C "$FRONTEND_DIR" .
    
    # 备份后端JAR文件
    log_info "备份后端JAR文件..."
    sudo tar -czf "$backup_dir/backend_backup.tar.gz" -C "$BACKEND_DIR" .
    
    # 备份配置文件
    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"
}

# 清理资源
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
        
        log_info "资源清理完成"
    else
        log_info "清理操作已取消"
    fi
}

# 监控服务
monitor_services() {
    log_step "服务监控面板"
    echo ""
    
    while true; do
        clear
        echo "=== 情绪博物馆服务监控 ==="
        echo "时间: $(date)"
        echo ""
        
        # 显示服务状态
        echo "📊 服务状态:"
        docker-compose -f "$COMPOSE_FILE" ps
        echo ""
        
        # 显示资源使用
        echo "💻 资源使用:"
        docker stats --no-stream --format "table {{.Container}}\t{{.CPUPerc}}\t{{.MemUsage}}\t{{.NetIO}}"
        echo ""
        
        # 显示磁盘使用
        echo "💾 磁盘使用:"
        df -h | grep -E "(Filesystem|/data)"
        echo ""
        
        # 显示目录信息
        echo "📁 目录信息:"
        echo "前端: $(du -sh $FRONTEND_DIR 2>/dev/null | cut -f1 || echo 'N/A')"
        echo "后端: $(du -sh $BACKEND_DIR 2>/dev/null | cut -f1 || echo 'N/A')"
        echo "日志: $(du -sh $LOG_DIR 2>/dev/null | cut -f1 || echo 'N/A')"
        echo ""
        
        echo "按 Ctrl+C 退出监控"
        sleep 5
    done
}

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

main "$@"