#!/bin/bash
# ============================================================
#  📨 등급별 텔레그램 알림 시스템
#  서버 배포 경로: /home/altablue/telegram_notify.sh
#
#  사용법:
#    ./telegram_notify.sh "CRITICAL" "서버가 해킹당했습니다!"
#    ./telegram_notify.sh "WARNING"  "의심스러운 파일 발견"
#    ./telegram_notify.sh "INFO"     "일일 스캔 완료"
#
#  등급:
#    CRITICAL (🔴) - 즉각 대응 필요 (백도어 발견, 파일 변조 등)
#    WARNING  (🟡) - 주의 필요 (의심 파일, 비정상 접근 등)
#    INFO     (🔵) - 정보성 (스캔 완료, 시스템 상태 등)
# ============================================================

# 설정 파일 로드
CONFIG_FILE="/etc/security-watchdog/config"
if [ -f "$CONFIG_FILE" ]; then
    source "$CONFIG_FILE"
else
    # 폴백: 직접 설정
    BOT_TOKEN="8095407589:AAFI7_XVRk1Z1J85hug5K9iYJQORI2DoDNI"
    CHAT_ID="5935655309"
    SERVER_NAME="AltaBlue"
fi

# ── 인자 처리 ──────────────────────────────────────
LEVEL="${1:-INFO}"
MESSAGE="${2:-메시지 없음}"
TIMESTAMP=$(date "+%Y-%m-%d %H:%M:%S")

# ── 등급별 이모지 및 헤더 ─────────────────────────
case "$LEVEL" in
    CRITICAL)
        EMOJI="🔴🚨"
        HEADER="긴급 보안 경고"
        ;;
    WARNING)
        EMOJI="🟡⚠️"
        HEADER="보안 경고"
        ;;
    INFO)
        EMOJI="🔵ℹ️"
        HEADER="보안 알림"
        ;;
    HEALTH)
        EMOJI="💚"
        HEADER="서버 상태"
        ;;
    DAILY)
        EMOJI="📊"
        HEADER="일일 보안 보고서"
        ;;
    *)
        EMOJI="📋"
        HEADER="알림"
        ;;
esac

# ── 메시지 포맷팅 ─────────────────────────────────
FORMATTED="${EMOJI} <b>${HEADER}</b>
━━━━━━━━━━━━━━━━━━━━
🖥️ 서버: ${SERVER_NAME:-AltaBlue}
🕐 시간: ${TIMESTAMP}
━━━━━━━━━━━━━━━━━━━━

${MESSAGE}"

# ── Rate Limiting ──────────────────────────────────
# 텔레그램 API 제한: 초당 30메시지, 그룹당 분당 20메시지
RATE_FILE="/tmp/.telegram_rate_$$"
RATE_GLOBAL="/tmp/.telegram_rate_global"

# 글로벌 속도 제한 (최소 1초 간격)
if [ -f "$RATE_GLOBAL" ]; then
    LAST_SEND=$(cat "$RATE_GLOBAL" 2>/dev/null || echo 0)
    NOW=$(date +%s)
    DIFF=$((NOW - LAST_SEND))
    if [ "$DIFF" -lt 1 ]; then
        sleep 1
    fi
fi

# ── 중복 알림 방지 ────────────────────────────────
# 동일 메시지 해시를 기반으로 쿨다운 적용
COOLDOWN=${ALERT_COOLDOWN:-300}
MSG_HASH=$(echo "$MESSAGE" | md5sum | cut -d' ' -f1)
COOLDOWN_FILE="/tmp/.telegram_cooldown_${MSG_HASH}"

if [ -f "$COOLDOWN_FILE" ]; then
    LAST_ALERT=$(cat "$COOLDOWN_FILE" 2>/dev/null || echo 0)
    NOW=$(date +%s)
    if [ $((NOW - LAST_ALERT)) -lt "$COOLDOWN" ]; then
        # CRITICAL 등급은 쿨다운 무시
        if [ "$LEVEL" != "CRITICAL" ]; then
            exit 0
        fi
    fi
fi

# ── 메시지 전송 ───────────────────────────────────
RESPONSE=$(curl -s -w "\n%{http_code}" -X POST \
    "https://api.telegram.org/bot${BOT_TOKEN}/sendMessage" \
    -d chat_id="${CHAT_ID}" \
    -d text="${FORMATTED}" \
    -d parse_mode="HTML" 2>&1)

HTTP_CODE=$(echo "$RESPONSE" | tail -1)
BODY=$(echo "$RESPONSE" | sed '$d')

# 전송 타임스탬프 기록
echo $(date +%s) > "$RATE_GLOBAL"

if [ "$HTTP_CODE" = "200" ]; then
    echo $(date +%s) > "$COOLDOWN_FILE"
    exit 0
else
    # 전송 실패 시 로그 기록
    LOG_DIR="${LOG_DIR:-/home/altablue/logs}"
    mkdir -p "$LOG_DIR"
    echo "[$(date '+%Y-%m-%d %H:%M:%S')] 텔레그램 전송 실패 (HTTP $HTTP_CODE): $BODY" >> "${LOG_DIR}/telegram_errors.log"

    # 재시도 (1회)
    sleep 2
    curl -s -X POST "https://api.telegram.org/bot${BOT_TOKEN}/sendMessage" \
        -d chat_id="${CHAT_ID}" \
        -d text="${FORMATTED}" \
        -d parse_mode="HTML" >/dev/null 2>&1

    exit 1
fi