#!/bin/bash
set -euo pipefail

# ==========================================================
# Alertmanager 测试脚本（含启动等待）
# ==========================================================

ALERTMANAGER_URL="http://localhost:9093"
TEST_ALERT_NAME_CRITICAL="NodeDown"
TEST_ALERT_NAME_WARNING="HighCPU"
TMP_LOG="/tmp/test-alertmanager.log"

# 等待参数
am_wait_attempts=30
am_wait_interval=2

GREEN="\033[1;32m"
RED="\033[1;31m"
YELLOW="\033[1;33m"
RESET="\033[0m"

# ==========================================================
# 函数定义
# ==========================================================

wait_for_alertmanager() {
  local attempt=1
  echo "[INFO] 等待 Alertmanager 启动中..."
  while (( attempt <= am_wait_attempts )); do
    if curl -fsS "${ALERTMANAGER_URL}/api/v2/status" >/dev/null 2>&1; then
      echo -e "${GREEN}[OK] Alertmanager 已就绪 (attempt=${attempt}/${am_wait_attempts})${RESET}"
      return 0
    fi
    echo "[..] Alertmanager 尚未就绪 (${attempt}/${am_wait_attempts})"
    sleep "${am_wait_interval}"
    (( attempt++ ))
  done
  echo -e "${RED}[ERROR] Alertmanager 在 ${am_wait_attempts} 次尝试后仍未就绪${RESET}"
  return 1
}

log_step() {
  echo -e "${YELLOW}==== $1 ====${RESET}"
}

# ==========================================================
# 主流程
# ==========================================================

log_step "查询 Alertmanager 当前告警列表开始"
echo "[INFO] Alertmanager 地址: $ALERTMANAGER_URL"

# Step 1: 等待 Alertmanager 启动
wait_for_alertmanager

# Step 2: 查询当前告警列表
echo "[INFO] 查询当前告警..."
sleep 1
curl -fsS "${ALERTMANAGER_URL}/api/v2/alerts" | jq '.' || {
  echo -e "${RED}[WARN] 无法解析返回 JSON，请检查 jq 是否安装${RESET}"
  curl -s "${ALERTMANAGER_URL}/api/v2/alerts"
}

# Step 3: 检查告警是否包含 NodeDown
if curl -fsS "${ALERTMANAGER_URL}/api/v2/alerts" | grep -q "${TEST_ALERT_NAME_CRITICAL}"; then
  echo -e "${GREEN}✅ 测试通过：Alertmanager 已成功接收告警 ${TEST_ALERT_NAME_CRITICAL}${RESET}"
else
  echo -e "${RED}❌ 测试失败：未检测到告警 ${TEST_ALERT_NAME_CRITICAL}${RESET}"
fi

log_step "测试结束"