#!/bin/bash
# 统一服务器连接脚本模板 - 智能快速连接 + 完整环境配置
# 使用说明：复制此文件并修改配置信息后使用

SESSION_NAME="dev-session-template"
CONNECTION_TOOL="remote-tool"

# 跳板机配置（可选，如果为空则直接连接目标服务器）
JUMP_HOST="your_jump_host"                     # 跳板机地址，设为空字符串则跳过
JUMP_PASSWORD="your_password_here"             # 跳板机密码（自动输入）

# 目标服务器配置
SERVER_HOST="your_server_host"                 # 最终目标服务器
DOCKER_CONTAINER="your_container"               # Docker容器名称
DOCKER_IMAGE="your_docker_image"                # Docker镜像名称

# 环境配置选项
AUTO_SETUP_ENVIRONMENT=true                    # 是否自动配置环境（首次）
QUICK_CONNECT_MODE=true                         # 快速连接模式（跳过已配置的环境设置）

# BOS配置信息（仅首次环境配置时使用）
BOS_ACCESS_KEY="your_bos_access_key"
BOS_SECRET_KEY="your_bos_secret_key"
BOS_BUCKET="your_bos_bucket_path"

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

# 智能检查tmux输出函数
check_tmux_output() {
    local session_name=$1
    local expected_pattern=$2
    local timeout=${3:-10}
    local check_interval=1
    local elapsed=0
    
    while [ $elapsed -lt $timeout ]; do
        local output=$(tmux capture-pane -t $session_name -p | grep -v '^$' | tail -10)
        if [[ "$output" =~ $expected_pattern ]]; then
            return 0
        fi
        sleep $check_interval
        elapsed=$((elapsed + check_interval))
    done
    return 1
}

