#!/bin/bash

echo "开始卸载 Docker..."

# 检测系统类型
if [ -x "$(command -v yum)" ]; then
    echo "检测到 CentOS/RHEL 系统"
    PKG_MGR="yum"
elif [ -x "$(command -v apt-get)" ]; then
    echo "检测到 Debian/Ubuntu 系统"
    PKG_MGR="apt-get"
else
    echo "不支持的包管理器"
    exit 1
fi

# 停止 Docker 服务
echo "停止 Docker 相关服务..."
sudo systemctl stop docker docker.socket containerd 2>/dev/null || true
sudo pkill -f docker 2>/dev/null || true

# 卸载 Docker 包
echo "卸载 Docker 软件包..."
if [ "$PKG_MGR" = "yum" ]; then
    sudo yum remove -y docker \
        docker-client \
        docker-client-latest \
        docker-common \
        docker-latest \
        docker-latest-logrotate \
        docker-logrotate \
        docker-engine \
        containerd \
        runc \
        docker-ce \
        docker-ce-cli \
        docker-ce-rootless-extras
else
    sudo apt-get purge -y docker-ce \
        docker-ce-cli \
        docker-ce-rootless-extras \
        containerd.io \
        docker-buildx-plugin \
        docker-compose-plugin
fi

# 删除所有 Docker 数据和配置
echo "删除 Docker 数据和配置..."
sudo rm -rf /var/lib/docker
sudo rm -rf /var/lib/containerd
sudo rm -rf /etc/docker
sudo rm -rf /var/run/docker.sock
sudo rm -rf /var/run/containerd
sudo rm -rf /etc/apparmor.d/docker
sudo rm -rf /usr/bin/docker*
sudo rm -rf /usr/libexec/docker
sudo rm -rf /usr/lib/docker*

# 删除 Docker 用户组（如果存在且没有其他用户）
echo "清理 Docker 用户组..."
if getent group docker >/dev/null; then
    echo "警告: 将删除 docker 用户组"
    sudo groupdel docker
fi

# 清理系统
echo "执行系统清理..."
if [ "$PKG_MGR" = "yum" ]; then
    sudo yum autoremove -y
else
    sudo apt-get autoremove -y
    sudo apt-get autoclean -y
fi

# 验证卸载
echo "验证卸载结果..."
if command -v docker >/dev/null 2>&1; then
    echo "❌ 卸载不彻底：仍检测到 docker 命令"
    exit 1
else
    echo "✅ Docker 已完全卸载"
    
    # 检查残留文件
    echo "检查残留文件..."
    RESIDUAL_FILES=$(find / -name "*docker*" -type f 2>/dev/null | head -10)
    if [ -n "$RESIDUAL_FILES" ]; then
        echo "发现以下残留文件："
        echo "$RESIDUAL_FILES"
        echo "建议手动检查这些文件是否需要保留"
    fi
fi

echo "Docker 卸载完成！"}

