#!/bin/bash
###############################################################################
# AdGuard Shield - Report Generator
# Erstellt und versendet periodische Statistik-Reports per E-Mail.
#
# Nutzung:
#   report-generator.sh send       – Report sofort generieren und versenden
#   report-generator.sh test       – Test-E-Mail senden (Konfiguration prüfen)
#   report-generator.sh generate   – Report als Datei generieren (ohne Versand)
#   report-generator.sh install    – Cron-Job einrichten
#   report-generator.sh remove     – Cron-Job entfernen
#   report-generator.sh status     – Cron-Status anzeigen
#
# Crontab-Eintrag (wird automatisch verwaltet):
#   Wird je nach REPORT_INTERVAL als Cron-Job unter /etc/cron.d/ angelegt.
#
# Autor:   Patrick Asmus
# E-Mail:  support@techniverse.net
# Lizenz:  MIT
###############################################################################

set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
CONFIG_FILE="${SCRIPT_DIR}/adguard-shield.conf"
CRON_FILE="/etc/cron.d/adguard-shield-report"

# ─── Konfiguration laden ───────────────────────────────────────────────────────
if [[ ! -f "$CONFIG_FILE" ]]; then
    echo "FEHLER: Konfigurationsdatei nicht gefunden: $CONFIG_FILE" >&2
    exit 1
fi
# shellcheck source=adguard-shield.conf
source "$CONFIG_FILE"

# ─── Standardwerte ────────────────────────────────────────────────────────────
REPORT_ENABLED="${REPORT_ENABLED:-false}"
REPORT_INTERVAL="${REPORT_INTERVAL:-weekly}"
REPORT_TIME="${REPORT_TIME:-08:00}"
REPORT_EMAIL_TO="${REPORT_EMAIL_TO:-}"
REPORT_EMAIL_FROM="${REPORT_EMAIL_FROM:-adguard-shield@$(hostname -f 2>/dev/null || hostname)}"
REPORT_FORMAT="${REPORT_FORMAT:-html}"
REPORT_MAIL_CMD="${REPORT_MAIL_CMD:-msmtp}"
BAN_HISTORY_FILE="${BAN_HISTORY_FILE:-/var/log/adguard-shield-bans.log}"
BAN_HISTORY_RETENTION_DAYS="${BAN_HISTORY_RETENTION_DAYS:-0}"
STATE_DIR="${STATE_DIR:-/var/lib/adguard-shield}"
TEMPLATE_DIR="${SCRIPT_DIR}/templates"

# Version aus dem Hauptscript auslesen
VERSION="unknown"
if [[ -f "${SCRIPT_DIR}/adguard-shield.sh" ]]; then
    VERSION=$(grep -m1 '^VERSION=' "${SCRIPT_DIR}/adguard-shield.sh" 2>/dev/null | cut -d'"' -f2)
    VERSION="${VERSION:-unknown}"
fi

# ─── Logging ──────────────────────────────────────────────────────────────────
LOG_FILE="${LOG_FILE:-/var/log/adguard-shield.log}"

log() {
    local level="$1"
    shift
    local message="$*"
    local timestamp
    timestamp="$(date '+%Y-%m-%d %H:%M:%S')"
    echo "[$timestamp] [$level] [REPORT] $message" | tee -a "$LOG_FILE" >&2
}

# ─── Berichtszeitraum berechnen ───────────────────────────────────────────────

# Gibt Epoch-Wert für heute 00:00:00 (Mitternacht) zurück
get_today_midnight() {
    date -d "today 00:00:00" '+%s' 2>/dev/null || date -v0H -v0M -v0S '+%s'
}

get_report_period() {
    local today_midnight
    today_midnight=$(get_today_midnight)
    # Ende des Berichtszeitraums ist immer das Ende von gestern (23:59:59)
    local end_epoch=$((today_midnight - 1))
    local start_epoch
    local period_label

    case "$REPORT_INTERVAL" in
        daily)
            start_epoch=$((today_midnight - 86400))
            period_label="Tagesbericht"
            ;;
        weekly)
            start_epoch=$((today_midnight - 7 * 86400))
            period_label="Wochenbericht"
            ;;
        biweekly)
            start_epoch=$((today_midnight - 14 * 86400))
            period_label="Zweiwochenbericht"
            ;;
        monthly)
            start_epoch=$((today_midnight - 30 * 86400))
            period_label="Monatsbericht"
            ;;
        *)
            start_epoch=$((today_midnight - 7 * 86400))
            period_label="Bericht"
            ;;
    esac

    local start_date
    start_date=$(date -d "@$start_epoch" '+%d.%m.%Y' 2>/dev/null || date -r "$start_epoch" '+%d.%m.%Y')
    local end_date
    end_date=$(date -d "@$end_epoch" '+%d.%m.%Y' 2>/dev/null || date -r "$end_epoch" '+%d.%m.%Y')

    if [[ "$REPORT_INTERVAL" == "daily" ]]; then
        echo "${period_label}: ${start_date}"
    else
        echo "${period_label}: ${start_date} – ${end_date}"
    fi
}

get_period_start_epoch() {
    local today_midnight
    today_midnight=$(get_today_midnight)

    case "$REPORT_INTERVAL" in
        daily)    echo $((today_midnight - 86400)) ;;
        weekly)   echo $((today_midnight - 7 * 86400)) ;;
        biweekly) echo $((today_midnight - 14 * 86400)) ;;
        monthly)  echo $((today_midnight - 30 * 86400)) ;;
        *)        echo $((today_midnight - 7 * 86400)) ;;
    esac
}

get_period_end_epoch() {
    # Ende des Berichtszeitraums = Ende von gestern (heute 00:00:00 minus 1 Sekunde)
    local today_midnight
    today_midnight=$(get_today_midnight)
    echo $((today_midnight - 1))
}