# 智能容器检查和处理函数（修复版 - 解决分支执行bug）
smart_container_setup() {
    local session_name=$1
    local container_name=$2
    local image_name=$3
    
    echo -e "${BLUE}🔍 智能检查Docker环境...${NC}"
    
    # 步骤1：精确检查容器是否存在
    echo -e "${BLUE}🔍 步骤1: 精确检查容器是否存在: $container_name${NC}"
    tmux send-keys -t $session_name "echo '=== CONTAINER_EXIST_CHECK_START ==='" Enter
    tmux send-keys -t $session_name "docker inspect $container_name >/dev/null 2>&1 && echo 'CONTAINER_EXISTS_YES' || echo 'CONTAINER_EXISTS_NO'" Enter
    sleep 3
    
    # 改进的输出捕获 - 只获取最后一行结果
    echo -e "${YELLOW}正在捕获检查结果...${NC}"
    tmux send-keys -t $session_name "echo '=== CAPTURE_POINT ==='" Enter
    sleep 1
    
    # 更精确的输出捕获 - 获取两个标记之间的内容
    local exist_output=$(tmux capture-pane -t $session_name -p | sed -n '/=== CONTAINER_EXIST_CHECK_START ===/,/=== CAPTURE_POINT ===/p' | grep -E "(CONTAINER_EXISTS_YES|CONTAINER_EXISTS_NO)" | tail -1)
    
    echo -e "${YELLOW}调试信息 - 容器存在性检查:${NC}"
    echo "检查结果: [$exist_output]"
    
    # 更严格的条件判断
    if [[ "$exist_output" == *"CONTAINER_EXISTS_YES"* ]]; then
        echo -e "${GREEN}✅ 容器已存在，进入快速连接模式...${NC}"
        handle_existing_container $session_name $container_name
        return $?
        
    elif [[ "$exist_output" == *"CONTAINER_EXISTS_NO"* ]]; then
        echo -e "${BLUE}🚀 容器不存在，进入创建模式...${NC}"
        handle_new_container $session_name $container_name $image_name
        return $?
        
    else
        echo -e "${RED}❌ 容器存在性检查结果异常${NC}"
        echo -e "${YELLOW}调试信息 - 原始输出:${NC}"
        tmux capture-pane -t $session_name -p | tail -10
        return 1
    fi
}
# 处理已存在容器的逻辑
handle_existing_container() {
    local session_name=$1
    local container_name=$2
    
    echo -e "${BLUE}🔍 步骤2: 检查容器运行状态...${NC}"
    tmux send-keys -t $session_name "echo '=== CONTAINER_STATUS_CHECK_START ==='" Enter
    tmux send-keys -t $session_name "docker inspect --format='{{.State.Running}}' $container_name" Enter
    sleep 2
    
    local status_output=$(tmux capture-pane -t $session_name -p | grep -A 3 "=== CONTAINER_STATUS_CHECK_START ===" | tail -2)
    echo -e "${YELLOW}调试信息 - 容器运行状态:${NC}"
    echo "$status_output"
    
    if echo "$status_output" | grep -q "true"; then
        echo -e "${GREEN}✅ 容器正在运行${NC}"
    else
        echo -e "${YELLOW}⚠️  容器已停止，正在启动...${NC}"
        tmux send-keys -t $session_name "echo '=== CONTAINER_START_BEGIN ==='" Enter
        tmux send-keys -t $session_name "docker start $container_name" Enter
        sleep 5
        
        # 验证启动是否成功
        local start_result=$(tmux capture-pane -t $session_name -p | grep -A 3 "=== CONTAINER_START_BEGIN ===" | tail -2)
        if echo "$start_result" | grep -q "$container_name"; then
            echo -e "${GREEN}✅ 容器启动成功${NC}"
        else
            echo -e "${RED}❌ 容器启动失败${NC}"
            echo -e "${YELLOW}错误信息: $start_result${NC}"
            return 1
        fi
    fi
    
    # 进入容器
    echo -e "${GREEN}🚪 步骤3: 进入现有容器...${NC}"
    tmux send-keys -t $session_name "docker exec -it $container_name zsh" Enter
    sleep 2
    
    # 检查是否成功进入zsh
    if ! check_tmux_output $session_name ".*@.*:.*#" 5; then
        echo -e "${YELLOW}⚠️  尝试启动zsh...${NC}"
        tmux send-keys -t $session_name "zsh" Enter
        sleep 2
    fi
    
    echo -e "${GREEN}✅ 快速连接完成！${NC}"
    return 0
}
# 处理新容器创建的逻辑
handle_new_container() {
    local session_name=$1
    local container_name=$2
    local image_name=$3
    
    echo -e "${BLUE}步骤1: 创建Docker容器 $container_name${NC}"
    tmux send-keys -t $session_name "echo '=== CONTAINER_CREATE_START ==='" Enter
    tmux send-keys -t $session_name "docker run --privileged --name=$container_name --ulimit core=-1 --security-opt seccomp=unconfined -dti --net=host --uts=host --ipc=host -v /home:/home -v /data1:/data1 -v /data2:/data2 -v /data3:/data3 -v /data4:/data4 --shm-size=256g --restart=always $image_name" Enter
    sleep 10
    
    # 验证容器创建
    echo -e "${BLUE}步骤2: 验证容器创建结果${NC}"
    tmux send-keys -t $session_name "echo '=== VERIFY_CREATE_START ==='" Enter
    tmux send-keys -t $session_name "docker inspect $container_name >/dev/null 2>&1 && echo 'CREATE_SUCCESS' || echo 'CREATE_FAILED'" Enter
    sleep 3
    
    local verify_output=$(tmux capture-pane -t $session_name -p | grep -A 3 "=== VERIFY_CREATE_START ===" | tail -2)
    
    if echo "$verify_output" | grep -q "CREATE_SUCCESS"; then
        echo -e "${GREEN}✅ 容器创建成功${NC}"
    else
        echo -e "${RED}❌ 容器创建失败${NC}"
        echo -e "${YELLOW}调试信息: $verify_output${NC}"
        return 1
    fi
    
    # 进入新创建的容器
    echo -e "${GREEN}🚪 步骤3: 进入新创建的容器...${NC}"
    tmux send-keys -t $session_name "docker exec -it $container_name bash" Enter
    sleep 3
    
    if ! check_tmux_output $session_name "root@.*#" 5; then
        echo -e "${RED}❌ 进入容器失败${NC}"
        return 1
    fi
    
    # 如果启用自动环境配置，执行完整配置
    if [ "$AUTO_SETUP_ENVIRONMENT" = "true" ]; then
        setup_full_environment $session_name
    else
        echo -e "${YELLOW}💡 容器已创建，如需配置环境请手动执行相关命令${NC}"
    fi
    
    return 0
}
# 完整环境配置函数（改进版 - 直接检测实际使用结果）
setup_full_environment() {
    local session_name=$1
    
    echo -e "${BLUE}🛠️  开始完整环境配置...${NC}"
    
    # 检查BOS工具
    echo -e "${BLUE}步骤1: 检查BOS工具${NC}"
    tmux send-keys -t $session_name "which bcecmd" Enter
    sleep 2
    
    if check_tmux_output $session_name "/.*bcecmd" 5; then
        echo -e "${GREEN}✅ BOS工具可用${NC}"
        
        # 配置BOS
        echo -e "${BLUE}步骤2: 配置BOS工具${NC}"
        echo "启动bcecmd配置..."
        tmux send-keys -t $session_name "bcecmd -c" Enter
        sleep 3  # 给bcecmd更多启动时间

        echo "输入Access Key..."
        tmux send-keys -t $session_name "$BOS_ACCESS_KEY" Enter
        sleep 0.1  # 调整为0.1秒间隔

        echo "输入Secret Key..."
        tmux send-keys -t $session_name "$BOS_SECRET_KEY" Enter
        sleep 0.1  # 调整为0.1秒间隔

        echo "使用默认配置（连续回车）..."
        for i in {1..11}; do
            echo "  发送第${i}个回车..."
            tmux send-keys -t $session_name "" Enter
            sleep 0.1  # 调整为0.1秒间隔
        done
        
        # 等待配置完成
        echo "等待bcecmd配置完成..."
        sleep 5

        # 改进：直接测试实际使用，不检测配置信息
        echo -e "${BLUE}步骤3: 测试BOS连接并下载配置文件${NC}"
        
        # 尝试下载第一个配置文件
        echo "测试下载.p10k.zsh配置..."
        tmux send-keys -t $session_name "echo '=== BOS_DOWNLOAD_START ==='" Enter
        tmux send-keys -t $session_name "bcecmd bos cp -y $BOS_BUCKET/.p10k.zsh /root" Enter
        sleep 5
        
        # 检查下载是否成功
        tmux send-keys -t $session_name "echo '=== BOS_DOWNLOAD_CHECK ==='" Enter
        tmux send-keys -t $session_name "if [ -f /root/.p10k.zsh ]; then echo 'BOS_DOWNLOAD_SUCCESS'; else echo 'BOS_DOWNLOAD_FAILED'; fi" Enter
        sleep 2
        
        # 检查下载结果
        local download_output=$(tmux capture-pane -t $session_name -p | tail -10)
        
        if echo "$download_output" | grep -q "BOS_DOWNLOAD_SUCCESS"; then
            echo -e "${GREEN}✅ BOS配置和连接成功！${NC}"
            
            # 继续下载其他配置文件
            echo "下载.zshrc配置..."
            tmux send-keys -t $session_name "bcecmd bos cp -y $BOS_BUCKET/.zshrc /root" Enter
            sleep 5

            echo "下载.zsh_history配置..."
            tmux send-keys -t $session_name "bcecmd bos cp -y $BOS_BUCKET/.zsh_history /root" Enter
            sleep 5

            # 验证所有配置文件
            echo -e "${BLUE}步骤4: 验证配置文件${NC}"
            tmux send-keys -t $session_name "ls -la /root/.zshrc /root/.p10k.zsh /root/.zsh_history" Enter
            sleep 2

            # 检查配置文件是否下载成功
            if check_tmux_output $session_name "\.zshrc.*\.p10k\.zsh.*\.zsh_history" 5; then
                echo -e "${GREEN}✅ 所有配置文件下载成功${NC}"
            else
                echo -e "${YELLOW}⚠️  部分配置文件可能下载失败，但核心配置已成功${NC}"
            fi
            
        else
            echo -e "${RED}❌ BOS连接或下载失败！${NC}"
            echo -e "${YELLOW}🔍 BOS操作输出：${NC}"
            echo -e "${CYAN}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
            echo "$download_output"
            echo -e "${CYAN}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
            echo ""
            echo -e "${YELLOW}💡 可能的原因：${NC}"
            echo -e "${CYAN}  1. Access Key 或 Secret Key 不正确${NC}"
            echo -e "${CYAN}  2. 网络连接问题${NC}"
            echo -e "${CYAN}  3. BOS Bucket路径不存在：$BOS_BUCKET${NC}"
            echo -e "${CYAN}  4. 权限不足或文件不存在${NC}"
            echo ""
            echo -e "${RED}🛑 BOS下载失败，切换到本地配置${NC}"
            echo -e "${YELLOW}📋 调试建议（如需要）：${NC}"
            echo -e "${CYAN}  1. 进入tmux session: tmux attach -t $session_name${NC}"
            echo -e "${CYAN}  2. 手动测试: bcecmd bos ls $BOS_BUCKET${NC}"
            echo -e "${CYAN}  3. 检查网络: ping bce.example.com${NC}"
            echo -e "${CYAN}  4. 验证密钥和bucket路径${NC}"
            echo ""
            echo -e "${BLUE}🔄 自动切换到本地备用配置...${NC}"
            setup_local_config $session_name
        fi
    else
        echo -e "${YELLOW}⚠️  BOS工具不可用，使用本地备用配置${NC}"
        setup_local_config $session_name
    fi
    
    # 创建工作目录
    echo -e "${BLUE}步骤5: 创建工作目录${NC}"
    tmux send-keys -t $session_name "mkdir -p /home/xuyehua" Enter
    sleep 1
    
    # 生成SSH密钥
    echo -e "${BLUE}步骤6: 生成SSH密钥${NC}"
    echo "生成SSH密钥（使用默认设置）..."
    tmux send-keys -t $session_name "ssh-keygen -t rsa -f ~/.ssh/id_rsa -N ''" Enter
    sleep 3
    
    # 启动zsh环境
    echo -e "${BLUE}步骤7: 启动zsh环境${NC}"
    tmux send-keys -t $session_name "echo '=== 启动zsh ==='" Enter
    tmux send-keys -t $session_name "zsh" Enter
    sleep 3
    
    # 显示SSH公钥
    echo -e "${BLUE}步骤8: 显示SSH公钥${NC}"
    tmux send-keys -t $session_name "echo \"=== SSH_KEY_DISPLAY_START ===\"" Enter
    tmux send-keys -t $session_name "if [ -f /root/.ssh/id_rsa.pub ]; then cat /root/.ssh/id_rsa.pub; else echo \"SSH公钥文件不存在\"; fi" Enter
    sleep 2
    
    # 捕获并显示公钥内容
    local ssh_key_output=$(tmux capture-pane -t $session_name -p | grep -A 3 "=== SSH_KEY_DISPLAY_START ===" | tail -2 | grep -v "^$" | head -1)
    if [ -n "$ssh_key_output" ] && ! echo "$ssh_key_output" | grep -q "不存在"; then
        echo -e "${GREEN}✅ SSH公钥已生成，内容如下：${NC}"
        echo -e "${CYAN}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
        echo -e "${YELLOW}$ssh_key_output${NC}"
        echo -e "${CYAN}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
        echo -e "${GREEN}💡 请复制上述公钥内容到目标服务器的authorized_keys文件${NC}"
    else
        echo -e "${YELLOW}⚠️  SSH公钥获取失败，请手动执行: cat /root/.ssh/id_rsa.pub${NC}"
    fi
    
    echo -e "${GREEN}✅ 完整环境配置完成！${NC}"
}
echo -e "${CYAN}=== 统一服务器连接脚本 ===${NC}"
echo "目标服务器: $SERVER_HOST"
echo "Docker容器: $DOCKER_CONTAINER"
echo "快速连接模式: $QUICK_CONNECT_MODE"
echo ""

