#!/bin/bash
# ===========================================
# mail-lib.sh — 邮件发送共享库
# 供 daily-report.sh / alert.sh 复用：
#   mail_send <主题> <HTML正文>  （带重试 + 日志 + 轮转）
# ===========================================

# ---------- 可配置项（环境变量可覆盖） ----------
: "${MAIL_TO:=m943790568@163.com}"                  # 收件人，多个用逗号分隔
: "${MAIL_FROM:=ASD 监控 <m943790568@163.com>}"     # 发件人显示名
: "${MAIL_ACCOUNT:=default}"                        # msmtp 账户（~/.msmtprc 中的 account）
: "${MAIL_LOG_DIR:=/root/rust/scripts/logs}"        # 日志目录
: "${MAIL_RETRY:=3}"                                # 发送重试次数
: "${MAIL_LOG_MAX_KB:=1024}"                        # mail.log 轮转阈值（KB）

mkdir -p "$MAIL_LOG_DIR" 2>/dev/null || true

# 写日志
mail_log() {
    local level="$1"; shift
    echo "[$(date '+%Y-%m-%d %H:%M:%S')] [${level}] $*" >> "${MAIL_LOG_DIR}/mail.log" 2>/dev/null || true
}

# 日志轮转：超过阈值保留末尾 200 行
mail_rotate_log() {
    local f="${MAIL_LOG_DIR}/mail.log"
    [ -f "$f" ] || return 0
    local size_kb=$(( $(stat -c %s "$f" 2>/dev/null || echo 0) / 1024 ))
    if [ "$size_kb" -gt "$MAIL_LOG_MAX_KB" ]; then
        tail -n 200 "$f" > "$f.tmp" 2>/dev/null && mv "$f.tmp" "$f" 2>/dev/null || true
        mail_log INFO "mail.log 已轮转"
    fi
}

# 发送 HTML 邮件（UTF-8），失败自动重试
# 用法: mail_send "主题" "HTML 正文"
mail_send() {
    local subject="$1" body="$2"
    mail_rotate_log

    local recipients headers
    recipients=$(echo "$MAIL_TO" | tr ',' ' ')
    headers=$(cat <<EOF
From: ${MAIL_FROM}
To: ${MAIL_TO}
Subject: ${subject}
MIME-Version: 1.0
Content-Type: text/html; charset=UTF-8
Content-Transfer-Encoding: 8bit
EOF
)

    local attempt=1 ok=0
    while [ "$attempt" -le "$MAIL_RETRY" ]; do
        if { echo "$headers"; echo; echo "$body"; } | msmtp -a "$MAIL_ACCOUNT" $recipients 2>>"${MAIL_LOG_DIR}/mail.log"; then
            ok=1
            mail_log INFO "邮件发送成功: ${subject} → ${MAIL_TO}（第 ${attempt} 次尝试）"
            break
        fi
        mail_log WARN "邮件发送失败（第 ${attempt}/${MAIL_RETRY} 次）: ${subject}"
        attempt=$((attempt + 1))
        [ "$attempt" -le "$MAIL_RETRY" ] && sleep $((attempt * 5))
    done

    if [ "$ok" -ne 1 ]; then
        mail_log ERROR "邮件最终发送失败: ${subject}"
        return 1
    fi
    return 0
}