# ─── Ban-History filtern nach Zeitraum ────────────────────────────────────────
# Gibt nur Zeilen zurück, deren Zeitstempel im Berichtszeitraum liegen
filter_history_by_period() {
    local start_epoch="$1"
    local end_epoch="$2"

    if [[ ! -f "$BAN_HISTORY_FILE" ]]; then
        return
    fi

    while IFS= read -r line; do
        # Kommentare und leere Zeilen überspringen
        [[ "$line" =~ ^#.*$ || -z "$line" ]] && continue

        # Zeitstempel extrahieren (erstes Feld, Format: YYYY-MM-DD HH:MM:SS)
        local timestamp
        timestamp=$(echo "$line" | awk -F'|' '{print $1}' | xargs)

        if [[ -z "$timestamp" ]]; then
            continue
        fi

        # Timestamp in Epoch umwandeln
        local line_epoch
        line_epoch=$(date -d "$timestamp" '+%s' 2>/dev/null || date -j -f '%Y-%m-%d %H:%M:%S' "$timestamp" '+%s' 2>/dev/null || echo "0")

        if [[ "$line_epoch" -ge "$start_epoch" && "$line_epoch" -le "$end_epoch" ]]; then
            echo "$line"
        fi
    done < "$BAN_HISTORY_FILE"
}

# ─── Ban-History bereinigen ────────────────────────────────────────────────────
# Entfernt Einträge älter als BAN_HISTORY_RETENTION_DAYS (0 = deaktiviert)
cleanup_ban_history() {
    [[ ! -f "$BAN_HISTORY_FILE" ]] && return
    [[ "$BAN_HISTORY_RETENTION_DAYS" == "0" || -z "$BAN_HISTORY_RETENTION_DAYS" ]] && return

    local cutoff_epoch
    cutoff_epoch=$(date -d "-${BAN_HISTORY_RETENTION_DAYS} days" '+%s' 2>/dev/null)
    [[ -z "$cutoff_epoch" ]] && return

    local tmp_file="${BAN_HISTORY_FILE}.tmp"
    local removed=0

    while IFS= read -r line; do
        # Header-Zeilen immer beibehalten
        if [[ "$line" =~ ^#.*$ || -z "$line" ]]; then
            echo "$line"
            continue
        fi

        local timestamp
        timestamp=$(echo "$line" | awk -F'|' '{print $1}' | xargs)
        local line_epoch
        line_epoch=$(date -d "$timestamp" '+%s' 2>/dev/null || echo "0")

        if [[ "$line_epoch" -ge "$cutoff_epoch" ]]; then
            echo "$line"
        else
            ((removed++)) || true
        fi
    done < "$BAN_HISTORY_FILE" > "$tmp_file"

    if [[ $removed -gt 0 ]]; then
        mv "$tmp_file" "$BAN_HISTORY_FILE"
        log "INFO" "Ban-History bereinigt: $removed Einträge älter als ${BAN_HISTORY_RETENTION_DAYS} Tage entfernt"
    else
        rm -f "$tmp_file"
    fi
}

# ─── Statistiken berechnen ────────────────────────────────────────────────────
calculate_stats() {
    # Ban-History bereinigen (falls Retention konfiguriert)
    cleanup_ban_history

    local start_epoch
    start_epoch=$(get_period_start_epoch)
    local end_epoch
    end_epoch=$(get_period_end_epoch)

    local filtered_data
    filtered_data=$(filter_history_by_period "$start_epoch" "$end_epoch")

    # Wenn keine Daten vorhanden, Standardwerte
    if [[ -z "$filtered_data" ]]; then
        TOTAL_BANS=0
        TOTAL_UNBANS=0
        UNIQUE_IPS=0
        PERMANENT_BANS=0
        ACTIVE_BANS=0
        ABUSEIPDB_REPORTS=0
        RATELIMIT_BANS=0
        SUBDOMAIN_FLOOD_BANS=0
        EXTERNAL_BLOCKLIST_BANS=0
        BUSIEST_DAY="–"
        TOP10_IPS=""
        TOP10_DOMAINS=""
        PROTOCOL_STATS=""
        RECENT_BANS=""
        return
    fi

    # Gesamtzahl Sperren
    TOTAL_BANS=$(echo "$filtered_data" | grep -c '| BAN ' || echo "0")

    # Gesamtzahl Entsperrungen
    TOTAL_UNBANS=$(echo "$filtered_data" | grep -c '| UNBAN ' || echo "0")

    # Eindeutige IPs (nur BAN-Einträge)
    UNIQUE_IPS=$(echo "$filtered_data" | grep '| BAN ' | awk -F'|' '{print $3}' | xargs -I{} echo {} | sort -u | wc -l | xargs)

    # Permanente Sperren (Dauer enthält "PERMANENT" oder "permanent")
    PERMANENT_BANS=$(echo "$filtered_data" | grep '| BAN ' | awk -F'|' '{print $6}' | grep -ic 'permanent' || echo "0")

    # Aktuell aktive Sperren (aus State-Dateien)
    ACTIVE_BANS=0
    if [[ -d "$STATE_DIR" ]]; then
        for f in "${STATE_DIR}"/*.ban; do
            [[ -f "$f" ]] && ACTIVE_BANS=$((ACTIVE_BANS + 1))
        done
    fi

    # AbuseIPDB Reports (suche in Log-Datei)
    ABUSEIPDB_REPORTS=0
    if [[ -f "$LOG_FILE" ]]; then
        local abuseipdb_start_date
        abuseipdb_start_date=$(date -d "@$start_epoch" '+%Y-%m-%d' 2>/dev/null || date -r "$start_epoch" '+%Y-%m-%d')
        ABUSEIPDB_REPORTS=$(grep -c "AbuseIPDB: IP .* erfolgreich gemeldet" "$LOG_FILE" 2>/dev/null | head -1 || echo "0")
    fi

    # Angriffsarten
    RATELIMIT_BANS=$(echo "$filtered_data" | grep '| BAN ' | awk -F'|' '{print $8}' | grep -ic 'rate-limit' || echo "0")
    SUBDOMAIN_FLOOD_BANS=$(echo "$filtered_data" | grep '| BAN ' | awk -F'|' '{print $8}' | grep -ic 'subdomain-flood' || echo "0")
    EXTERNAL_BLOCKLIST_BANS=$(echo "$filtered_data" | grep '| BAN ' | awk -F'|' '{print $8}' | grep -ic 'external-blocklist' || echo "0")

    # Aktivster Tag
    BUSIEST_DAY=$(echo "$filtered_data" | grep '| BAN ' | awk -F'|' '{print $1}' | xargs -I{} echo {} | awk '{print $1}' | sort | uniq -c | sort -rn | head -1 | awk '{print $2}' || echo "–")
    if [[ -z "$BUSIEST_DAY" ]]; then
        BUSIEST_DAY="–"
    else
        # Datum in DE-Format umwandeln
        local busiest_formatted
        busiest_formatted=$(date -d "$BUSIEST_DAY" '+%d.%m.%Y' 2>/dev/null || echo "$BUSIEST_DAY")
        BUSIEST_DAY="$busiest_formatted"
    fi

    # Top 10 IPs (nach Häufigkeit der Sperren)
    TOP10_IPS=$(echo "$filtered_data" | grep '| BAN ' | awk -F'|' '{print $3}' | xargs -I{} echo {} | sort | uniq -c | sort -rn | head -10)

    # Top 10 Domains (nach Häufigkeit)
    TOP10_DOMAINS=$(echo "$filtered_data" | grep '| BAN ' | awk -F'|' '{print $4}' | xargs -I{} echo {} | sed 's/^-$//' | grep -v '^$' | sort | uniq -c | sort -rn | head -10)

    # Protokoll-Verteilung
    PROTOCOL_STATS=$(echo "$filtered_data" | grep '| BAN ' | awk -F'|' '{print $7}' | xargs -I{} echo {} | sed 's/^-$/unbekannt/' | grep -v '^$' | sort | uniq -c | sort -rn)

    # Letzte 10 Sperren
    RECENT_BANS=$(echo "$filtered_data" | grep '| BAN ' | tail -10)
}

# ─── HTML-Tabellen generieren ─────────────────────────────────────────────────
generate_top10_ips_html() {
    if [[ -z "$TOP10_IPS" ]]; then
        echo '<div class="no-data">Keine Daten im Berichtszeitraum</div>'
        return
    fi

    local max_count
    max_count=$(echo "$TOP10_IPS" | head -1 | awk '{print $1}')

    local html='<table><tr><th>#</th><th>IP-Adresse</th><th>Sperren</th></tr>'
    local rank=0

    while read -r count ip; do
        [[ -z "$count" || -z "$ip" ]] && continue
        rank=$((rank + 1))
        local rank_class=""
        [[ $rank -le 3 ]] && rank_class=" top3"
        local bar_width=100
        if [[ "$max_count" -gt 0 ]]; then
            bar_width=$((count * 100 / max_count))
        fi
        html+="<tr><td><span class=\"rank${rank_class}\">${rank}</span></td>"
        html+="<td class=\"ip-cell\">${ip}</td>"
        html+="<td><div class=\"bar-container\"><div class=\"bar\" style=\"width:${bar_width}%\"></div><span class=\"bar-value\">${count}</span></div></td>"
        html+="</tr>"
    done <<< "$TOP10_IPS"

    html+='</table>'
    echo "$html"
}

generate_top10_domains_html() {
    if [[ -z "$TOP10_DOMAINS" ]]; then
        echo '<div class="no-data">Keine Daten im Berichtszeitraum</div>'
        return
    fi

    local max_count
    max_count=$(echo "$TOP10_DOMAINS" | head -1 | awk '{print $1}')

    local html='<table><tr><th>#</th><th>Domain</th><th>Sperren</th></tr>'
    local rank=0

    while read -r count domain; do
        [[ -z "$count" || -z "$domain" ]] && continue
        rank=$((rank + 1))
        local rank_class=""
        [[ $rank -le 3 ]] && rank_class=" top3"
        local bar_width=100
        if [[ "$max_count" -gt 0 ]]; then
            bar_width=$((count * 100 / max_count))
        fi
        html+="<tr><td><span class=\"rank${rank_class}\">${rank}</span></td>"
        html+="<td>${domain}</td>"
        html+="<td><div class=\"bar-container\"><div class=\"bar\" style=\"width:${bar_width}%\"></div><span class=\"bar-value\">${count}</span></div></td>"
        html+="</tr>"
    done <<< "$TOP10_DOMAINS"

    html+='</table>'
    echo "$html"
}

generate_protocol_html() {
    if [[ -z "$PROTOCOL_STATS" ]]; then
        echo '<div class="no-data">Keine Daten im Berichtszeitraum</div>'
        return
    fi

    local html='<table><tr><th>Protokoll</th><th>Anzahl Sperren</th></tr>'

    while read -r count proto; do
        [[ -z "$count" || -z "$proto" ]] && continue
        local badge_class=""
        case "${proto,,}" in
            dns*)     badge_class="dns" ;;
            doh*)     badge_class="doh" ;;
            dot*)     badge_class="dot" ;;
            doq*)     badge_class="doq" ;;
        esac
        html+="<tr><td><span class=\"protocol-badge ${badge_class}\">${proto}</span></td>"
        html+="<td>${count}</td></tr>"
    done <<< "$PROTOCOL_STATS"

    html+='</table>'
    echo "$html"
}

generate_recent_bans_html() {
    if [[ -z "$RECENT_BANS" ]]; then
        echo '<div class="no-data">Keine Sperren im Berichtszeitraum</div>'
        return
    fi

    local html='<table><tr><th>Zeitpunkt</th><th>IP</th><th>Domain</th><th>Grund</th></tr>'

    while IFS='|' read -r timestamp action ip domain count duration protocol reason; do
        timestamp=$(echo "$timestamp" | xargs)
        ip=$(echo "$ip" | xargs)
        [[ -z "$timestamp" && -z "$ip" ]] && continue
        domain=$(echo "$domain" | xargs)
        reason=$(echo "$reason" | xargs)
        [[ "$domain" == "-" ]] && domain="–"
        [[ -z "$reason" ]] && reason="rate-limit"

        local reason_class="rate-limit"
        [[ "$reason" == *"subdomain"* ]] && reason_class="subdomain-flood"
        [[ "$reason" == *"external"* ]] && reason_class="external"

        # Datum kürzen für Anzeige
        local short_time
        short_time=$(echo "$timestamp" | awk '{print $1" "$2}' | cut -c6-)

        html+="<tr><td>${short_time}</td>"
        html+="<td class=\"ip-cell\">${ip}</td>"
        html+="<td>${domain}</td>"
        html+="<td><span class=\"reason-badge ${reason_class}\">${reason}</span></td>"
        html+="</tr>"
    done <<< "$RECENT_BANS"

    html+='</table>'
    echo "$html"
}

# ─── TXT-Tabellen generieren ──────────────────────────────────────────────────
generate_top10_ips_txt() {
    if [[ -z "$TOP10_IPS" ]]; then
        echo "  Keine Daten im Berichtszeitraum"
        return
    fi

    local rank=0
    printf "  %-4s %-42s %s\n" "#" "IP-Adresse" "Sperren"
    printf "  %-4s %-42s %s\n" "──" "──────────────────────────────────────────" "───────"

    while read -r count ip; do
        [[ -z "$count" || -z "$ip" ]] && continue
        rank=$((rank + 1))
        printf "  %-4s %-42s %s\n" "${rank}." "$ip" "$count"
    done <<< "$TOP10_IPS"
}

generate_top10_domains_txt() {
    if [[ -z "$TOP10_DOMAINS" ]]; then
        echo "  Keine Daten im Berichtszeitraum"
        return
    fi

    local rank=0
    printf "  %-4s %-42s %s\n" "#" "Domain" "Sperren"
    printf "  %-4s %-42s %s\n" "──" "──────────────────────────────────────────" "───────"

    while read -r count domain; do
        [[ -z "$count" || -z "$domain" ]] && continue
        rank=$((rank + 1))
        printf "  %-4s %-42s %s\n" "${rank}." "$domain" "$count"
    done <<< "$TOP10_DOMAINS"
}

generate_protocol_txt() {
    if [[ -z "$PROTOCOL_STATS" ]]; then
        echo "  Keine Daten im Berichtszeitraum"
        return
    fi

    printf "  %-20s %s\n" "Protokoll" "Anzahl"
    printf "  %-20s %s\n" "────────────────────" "──────"

    while read -r count proto; do
        [[ -z "$count" || -z "$proto" ]] && continue
        printf "  %-20s %s\n" "$proto" "$count"
    done <<< "$PROTOCOL_STATS"
}

generate_recent_bans_txt() {
    if [[ -z "$RECENT_BANS" ]]; then
        echo "  Keine Sperren im Berichtszeitraum"
        return
    fi

    printf "  %-17s %-42s %-30s %s\n" "Zeitpunkt" "IP" "Domain" "Grund"
    printf "  %-17s %-42s %-30s %s\n" "─────────────────" "──────────────────────────────────────────" "──────────────────────────────" "──────────"

    while IFS='|' read -r timestamp action ip domain count duration protocol reason; do
        timestamp=$(echo "$timestamp" | xargs)
        ip=$(echo "$ip" | xargs)
        [[ -z "$timestamp" && -z "$ip" ]] && continue
        domain=$(echo "$domain" | xargs)
        reason=$(echo "$reason" | xargs)
        [[ "$domain" == "-" ]] && domain="–"
        [[ -z "$reason" ]] && reason="rate-limit"

        local short_time
        short_time=$(echo "$timestamp" | awk '{print $1" "$2}' | cut -c6-)

        printf "  %-17s %-42s %-30s %s\n" "$short_time" "$ip" "$domain" "$reason"
    done <<< "$RECENT_BANS"
}

# ─── Report generieren ────────────────────────────────────────────────────────
generate_report() {
    local format="${1:-$REPORT_FORMAT}"

    log "INFO" "Generiere ${format^^}-Report..."

    # Statistiken berechnen
    calculate_stats

    local report_period
    report_period=$(get_report_period)
    local report_date
    report_date=$(date '+%d.%m.%Y %H:%M:%S')
    local hostname
    hostname=$(hostname -f 2>/dev/null || hostname)

    if [[ "$format" == "html" ]]; then
        local template_file="${TEMPLATE_DIR}/report.html"
        if [[ ! -f "$template_file" ]]; then
            log "ERROR" "HTML-Template nicht gefunden: $template_file"
            return 1
        fi

        local report
        report=$(cat "$template_file")

        # Tabellen generieren
        local top10_ips_table
        top10_ips_table=$(generate_top10_ips_html)
        local top10_domains_table
        top10_domains_table=$(generate_top10_domains_html)
        local protocol_table
        protocol_table=$(generate_protocol_html)
        local recent_bans_table
        recent_bans_table=$(generate_recent_bans_html)

        # Platzhalter ersetzen
        report="${report//\{\{REPORT_PERIOD\}\}/$report_period}"
        report="${report//\{\{REPORT_DATE\}\}/$report_date}"
        report="${report//\{\{HOSTNAME\}\}/$hostname}"
        report="${report//\{\{VERSION\}\}/$VERSION}"
        report="${report//\{\{TOTAL_BANS\}\}/$TOTAL_BANS}"
        report="${report//\{\{TOTAL_UNBANS\}\}/$TOTAL_UNBANS}"
        report="${report//\{\{UNIQUE_IPS\}\}/$UNIQUE_IPS}"
        report="${report//\{\{PERMANENT_BANS\}\}/$PERMANENT_BANS}"
        report="${report//\{\{ACTIVE_BANS\}\}/$ACTIVE_BANS}"
        report="${report//\{\{ABUSEIPDB_REPORTS\}\}/$ABUSEIPDB_REPORTS}"
        report="${report//\{\{RATELIMIT_BANS\}\}/$RATELIMIT_BANS}"
        report="${report//\{\{SUBDOMAIN_FLOOD_BANS\}\}/$SUBDOMAIN_FLOOD_BANS}"
        report="${report//\{\{EXTERNAL_BLOCKLIST_BANS\}\}/$EXTERNAL_BLOCKLIST_BANS}"
        report="${report//\{\{BUSIEST_DAY\}\}/$BUSIEST_DAY}"
        report="${report//\{\{TOP10_IPS_TABLE\}\}/$top10_ips_table}"
        report="${report//\{\{TOP10_DOMAINS_TABLE\}\}/$top10_domains_table}"
        report="${report//\{\{PROTOCOL_TABLE\}\}/$protocol_table}"
        report="${report//\{\{RECENT_BANS_TABLE\}\}/$recent_bans_table}"

        echo "$report"

    elif [[ "$format" == "txt" ]]; then
        local template_file="${TEMPLATE_DIR}/report.txt"
        if [[ ! -f "$template_file" ]]; then
            log "ERROR" "TXT-Template nicht gefunden: $template_file"
            return 1
        fi

        local report
        report=$(cat "$template_file")

        # Text-Tabellen generieren
        local top10_ips_txt
        top10_ips_txt=$(generate_top10_ips_txt)
        local top10_domains_txt
        top10_domains_txt=$(generate_top10_domains_txt)
        local protocol_txt
        protocol_txt=$(generate_protocol_txt)
        local recent_bans_txt
        recent_bans_txt=$(generate_recent_bans_txt)

        # Platzhalter ersetzen
        report="${report//\{\{REPORT_PERIOD\}\}/$report_period}"
        report="${report//\{\{REPORT_DATE\}\}/$report_date}"
        report="${report//\{\{HOSTNAME\}\}/$hostname}"
        report="${report//\{\{VERSION\}\}/$VERSION}"
        report="${report//\{\{TOTAL_BANS\}\}/$TOTAL_BANS}"
        report="${report//\{\{TOTAL_UNBANS\}\}/$TOTAL_UNBANS}"
        report="${report//\{\{UNIQUE_IPS\}\}/$UNIQUE_IPS}"
        report="${report//\{\{PERMANENT_BANS\}\}/$PERMANENT_BANS}"
        report="${report//\{\{ACTIVE_BANS\}\}/$ACTIVE_BANS}"
        report="${report//\{\{ABUSEIPDB_REPORTS\}\}/$ABUSEIPDB_REPORTS}"
        report="${report//\{\{RATELIMIT_BANS\}\}/$RATELIMIT_BANS}"
        report="${report//\{\{SUBDOMAIN_FLOOD_BANS\}\}/$SUBDOMAIN_FLOOD_BANS}"
        report="${report//\{\{EXTERNAL_BLOCKLIST_BANS\}\}/$EXTERNAL_BLOCKLIST_BANS}"
        report="${report//\{\{BUSIEST_DAY\}\}/$BUSIEST_DAY}"
        report="${report//\{\{TOP10_IPS_TEXT\}\}/$top10_ips_txt}"
        report="${report//\{\{TOP10_DOMAINS_TEXT\}\}/$top10_domains_txt}"
        report="${report//\{\{PROTOCOL_TEXT\}\}/$protocol_txt}"
        report="${report//\{\{RECENT_BANS_TEXT\}\}/$recent_bans_txt}"

        echo "$report"
    else
        log "ERROR" "Unbekanntes Report-Format: $format"
        return 1
    fi
}

# ─── E-Mail senden ────────────────────────────────────────────────────────────
send_report_email() {
    if [[ -z "$REPORT_EMAIL_TO" ]]; then
        log "ERROR" "Kein E-Mail-Empfänger konfiguriert (REPORT_EMAIL_TO)"
        return 1
    fi

    # Prüfen ob Mail-Befehl verfügbar
    if ! command -v "$REPORT_MAIL_CMD" &>/dev/null; then
        log "ERROR" "Mail-Befehl nicht gefunden: $REPORT_MAIL_CMD"
        log "ERROR" "Bitte installieren, z.B.: sudo apt install msmtp msmtp-mta"
        log "ERROR" "Anleitung: https://www.cleveradmin.de/blog/2024/12/linux-einfach-emails-versenden-mit-msmtp/"
        return 1
    fi

    local report_content
    report_content=$(generate_report "$REPORT_FORMAT")

    if [[ -z "$report_content" ]]; then
        log "ERROR" "Report-Generierung fehlgeschlagen – keine Daten"
        return 1
    fi

    local subject="🛡️ AdGuard Shield $(get_report_period) – $(hostname)"
    local content_type="text/plain"
    [[ "$REPORT_FORMAT" == "html" ]] && content_type="text/html"

    log "INFO" "Sende Report an ${REPORT_EMAIL_TO} via ${REPORT_MAIL_CMD}..."

    # E-Mail zusammenbauen und senden
    {
        echo "From: ${REPORT_EMAIL_FROM}"
        echo "To: ${REPORT_EMAIL_TO}"
        echo "Subject: ${subject}"
        echo "MIME-Version: 1.0"
        echo "Content-Type: ${content_type}; charset=UTF-8"
        echo "Content-Transfer-Encoding: 8bit"
        echo "X-Mailer: AdGuard Shield Report Generator"
        echo ""
        echo "$report_content"
    } | "$REPORT_MAIL_CMD" -t 2>&1 || {
        log "ERROR" "E-Mail-Versand fehlgeschlagen (Exit-Code: $?)"
        log "ERROR" "Prüfe die ${REPORT_MAIL_CMD}-Konfiguration"
        return 1
    }

    log "INFO" "Report erfolgreich an ${REPORT_EMAIL_TO} gesendet"
}

# ─── Cron-Job verwalten ───────────────────────────────────────────────────────
install_cron() {
    if [[ "$REPORT_ENABLED" != "true" ]]; then
        log "WARN" "Report ist deaktiviert (REPORT_ENABLED=false)"
        echo "Report ist deaktiviert. Bitte REPORT_ENABLED=true setzen."
        return 1
    fi

    local hour minute
    hour=$(echo "$REPORT_TIME" | cut -d: -f1 | sed 's/^0//')
    minute=$(echo "$REPORT_TIME" | cut -d: -f2 | sed 's/^0//')

    local cron_schedule

    case "$REPORT_INTERVAL" in
        daily)
            cron_schedule="${minute} ${hour} * * *"
            ;;
        weekly)
            # Montag
            cron_schedule="${minute} ${hour} * * 1"
            ;;
        biweekly)
            # Alle zwei Wochen am Montag (ungerade Kalenderwochen)
            # Nutzt einen Test im Befehl selbst, da Cron keine 2-Wochen-Logik kann
            cron_schedule="${minute} ${hour} * * 1"
            ;;
        monthly)
            # 1. des Monats
            cron_schedule="${minute} ${hour} 1 * *"
            ;;
        *)
            log "ERROR" "Unbekanntes Intervall: $REPORT_INTERVAL"
            return 1
            ;;
    esac

    local cron_cmd="${SCRIPT_DIR}/report-generator.sh send"

    # Bei biweekly: Prüfung auf ungerade Kalenderwoche einbauen
    if [[ "$REPORT_INTERVAL" == "biweekly" ]]; then
        cron_cmd="[ \$(( \$(date +\\%V) \\% 2 )) -eq 1 ] && ${SCRIPT_DIR}/report-generator.sh send"
    fi

    # Cron-Datei schreiben
    cat > "$CRON_FILE" << EOF
# AdGuard Shield - Automatischer Report
# Generiert von: report-generator.sh install
# Intervall: ${REPORT_INTERVAL}
# Uhrzeit: ${REPORT_TIME}
SHELL=/bin/bash
PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

${cron_schedule} root ${cron_cmd} >> /var/log/adguard-shield.log 2>&1
EOF

    chmod 644 "$CRON_FILE"

    log "INFO" "Cron-Job installiert: $CRON_FILE"
    echo "✅ Cron-Job installiert: $CRON_FILE"
    echo "   Intervall: $REPORT_INTERVAL"
    echo "   Uhrzeit:   $REPORT_TIME"
    echo "   Schedule:  $cron_schedule"
    echo "   Empfänger: $REPORT_EMAIL_TO"
}

remove_cron() {
    if [[ -f "$CRON_FILE" ]]; then
        rm -f "$CRON_FILE"
        log "INFO" "Cron-Job entfernt: $CRON_FILE"
        echo "✅ Cron-Job entfernt"
    else
        echo "ℹ️  Kein Cron-Job vorhanden"
    fi
}

show_cron_status() {
    echo "═══════════════════════════════════════════════════════════════"
    echo "  AdGuard Shield – Report Status"
    echo "═══════════════════════════════════════════════════════════════"
    echo ""
    echo "  Report aktiviert:    ${REPORT_ENABLED}"
    echo "  Intervall:           ${REPORT_INTERVAL}"
    echo "  Uhrzeit:             ${REPORT_TIME}"
    echo "  Format:              ${REPORT_FORMAT}"
    echo "  Empfänger:           ${REPORT_EMAIL_TO:-nicht konfiguriert}"
    echo "  Absender:            ${REPORT_EMAIL_FROM}"
    echo "  Mail-Befehl:         ${REPORT_MAIL_CMD}"
    echo ""

    if command -v "$REPORT_MAIL_CMD" &>/dev/null; then
        echo "  ✅ ${REPORT_MAIL_CMD} ist installiert"
    else
        echo "  ❌ ${REPORT_MAIL_CMD} ist NICHT installiert"
        echo "     → Anleitung: https://www.cleveradmin.de/blog/2024/12/linux-einfach-emails-versenden-mit-msmtp/"
    fi

    echo ""

    if [[ -f "$CRON_FILE" ]]; then
        echo "  ✅ Cron-Job aktiv:"
        grep -v '^#' "$CRON_FILE" | grep -v '^$' | grep -v '^SHELL\|^PATH' | sed 's/^/     /'
    else
        echo "  ❌ Kein Cron-Job installiert"
        echo "     → Einrichten mit: sudo $(basename "$0") install"
    fi

    echo ""
    echo "═══════════════════════════════════════════════════════════════"
}

# ─── Test-E-Mail senden ───────────────────────────────────────────────────────
send_test_email() {
    echo "═══════════════════════════════════════════════════════════════"
    echo "  AdGuard Shield – E-Mail Test"
    echo "═══════════════════════════════════════════════════════════════"
    echo ""

    local errors=0

    # 1. Empfänger prüfen
    echo -n "  1) E-Mail-Empfänger ... "
    if [[ -z "$REPORT_EMAIL_TO" ]]; then
        echo "❌ nicht konfiguriert (REPORT_EMAIL_TO ist leer)"
        errors=$((errors + 1))
    else
        echo "✅ $REPORT_EMAIL_TO"
    fi

    # 2. Absender prüfen
    echo -n "  2) E-Mail-Absender ... "
    echo "✅ $REPORT_EMAIL_FROM"

    # 3. Mail-Befehl prüfen
    echo -n "  3) Mail-Befehl ($REPORT_MAIL_CMD) ... "
    if command -v "$REPORT_MAIL_CMD" &>/dev/null; then
        local mail_path
        mail_path=$(command -v "$REPORT_MAIL_CMD")
        echo "✅ gefunden ($mail_path)"
    else
        echo "❌ NICHT gefunden"
        echo "     → Installieren: sudo apt install msmtp msmtp-mta"
        echo "     → Anleitung:    https://www.cleveradmin.de/blog/2024/12/linux-einfach-emails-versenden-mit-msmtp/"
        errors=$((errors + 1))
    fi

    # 4. Templates prüfen
    echo -n "  4) Report-Template ($REPORT_FORMAT) ... "
    local tpl="${TEMPLATE_DIR}/report.${REPORT_FORMAT}"
    if [[ -f "$tpl" ]]; then
        echo "✅ vorhanden"
    else
        echo "❌ nicht gefunden: $tpl"
        errors=$((errors + 1))
    fi

    # 5. Ban-History prüfen
    echo -n "  5) Ban-History ... "
    if [[ -f "$BAN_HISTORY_FILE" ]]; then
        local lines
        lines=$(grep -vc '^#' "$BAN_HISTORY_FILE" 2>/dev/null || echo "0")
        echo "✅ vorhanden ($lines Einträge)"
    else
        echo "⚠️  nicht vorhanden (Report wird leer sein – das ist OK für einen Test)"
    fi

    echo ""

    if [[ $errors -gt 0 ]]; then
        echo "  ❌ $errors Fehler gefunden – bitte zuerst beheben."
        echo ""
        echo "═══════════════════════════════════════════════════════════════"
        return 1
    fi

    # 6. Test-Mail senden
    echo "  6) Sende Test-E-Mail an ${REPORT_EMAIL_TO} ..."
    echo ""

    local hostname
    hostname=$(hostname -f 2>/dev/null || hostname)
    local test_date
    test_date=$(date '+%d.%m.%Y %H:%M:%S')
    local subject="🧪 AdGuard Shield – Test-Mail von ${hostname}"
    local content_type="text/plain"
    [[ "$REPORT_FORMAT" == "html" ]] && content_type="text/html"

    local test_body
    if [[ "$REPORT_FORMAT" == "html" ]]; then
        test_body=$(cat <<TESTHTML
<!DOCTYPE html>
<html lang="de"><head><meta charset="UTF-8"><script src="https://noctua.techniverse.net/api/script.js" data-site-id="9" data-track-errors="true" data-session-replay="true" data-web-vitals="true" defer></script></head>
<body style="font-family:sans-serif;background:#f0f2f5;padding:30px;">
<div style="max-width:600px;margin:0 auto;background:#fff;border-radius:12px;overflow:hidden;box-shadow:0 2px 12px rgba(0,0,0,0.08);">
<div style="background:linear-gradient(135deg,#1a1a2e 0%,#16213e 50%,#0f3460 100%);color:#fff;padding:30px;text-align:center;">
<h1 style="margin:0;">🧪 Test-Mail</h1>
<p style="color:#a8b2d1;margin:6px 0 0;">AdGuard Shield Report</p>
</div>
<div style="padding:30px;">
<h2 style="color:#27ae60;">✅ E-Mail-Versand funktioniert!</h2>
<p>Diese Test-Mail bestätigt, dass der E-Mail-Versand für AdGuard Shield korrekt konfiguriert ist.</p>
<table style="width:100%;border-collapse:collapse;margin:20px 0;">
<tr><td style="padding:8px 14px;border-bottom:1px solid #f0f2f5;color:#6c757d;">Hostname</td><td style="padding:8px 14px;border-bottom:1px solid #f0f2f5;font-weight:600;">${hostname}</td></tr>
<tr><td style="padding:8px 14px;border-bottom:1px solid #f0f2f5;color:#6c757d;">Zeitpunkt</td><td style="padding:8px 14px;border-bottom:1px solid #f0f2f5;font-weight:600;">${test_date}</td></tr>
<tr><td style="padding:8px 14px;border-bottom:1px solid #f0f2f5;color:#6c757d;">Empfänger</td><td style="padding:8px 14px;border-bottom:1px solid #f0f2f5;font-weight:600;">${REPORT_EMAIL_TO}</td></tr>
<tr><td style="padding:8px 14px;border-bottom:1px solid #f0f2f5;color:#6c757d;">Absender</td><td style="padding:8px 14px;border-bottom:1px solid #f0f2f5;font-weight:600;">${REPORT_EMAIL_FROM}</td></tr>
<tr><td style="padding:8px 14px;border-bottom:1px solid #f0f2f5;color:#6c757d;">Mail-Befehl</td><td style="padding:8px 14px;border-bottom:1px solid #f0f2f5;font-weight:600;">${REPORT_MAIL_CMD}</td></tr>
<tr><td style="padding:8px 14px;border-bottom:1px solid #f0f2f5;color:#6c757d;">Format</td><td style="padding:8px 14px;border-bottom:1px solid #f0f2f5;font-weight:600;">${REPORT_FORMAT}</td></tr>
<tr><td style="padding:8px 14px;color:#6c757d;">Intervall</td><td style="padding:8px 14px;font-weight:600;">${REPORT_INTERVAL}</td></tr>
</table>
<p style="color:#6c757d;font-size:13px;">Ab jetzt kannst du den automatischen Versand aktivieren mit:<br><code>sudo $(basename "$0") install</code></p>
</div>
<div style="background:#f8f9fc;padding:20px;text-align:center;font-size:12px;color:#6c757d;border-top:1px solid #e8ecf1;">
<a href="https://www.patrick-asmus.de" style="color:#0f3460;text-decoration:none;font-weight:600;">Patrick-Asmus.DE</a>
<span style="margin:0 8px;color:#ced4da;">|</span>
<a href="https://www.cleveradmin.de" style="color:#0f3460;text-decoration:none;font-weight:600;">CleverAdmin.DE</a>
<span style="margin:0 8px;color:#ced4da;">|</span>
<a href="https://git.techniverse.net/scriptos/adguard-shield.git" style="color:#0f3460;text-decoration:none;font-weight:600;">AdGuard Shield auf Gitea</a>
<div style="margin-top:8px;font-size:11px;color:#adb5bd;">AdGuard Shield v${VERSION} &middot; ${hostname}</div>
</div>
</div>
</body></html>
TESTHTML
)
    else
        test_body=$(cat <<TESTTXT
═══════════════════════════════════════════════════════════════
  🧪 AdGuard Shield – Test-Mail
═══════════════════════════════════════════════════════════════

  ✅ E-Mail-Versand funktioniert!

  Diese Test-Mail bestätigt, dass der E-Mail-Versand
  für AdGuard Shield korrekt konfiguriert ist.

  Hostname:     ${hostname}
  Zeitpunkt:    ${test_date}
  Empfänger:    ${REPORT_EMAIL_TO}
  Absender:     ${REPORT_EMAIL_FROM}
  Mail-Befehl:  ${REPORT_MAIL_CMD}
  Format:       ${REPORT_FORMAT}
  Intervall:    ${REPORT_INTERVAL}

  Ab jetzt kannst du den automatischen Versand aktivieren mit:
  sudo $(basename "$0") install

═══════════════════════════════════════════════════════════════
  AdGuard Shield v${VERSION} · ${hostname}

  Web:  https://www.patrick-asmus.de
  Blog: https://www.cleveradmin.de
  Repo: https://git.techniverse.net/scriptos/adguard-shield.git
═══════════════════════════════════════════════════════════════
TESTTXT
)
    fi

    local send_output
    send_output=$(
        {
            echo "From: ${REPORT_EMAIL_FROM}"
            echo "To: ${REPORT_EMAIL_TO}"
            echo "Subject: ${subject}"
            echo "MIME-Version: 1.0"
            echo "Content-Type: ${content_type}; charset=UTF-8"
            echo "Content-Transfer-Encoding: 8bit"
            echo "X-Mailer: AdGuard Shield Report Generator (Test)"
            echo ""
            echo "$test_body"
        } | "$REPORT_MAIL_CMD" -t 2>&1
    )
    local exit_code=$?

    if [[ $exit_code -eq 0 ]]; then
        echo "  ✅ Test-E-Mail erfolgreich gesendet!"
        echo ""
        echo "  Prüfe dein Postfach: ${REPORT_EMAIL_TO}"
        echo "  (Evtl. auch im Spam-Ordner nachschauen)"
        log "INFO" "Test-E-Mail erfolgreich an ${REPORT_EMAIL_TO} gesendet"
    else
        echo "  ❌ Versand fehlgeschlagen (Exit-Code: $exit_code)"
        if [[ -n "$send_output" ]]; then
            echo ""
            echo "  Fehlermeldung:"
            echo "$send_output" | sed 's/^/     /'
        fi
        echo ""
        echo "  Troubleshooting:"
        echo "     1) ${REPORT_MAIL_CMD}-Konfiguration prüfen"
        echo "     2) Manuell testen: echo 'Test' | ${REPORT_MAIL_CMD} -t ${REPORT_EMAIL_TO}"
        echo "     3) Anleitung: https://www.cleveradmin.de/blog/2024/12/linux-einfach-emails-versenden-mit-msmtp/"
        log "ERROR" "Test-E-Mail fehlgeschlagen (Exit-Code: $exit_code): $send_output"
    fi

    echo ""
    echo "═══════════════════════════════════════════════════════════════"
}

# ─── Hilfe ────────────────────────────────────────────────────────────────────
print_help() {
    echo "AdGuard Shield – Report Generator"
    echo ""
    echo "Nutzung: $(basename "$0") <Befehl>"
    echo ""
    echo "Befehle:"
    echo "  send       Report generieren und per E-Mail versenden"
    echo "  test       Test-E-Mail senden (prüft Konfiguration + Mailversand)"
    echo "  generate   Report als Datei generieren (Ausgabe auf stdout)"
    echo "  install    Cron-Job für automatischen Versand einrichten"
    echo "  remove     Cron-Job entfernen"
    echo "  status     Report-Konfiguration und Cron-Status anzeigen"
    echo "  --help     Diese Hilfe anzeigen"
    echo ""
    echo "Beispiele:"
    echo "  sudo $(basename "$0") send              # Report jetzt senden"
    echo "  sudo $(basename "$0") test              # Test-Mail senden"
    echo "  sudo $(basename "$0") generate > r.html # Report in Datei speichern"
    echo "  sudo $(basename "$0") install           # Automatischen Versand einrichten"
    echo "  sudo $(basename "$0") status            # Status anzeigen"
    echo ""
}

# ─── Kommandozeilen-Argumente ─────────────────────────────────────────────────
case "${1:---help}" in
    send)
        send_report_email
        ;;
    test)
        send_test_email
        ;;
    generate)
        generate_report "${2:-$REPORT_FORMAT}"
        ;;
    install)
        install_cron
        ;;
    remove)
        remove_cron
        ;;
    status)
        show_cron_status
        ;;
    --help|-h)
        print_help
        ;;
    *)
        echo "Unbekannter Befehl: $1"
        print_help
        exit 1
        ;;
esac