# 检查session是否已存在
if tmux has-session -t $SESSION_NAME 2>/dev/null; then
    echo -e "${GREEN}✅ Session '$SESSION_NAME' 已存在${NC}"
    
    # 智能检查当前状态
    echo -e "${BLUE}🔍 检查当前环境状态...${NC}"
    current_output=$(tmux capture-pane -t $SESSION_NAME -p | tail -3)
    
    if echo "$current_output" | grep -q "root@.*:.*#\|.*@.*-.*:.*\$"; then
        echo -e "${GREEN}✅ 当前已在Docker容器内${NC}"
        echo -e "${YELLOW}使用: tmux attach -t $SESSION_NAME${NC}"
    else
        echo -e "${YELLOW}⚠️  当前不在Docker容器内${NC}"
        echo -e "${BLUE}🚀 尝试自动进入Docker容器...${NC}"
        smart_container_setup "$SESSION_NAME" "$DOCKER_CONTAINER" "$DOCKER_IMAGE"
    fi
    exit 0
fi

echo -e "${BLUE}🚀 创建新的tmux session: $SESSION_NAME${NC}"
tmux new-session -d -s $SESSION_NAME

echo -e "${BLUE}📡 步骤1: 启动连接工具 ($CONNECTION_TOOL)${NC}"
tmux send-keys -t $SESSION_NAME "$CONNECTION_TOOL" Enter

