#!/bin/bash

# 来源徽章后处理测试脚本

echo "=========================================="
echo "  来源徽章后处理功能测试"
echo "=========================================="
echo ""

# 检查后端服务状态
echo "1. 检查后端服务..."
if lsof -i :8080 > /dev/null 2>&1; then
    echo "   ✅ 后端服务运行中 (端口 8080)"
else
    echo "   ❌ 后端服务未运行"
    echo "   请先启动: cd server && go run cmd/server/main.go"
    exit 1
fi

echo ""
echo "2. 测试 API 端点..."

# 获取第一个应用ID（需要有效的JWT token）
# 这里仅作示例，实际需要有效的认证
echo "   提示：需要手动测试以下功能："
echo ""
echo "   📋 前端测试步骤："
echo "   1. 访问 http://localhost:3000"
echo "   2. 登录系统"
echo "   3. 进入任意应用的聊天界面"
echo "   4. 提问："居住证办理条件是什么？""
echo "   5. 观察回答中的徽章显示"
echo ""
echo "   ✅ 预期结果："
echo "   - 每个事实陈述后有 🔵 蓝色「知识库」徽章"
echo "   - 每个建议说明后有 🟠 橙色「AI建议」徽章"
echo "   - 末尾有完整的「来源说明」块"
echo ""

echo "3. 数据库验证..."
echo "   运行以下 SQL 查询最新的回答："
echo ""
echo "   psql -U postgres -d govai -c \\"
echo "   \"SELECT LEFT(ai_response, 500) as response_preview"
echo "    FROM app_usage_logs"
echo "    WHERE ai_response IS NOT NULL"
echo "    ORDER BY created_at DESC"
echo "    LIMIT 1;\""
echo ""
echo "   查找 [[知识库:xxx]] 或 [[AI建议]] 标记"
echo ""

echo "4. 验证后处理函数..."
cd /Users/freedak/Documents/AIDashboard/GovAI/server
if go build ./internal/handler/... 2>&1 | grep -q "error"; then
    echo "   ❌ 编译失败"
    go build ./internal/handler/...
    exit 1
else
    echo "   ✅ 后处理函数编译通过"
fi

echo ""
echo "=========================================="
echo "  测试准备完成"
echo "=========================================="
echo ""
echo "📖 详细报告："
echo "   /Users/freedak/Documents/AIDashboard/GovAI/CITATION_POST_PROCESSING_REPORT.md"
echo ""
echo "📝 解决方案文档："
echo "   /Users/freedak/Documents/AIDashboard/GovAI/CITATION_BADGE_SOLUTION.md"
echo ""
echo "🎯 下一步："
echo "   1. 如果后端未运行，启动服务"
echo "   2. 打开前端进行实际测试"
echo "   3. 观察徽章显示效果"
echo ""