#!/bin/bash
# Loki 集成验证脚本

set -e

echo "========================================="
echo "Loki 日志收集系统验证"
echo "========================================="
echo ""

# 颜色定义
GREEN='\033[0;32m'
RED='\033[0;31m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

# 检查服务状态
echo "1. 检查服务状态..."
echo "-------------------"
docker-compose ps

echo ""
echo "2. 检查 Loki 健康状态..."
echo "-------------------"
if curl -s http://localhost:3100/ready | grep -q "ready"; then
    echo -e "${GREEN}✓ Loki 服务正常${NC}"
else
    echo -e "${RED}✗ Loki 服务异常${NC}"
    exit 1
fi

echo ""
echo "3. 检查 Promtail 健康状态..."
echo "-------------------"
if curl -s http://localhost:9080/ready | grep -q "ready"; then
    echo -e "${GREEN}✓ Promtail 服务正常${NC}"
else
    echo -e "${RED}✗ Promtail 服务异常${NC}"
    exit 1
fi

echo ""
echo "4. 生成测试日志..."
echo "-------------------"
curl -X POST http://localhost:8111/invoke \
  -H "Content-Type: application/json" \
  -d '{"algorithm": "PrimeChecker", "params": {"number": 17}}' \
  -s -o /dev/null -w "HTTP Status: %{http_code}\n"

echo ""
echo "5. 等待日志收集 (5秒)..."
sleep 5

echo ""
echo "6. 查询 Loki 日志..."
echo "-------------------"
LOGS=$(curl -G -s "http://localhost:3100/loki/api/v1/query_range" \
  --data-urlencode 'query={job="functional-scaffold-app"}' \
  --data-urlencode 'limit=5')

if echo "$LOGS" | jq -e '.data.result | length > 0' > /dev/null 2>&1; then
    echo -e "${GREEN}✓ 成功查询到日志${NC}"
    echo ""
    echo "最近的日志条目："
    echo "$LOGS" | jq -r '.data.result[0].values[-1][1]' | head -3
else
    echo -e "${YELLOW}⚠ 暂时没有查询到日志，可能需要等待更长时间${NC}"
fi

echo ""
echo "7. 检查 Grafana 数据源..."
echo "-------------------"
DATASOURCES=$(curl -s -u admin:admin http://localhost:3000/api/datasources)
if echo "$DATASOURCES" | jq -e '.[] | select(.name == "Loki")' > /dev/null 2>&1; then
    echo -e "${GREEN}✓ Loki 数据源已配置${NC}"
else
    echo -e "${RED}✗ Loki 数据源未配置${NC}"
fi

if echo "$DATASOURCES" | jq -e '.[] | select(.name == "Prometheus")' > /dev/null 2>&1; then
    echo -e "${GREEN}✓ Prometheus 数据源已配置${NC}"
else
    echo -e "${RED}✗ Prometheus 数据源未配置${NC}"
fi

echo ""
echo "========================================="
echo "验证完成！"
echo "========================================="
echo ""
echo "访问地址："
echo "  - Grafana: http://localhost:3000 (admin/admin)"
echo "  - Loki: http://localhost:3100"
echo "  - Promtail: http://localhost:9080"
echo ""
echo "查看日志："
echo "  1. 访问 Grafana Explore: http://localhost:3000/explore"
echo "  2. 选择 Loki 数据源"
echo "  3. 输入查询: {job=\"functional-scaffold-app\"}"
echo ""