#!/bin/bash

# 简单的AI连接测试脚本
# 让你确认是否已经成功接入AI

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

# 显示标题
echo -e "${CYAN}"
echo "╔══════════════════════════════════════╗"
echo "║           AI连接测试工具             ║"
echo "║      确认你是否已接入AI助手         ║"
echo "╚══════════════════════════════════════╝"
echo -e "${NC}"

# 配置API密钥（请先修改这里！）
API_KEY="sk-61d6716fe6b2452d94a0cee3bc5c4e2e"

# 测试函数：简单的AI对话
test_ai_connection() {
    echo -e "${BLUE}🔄 尝试连接AI助手...${NC}"
    echo -e "${YELLOW}📡 发送测试请求到DeepSeek API...${NC}"
    
    # 简单的测试请求
    response=$(curl -s -X POST "https://api.deepseek.com/v1/chat/completions" \
        -H "Content-Type: application/json" \
        -H "Authorization: Bearer $API_KEY" \
        -d '{
            "model": "deepseek-coder",
            "messages": [
                {
                    "role": "system", 
                    "content": "你是一个AI助手。请用中文简短回复，确认连接成功。"
                },
                {
                    "role": "user",
                    "content": "请回复'随机50个字注意一定要50个字或50个字以上。'来确认我们的连接正常。"
                }
            ],
            "max_tokens": 50,
            "temperature": 0.1
        }')
    
    # 检查响应
    if [ $? -ne 0 ]; then
        echo -e "${RED}❌ 网络连接失败${NC}"
        return 1
    fi
    
    # 提取AI回复
    ai_reply=$(echo "$response" | grep -o '"content":"[^"]*"' | cut -d'"' -f4)
    
    if [ -z "$ai_reply" ]; then
        error_msg=$(echo "$response" | grep -o '"message":"[^"]*"' | cut -d'"' -f4)
        if [ -n "$error_msg" ]; then
            echo -e "${RED}❌ API错误: $error_msg${NC}"
        else
            echo -e "${RED}❌ 无法解析AI响应${NC}"
        fi
        return 1
    fi
    
    echo -e "${GREEN}✅ AI回复: $ai_reply${NC}"
    return 0
}

# 检查API密钥
check_api_key() {
    echo -e "${YELLOW}🔑 检查API密钥配置...${NC}"
    
    if [ "$API_KEY" = "sk-your-api-key-here" ] || [ -z "$API_KEY" ]; then
        echo -e "${RED}❌ 请先设置你的DeepSeek API密钥！${NC}"
        echo ""
        echo -e "${CYAN}修改方法:${NC}"
        echo "1. 打开脚本: nano ai_test.sh"
        echo "2. 找到 API_KEY= 这一行"
        echo "3. 替换为你的真实API密钥"
        echo "4. 保存并重新运行脚本"
        echo ""
        return 1
    fi
    
    if [[ "$API_KEY" != sk-* ]]; then
        echo -e "${RED}❌ API密钥格式不正确${NC}"
        return 1
    fi
    
    echo -e "${GREEN}✅ API密钥格式正确${NC}"
    return 0
}

# 系统信息显示
show_system_info() {
    echo -e "${CYAN}🖥️ 系统信息:${NC}"
    echo "主机名: $(hostname)"
    echo "系统: $(uname -srm)"
    echo "时间: $(date)"
    echo "工作目录: $(pwd)"
    echo ""
}

# 主测试流程
main_test() {
    echo -e "${BLUE}开始AI连接测试...${NC}"
    echo ""
    
    # 显示系统信息
    show_system_info
    
    # 检查API密钥
    if ! check_api_key; then
        exit 1
    fi
    
    echo -e "${YELLOW}⏳ 正在进行连接测试...${NC}"
    echo ""
    
    # 测试AI连接
    if test_ai_connection; then
        echo ""
        echo -e "${GREEN}🎉 恭喜！AI连接测试成功！${NC}"
        echo -e "${GREEN}🤖 你现在已经正式接入AI助手了！${NC}"
        echo ""
        echo -e "${CYAN}接下来你可以:${NC}"
        echo "✅ 运行完整的AI修复脚本"
        echo "✅ 使用AI助手解决各种问题"
        echo "✅ 享受AI驱动的自动化体验"
    else
        echo ""
        echo -e "${RED}😞 AI连接测试失败${NC}"
        echo -e "${YELLOW}请检查:${NC}"
        echo "🔹 API密钥是否正确"
        echo "🔹 网络连接是否正常"
        echo "🔹 账户余额是否充足"
        echo "🔹 API密钥是否有访问权限"
    fi
}

# 交互式选项
case "${1:-}" in
    "help")
        echo -e "${CYAN}使用方法:${NC}"
        echo "  ./ai_test.sh       - 运行AI连接测试"
        echo "  ./ai_test.sh help  - 显示帮助信息"
        ;;
    *)
        main_test
        ;;
esac