# 启动服务器
start_server() {
    if is_server_running; then
        print_color "服务器已在运行 (PID: $(cat "$PID_FILE"))" "$YELLOW"
        return
    fi
    
    print_color "启动服务器端口: $SERVER_PORT" "$GREEN"
    echo $$ > "$PID_FILE"
    
    # 主服务器循环
    while true; do
        log "等待客户端连接..."
        nc -l -p $SERVER_PORT -c "
            client_ip=\$(echo \$SSH_CLIENT | awk '{print \$1}')
            [[ -z \"\$client_ip\" ]] && client_ip=\"unknown\"
            
            read -r data
            timestamp=\$(date '+%Y-%m-%d %H:%M:%S')
            
            if echo \"\$data\" | grep -q \"HEARTBEAT|\"; then
                IFS=\"|\" read -r heartbeat serial hostname system <<< \"\$data\"
                
                # 更新客户端信息
                grep -v \"^\$serial|\" \"$CLIENTS_FILE\" > /tmp/clients.tmp 2>/dev/null
                echo \"\$serial|\$client_ip|\$hostname|\$system|\$timestamp\" >> /tmp/clients.tmp
                mv /tmp/clients.tmp \"$CLIENTS_FILE\" 2>/dev/null
                
                echo \"ACK:OK\"
                log \"客户端 \$serial 已连接 - \$hostname (\$client_ip)\"
            else
                echo \"ACK:UNKNOWN\"
            fi
        " 2>/dev/null
        sleep 1
    done
}

# 停止服务器
stop_server() {
    if is_server_running; then
        local pid=$(cat "$PID_FILE")
        print_color "停止服务器 (PID: $pid)" "$GREEN"
        kill "$pid"
        rm -f "$PID_FILE"
    else
        print_color "服务器未运行" "$YELLOW"
    fi
}

# 显示客户端列表
show_clients() {
    if [[ ! -s "$CLIENTS_FILE" ]]; then
        print_color "没有客户端连接" "$YELLOW"
        return
    fi
    
    local count=$(wc -l < "$CLIENTS_FILE" 2>/dev/null || echo "0")
    print_color "已连接客户端 ($count):" "$BLUE"
    
    print_color "┌────────────┬───────────────┬──────────────────┬─────────────────┬─────────────────────┐" "$CYAN"
    print_color "│   序列号   │    IP地址     │      主机名      │      系统       │     最后在线       │" "$CYAN"
    print_color "├────────────┼───────────────┼──────────────────┼─────────────────┼─────────────────────┤" "$CYAN"
    
    while IFS='|' read -r serial ip hostname system last_seen; do
        if [[ -n "$serial" ]]; then
            printf "│ ${GREEN}%-10s${NC} │ ${YELLOW}%-13s${NC} │ ${BLUE}%-16s${NC} │ ${PURPLE}%-15s${NC} │ ${GREEN}%-19s${NC} │\n" \
                   "$serial" "$ip" "$hostname" "$system" "$last_seen"
        fi
    done < "$CLIENTS_FILE"
    
    print_color "└────────────┴───────────────┴──────────────────┴─────────────────┴─────────────────────┘" "$CYAN"
}

# 发送命令到客户端
send_command() {
    local serial="$1"
    local command="$2"
    
    [[ -z "$serial" || -z "$command" ]] && {
        print_color "用法: send <序列号> <命令>" "$RED"
        return
    }
    
    local client_info=$(grep "^$serial|" "$CLIENTS_FILE")
    [[ -z "$client_info" ]] && {
        print_color "客户端未找到: $serial" "$RED"
        return
    }
    
    IFS='|' read -r serial ip hostname system last_seen <<< "$client_info"
    
    print_color "发送命令到 $serial ($hostname)..." "$GREEN"
    print_color "命令: $command" "$YELLOW"
    
    # 发送命令
    echo "COMMAND:$command" | timeout 5 nc -w 3 "$ip" 5556
    
    [[ $? -eq 0 ]] && {
        log "命令发送成功: $serial -> $command"
        print_color "命令发送成功!" "$GREEN"
    } || {
        log "命令发送失败: $serial -> $command"
        print_color "命令发送失败!" "$RED"
    }
}

# 执行脚本
execute_script() {
    local serial="$1"
    local script_name="$2"
    
    [[ -z "$serial" || -z "$script_name" ]] && {
        print_color "用法: script <序列号> <脚本名>" "$RED"
        show_scripts
        return
    }
    
    local script_path="$SCRIPT_DIR/$script_name"
    [[ ! -f "$script_path" ]] && {
        print_color "脚本不存在: $script_name" "$RED"
        show_scripts
        return
    }
    
    local client_info=$(grep "^$serial|" "$CLIENTS_FILE")
    [[ -z "$client_info" ]] && {
        print_color "客户端未找到: $serial" "$RED"
        return
    }
    
    IFS='|' read -r serial ip hostname system last_seen <<< "$client_info"
    
    print_color "执行脚本到 $serial ($hostname)..." "$GREEN"
    print_color "脚本: $script_name" "$YELLOW"
    
    echo "SCRIPT:$script_name" | timeout 5 nc -w 3 "$ip" 5556
    
    [[ $? -eq 0 ]] && {
        log "脚本执行成功: $serial -> $script_name"
        print_color "脚本执行成功!" "$GREEN"
    } || {
        log "脚本执行失败: $serial -> $script_name"
        print_color "脚本执行失败!" "$RED"
    }
}

# 显示可用脚本
show_scripts() {
    print_color "可用脚本:" "$BLUE"
    ls "$SCRIPT_DIR"/*.sh 2>/dev/null | xargs -n 1 basename | while read script; do
        print_color "  📜 $script" "$GREEN"
    done || print_color "  无可用脚本" "$YELLOW"
}

# 广播命令
broadcast_command() {
    local command="$1"
    [[ -z "$command" ]] && {
        print_color "用法: broadcast <命令>" "$RED"
        return
    }
    
    local count=0
    print_color "广播命令: $command" "$YELLOW"
    
    while IFS='|' read -r serial ip hostname system last_seen; do
        [[ -n "$serial" ]] && {
            echo "发送到 $serial..."
            echo "COMMAND:$command" | timeout 3 nc -w 2 "$ip" 5556 &
            count=$((count + 1))
        }
    done < "$CLIENTS_FILE"
    
    wait
    print_color "广播完成! 发送到 $count 个客户端" "$GREEN"
}

# 显示服务器状态
show_status() {
    print_color "=== 服务器状态 ===" "$BLUE"
    print_color "端口: $SERVER_PORT" "$GREEN"
    print_color "运行状态: $(is_server_running && echo '运行中' || echo '未运行')" "$YELLOW"
    print_color "客户端数: $(wc -l < "$CLIENTS_FILE" 2>/dev/null || echo "0")" "$CYAN"
    print_color "日志文件: $LOG_FILE" "$PURPLE"
}

# 主菜单
main_menu() {
    while true; do
        clear
        print_color "╔══════════════════════════════════════════════════════════════╗" "$CYAN"
        print_color "║                   服务器控制器 v3.0                         ║" "$BLUE"
        print_color "║                 Server Controller v3.0                      ║" "$BLUE"
        print_color "╠══════════════════════════════════════════════════════════════╣" "$CYAN"
        print_color "║ 端口: $SERVER_PORT | 客户端: $(wc -l < "$CLIENTS_FILE" 2>/dev/null || echo "0") | 状态: $(is_server_running && echo -e "${GREEN}运行中${NC}" || echo -e "${RED}未运行${NC}") ║" "$GREEN"
        print_color "╚══════════════════════════════════════════════════════════════╝" "$CYAN"
        echo
        
        print_color "1. 启动服务器" "$GREEN"
        print_color "2. 停止服务器" "$RED"
        print_color "3. 客户端列表" "$CYAN"
        print_color "4. 发送命令" "$YELLOW"
        print_color "5. 执行脚本" "$BLUE"
        print_color "6. 广播命令" "$PURPLE"
        print_color "7. 显示脚本" "$CYAN"
        print_color "8. 服务器状态" "$GREEN"
        print_color "9. 查看日志" "$YELLOW"
        print_color "0. 退出" "$RED"
        echo
        
        read -p "请选择操作 [0-9]: " choice
        
        case $choice in
            1) start_server ;;
            2) stop_server ;;
            3) show_clients ;;
            4)
                read -p "序列号: " serial
                read -p "命令: " cmd
                send_command "$serial" "$cmd"
                ;;
            5)
                read -p "序列号: " serial
                read -p "脚本名: " script
                execute_script "$serial" "$script"
                ;;
            6)
                read -p "广播命令: " cmd
                broadcast_command "$cmd"
                ;;
            7) show_scripts ;;
            8) show_status ;;
            9) tail -f "$LOG_FILE" ;;
            0)
                print_color "再见!" "$GREEN"
                exit 0
                ;;
            *) print_color "无效选择!" "$RED" ;;
        esac
        
        echo
        read -p "按回车继续..."
    done
}

# 命令行处理
case "${1:-}" in
    start)
        init_server
        start_server
        ;;
    stop)
        stop_server
        ;;
    status)
        init_server
        show_status
        ;;
    *)
        init_server
        main_menu
        ;;
esac