#!/bin/bash

echo "🧪 测试大模型微调平台 - 所有服务"
echo "=================================="
echo ""

BASE_URL="http://localhost"

# 检查服务是否运行
echo "1. 检查服务状态..."
echo ""

# 检查端口 8000 (Web 前端)
if curl -s "${BASE_URL}:8000" > /dev/null 2>&1; then
    echo "✅ Web 前端服务正在运行 (端口 8000)"
else
    echo "❌ Web 前端服务未运行 (端口 8000)"
fi

# 检查端口 8001 (FastAPI)
if curl -s "${BASE_URL}:8001" > /dev/null 2>&1; then
    echo "✅ FastAPI 服务正在运行 (端口 8001)"
else
    echo "❌ FastAPI 服务未运行 (端口 8001)"
fi

echo ""
echo "=================================="
echo ""

# 获取本机IP
SERVER_IP=$(hostname -I | awk '{print $1}')

echo "📱 访问地址:"
echo ""
echo "前端页面:"
echo "   - 主页: http://$SERVER_IP:8000/pages/main.html"
echo "   - 登录: http://$SERVER_IP:8000/pages/login.html"
echo ""
echo "API 服务:"
echo "   - API 根路径: http://$SERVER_IP:8001/"
echo "   - API 健康检查: http://$SERVER_IP:8001/api/health"
echo "   - API 文档: http://$SERVER_IP:8001/docs"
echo ""

echo "=================================="
echo "2. 测试 API 接口..."
echo ""

# 测试 API
echo "测试根路径:"
curl -s "${BASE_URL}:8001/" | python3 -m json.tool 2>/dev/null || curl -s "${BASE_URL}:8001/"
echo ""

echo "测试健康检查:"
curl -s "${BASE_URL}:8001/api/health" | python3 -m json.tool 2>/dev/null || echo "请求失败"
echo ""

echo "测试数据集 API:"
curl -s "${BASE_URL}:8001/api/datasets" | python3 -m json.tool 2>/dev/null || echo "请求失败"
echo ""

echo "=================================="
echo "3. 测试前端页面..."
echo ""

# 测试前端页面
echo "测试主页:"
if curl -s -I "${BASE_URL}:8000/pages/main.html" | grep -q "200 OK"; then
    echo "✅ 主页可访问"
else
    echo "❌ 主页无法访问"
fi

echo "测试登录页:"
if curl -s -I "${BASE_URL}:8000/pages/login.html" | grep -q "200 OK"; then
    echo "✅ 登录页可访问"
else
    echo "❌ 登录页无法访问"
fi

echo ""
echo "=================================="
echo "✅ 测试完成！"
echo ""
echo "💡 如果服务未运行，请使用以下命令启动:"
echo "   ./total_start.sh"
echo ""