echo -e "${YELLOW}✋ 请完成指纹认证...${NC}"
sleep 2

# 连接流程
if [ -n "$JUMP_HOST" ] && [ "$JUMP_HOST" != "" ]; then
    echo -e "${BLUE}🔗 步骤2: 连接到跳板机 ($JUMP_HOST)${NC}"
    tmux send-keys -t $SESSION_NAME "ssh $JUMP_HOST" Enter
    
    echo -e "${BLUE}🔑 自动输入跳板机密码...${NC}"
    sleep 3
    tmux send-keys -t $SESSION_NAME "$JUMP_PASSWORD" Enter
    
    echo -e "${BLUE}🎯 步骤3: 从跳板机连接到目标服务器 ($SERVER_HOST)${NC}"
    sleep 2
    tmux send-keys -t $SESSION_NAME "ssh $SERVER_HOST" Enter
else
    echo -e "${BLUE}🎯 步骤2: 直接连接到目标服务器 ($SERVER_HOST)${NC}"
    tmux send-keys -t $SESSION_NAME "ssh $SERVER_HOST" Enter
fi

echo -e "${YELLOW}等待服务器连接建立...${NC}"
sleep 3

# 智能Docker环境设置
echo -e "${BLUE}🐳 步骤4: 智能Docker环境设置${NC}"
smart_container_setup "$SESSION_NAME" "$DOCKER_CONTAINER" "$DOCKER_IMAGE"

echo -e "${GREEN}✅ 连接流程完成${NC}"
echo -e "${YELLOW}使用: tmux attach -t $SESSION_NAME${NC}"

echo ""
echo -e "${CYAN}💡 功能说明：${NC}"
if [ -n "$JUMP_HOST" ] && [ "$JUMP_HOST" != "" ]; then
    echo -e "${CYAN}  🔗 连接流程: $CONNECTION_TOOL → $JUMP_HOST → $SERVER_HOST → Docker容器${NC}"
else
    echo -e "${CYAN}  🎯 连接流程: $CONNECTION_TOOL → $SERVER_HOST → Docker容器${NC}"
fi
echo -e "${CYAN}  🚀 智能模式: 自动检测容器状态，快速进入或完整配置${NC}"
echo -e "${CYAN}  ⚡ 快速连接: 现有容器直接进入，无需重复配置${NC}"
echo -e "${CYAN}  🛠️  完整配置: 新容器自动配置BOS、SSH密钥、zsh环境${NC}"
echo -e "${CYAN}  📱 使用方式: tmux attach -t $SESSION_NAME${NC}"