#!/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
    echo "❌ 설정 파일을 찾을 수 없습니다: $CONFIG_FILE" >&2
    echo "   설치: sudo /home/altablue/deploy_security.sh" >&2
    exit 1
fi

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

# ── 등급별 이모지 및 헤더 ─────────────────────────
case "$LEVEL" in
    CRITICAL)
        EMOJI="🔴🚨"
        HEADER="긴급 보안 경고"
        ;;
    WARNING)
        EMOJI="🟡⚠️"
        HEADER="보안 경고"
        ;;
    HIGH)
        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 (flock 기반 동시 접근 보호) ──────
LOCK_FILE="/tmp/.telegram_notify.lock"
RATE_GLOBAL="/tmp/.telegram_rate_global"

(
    # 잠금 획득 (최대 5초 대기)
    flock -w 5 200 || {
        echo "[$(date '+%Y-%m-%d %H:%M:%S')] ⚠️ 잠금 획득 실패 (다른 알림 전송 중)" >&2
        exit 1
    }

    # 글로벌 속도 제한 (최소 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')

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

    if [ "$HTTP_CODE" = "200" ]; then
        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

) 200>"$LOCK_FILE"