#!/bin/bash

# --- Script Menu Refresh Sistem dengan Auto-Scheduler & Status Checker (Versi Perbaikan) ---
# Nama: fresh
# Fungsi: Menu interaktif untuk merefresh CPU, RAM, Swap, penjadwalan, dan status
# Author: AI Assistant
# -------------------------------------------------------------------------

# Fungsi untuk menjalankan refresh otomatis (tanpa interaksi)
run_auto_refresh() {
    echo "$(date): Memulai auto-refresh..." >> /var/log/fresh-auto.log
    sync
    echo 3 > /proc/sys/vm/drop_caches
    swapoff -a
    swapon -a
    echo "$(date): Auto-refresh selesai." >> /var/log/fresh-auto.log
}

# Cek argumen --auto untuk mode otomatis
if [ "$1" == "--auto" ]; then
    run_auto_refresh
    exit 0
fi

# Cek apakah script dijalankan dengan sudo (untuk mode interaktif)
if [ "$(id -u)" -ne 0 ]; then
   echo -e "\e[31mError: Script ini harus dijalankan dengan hak akses root (sudo).\e[0m" >&2
   exit 1
fi

# Fungsi untuk menampilkan menu utama
tampilkan_menu() {
    clear
    echo -e "\e[36m========================================\e[0m"
    echo -e "\e[1;32m        MENU REFRESH SISTEM VPS       \e[0m"
    echo -e "\e[36m========================================\e[0m"
    echo "Pilih tindakan yang ingin dilakukan:"
    echo "1. Kelola Proses CPU (buka htop)"
    echo "2. Refresh RAM (Hapus Cache)"
    echo "3. Refresh Swap"
    echo "4. Refresh Semua (RAM + Swap)"
    echo "5. Lihat Status Memori Saat Ini"
    echo "6. Jadwalkan Auto Refresh"
    echo "7. Cek Status Auto Refresh"
    echo "0. Keluar"
    echo -e "\e[36m----------------------------------------\e[0m"
}

# Fungsi untuk menu penjadwalan
menu_scheduler() {
    while true; do
        clear
        echo -e "\e[36m========================================\e[0m"
        echo -e "\e[1;34m       MENU JADWAL AUTO REFRESH       \e[0m"
        echo -e "\e[36m========================================\e[0m"
        echo "Pilih interval waktu:"
        echo "1. Setiap 1 Jam"
        echo "2. Setiap 3 Jam"
        echo "3. Setiap 6 Jam"
        echo "4. Setiap 12 Jam"
        echo "5. Setiap 24 Jam (1 Hari)"
        echo "6. Setiap 1 Minggu"
        echo "7. Setiap 2 Minggu"
        echo "8. Setiap 1 Bulan"
        echo "9. Hapus Semua Jadwal"
        echo "0. Kembali ke Menu Utama"
        echo -e "\e[36m----------------------------------------\e[0m"
        read -p "Masukkan pilihan Anda: " pilihan_sched

        case $pilihan_sched in
            1) cron_interval="0 * * * *"; break ;; # Setiap jam penuh
            2) cron_interval="0 */3 * * *"; break ;; # Setiap 3 jam
            3) cron_interval="0 */6 * * *"; break ;; # Setiap 6 jam
            4) cron_interval="0 */12 * * *"; break ;; # Setiap 12 jam
            5) cron_interval="0 0 * * *"; break ;; # Setiap hari pukul 00:00
            6) cron_interval="0 0 * * 0"; break ;; # Setiap Minggu pukul 00:00
            7) cron_interval="0 0 */14 * *"; break ;; # Setiap 14 hari (2 minggu)
            8) cron_interval="0 0 1 * *"; break ;; # Setiap bulan pukul 00:00
            9)
                echo "Menghapus semua jadwal 'fresh --auto'..."
                # Hapus cron job dengan cara yang lebih aman
                crontab -l 2>/dev/null | grep -v "fresh --auto" | crontab -
                echo -e "\e[32mSemua jadwal berhasil dihapus.\e[0m"
                pause
                return
                ;;
            0) return ;;
            *) 
                echo -e "\e[31mPilihan tidak valid.\e[0m"
                pause
                ;;
        esac
    done

    # Validasi format cron
    if ! echo "$cron_interval" | grep -qE '^([0-5]?[0-9]|\*) [0-2]?[0-9] \* \* \*(|\*)$'; then
        echo -e "\e[31mFormat waktu tidak valid!\e[0m"
        pause
        return
    fi

    # Tambahkan cron job dengan metode yang lebih aman
    echo "Menambahkan jadwal baru ke crontab..."
    
    # Hapus cron job lama (jika ada) dan tambahkan yang baru
    (crontab -l 2>/dev/null | grep -v "fresh --auto") | crontab -
    echo "$cron_interval sudo /usr/local/bin/fresh --auto >> /var/log/fresh-auto.log 2>&1" | crontab -
    
    # Verifikasi apakah cron job berhasil ditambahkan
    if crontab -l 2>/dev/null | grep -q "fresh --auto"; then
        echo -e "\e[32mJadwal berhasil dibuat!\e[0m"
        echo "Log akan disimpan di /var/log/fresh-auto.log"
    else
        echo -e "\e[31mGagal membuat jadwal. Periksa format cron.\e[0m"
    fi
    
    pause
}

# Fungsi untuk mengecek status auto-refresh
cek_status_auto_refresh() {
    clear
    echo -e "\e[36m========================================\e[0m"
    echo -e "\e[1;35m       STATUS AUTO REFRESH            \e[0m"
    echo -e "\e[36m========================================\e[0m"
    
    # Cek cron job dengan pengecekan yang lebih akurat
    if crontab -l 2>/dev/null | grep -q "fresh --auto.*>> /var/log/fresh-auto.log"; then
        echo -e "\e[32mStatus: AKTIF\e[0m"
        echo ""
        echo "Jadwal yang sedang berjalan:"
        crontab -l 2>/dev/null | grep --color=auto "fresh --auto"
    else
        echo -e "\e[31mStatus: TIDAK AKTIF\e[0m"
        echo ""
        echo "Tidak ada jadwal auto-refresh yang ditemukan."
    fi
    echo ""
    echo -e "\e[33mLog dapat dilihat di: /var/log/fresh-auto.log\e[0m"
    echo -e "\e[36m----------------------------------------\e[0m"
    pause
}

# Fungsi untuk pause
pause() {
    read -p "Tekan [Enter] untuk melanjutkan..."
}

# Loop utama program
while true; do
    tampilkan_menu
    read -p "Masukkan pilihan Anda [1-7] atau 0 untuk keluar: " pilihan

    case $pilihan in
        1)
            echo -e "\e[33mMembuka htop untuk mengelola proses CPU...\e[0m"
            pause
            htop
            ;;
        2)
            echo -e "\e[33mMembersihkan cache RAM...\e[0m"
            sync && echo 3 > /proc/sys/vm/drop_caches
            echo -e "\e[32mCache RAM berhasil dibersihkan.\e[0m"
            pause
            ;;
        3)
            echo -e "\e[31mPERINGATAN: Pastikan RAM Anda mencukupi!\e[0m"
            free -h
            read -p "Apakah Anda yakin ingin melanjutkan? (y/n) " konfirmasi
            if [[ $konfirmasi =~ ^[Yy]$ ]]; then
                swapoff -a && swapon -a
                echo -e "\e[32mSwap berhasil direfresh.\e[0m"
            else
                echo "Operasi dibatalkan."
            fi
            pause
            ;;
        4)
            echo -e "\e[31mPERINGATAN: Ini akan membersihkan cache RAM dan merefresh Swap.\e[0m"
            free -h
            read -p "Apakah Anda yakin ingin melanjutkan? (y/n) " konfirmasi
            if [[ $konfirmasi =~ ^[Yy]$ ]]; then
                run_auto_refresh
                echo -e "\e[32mSemua proses refresh selesai!\e[0m"
            else
                echo "Operasi dibatalkan."
            fi
            pause
            ;;
        5)
            echo -e "\e[33mStatus Memori Saat Ini:\e[0m"
            free -h
            echo ""
            echo -e "\e[33mPenggunaan Swap:\e[0m"
            swapon --show
            pause
            ;;
        6)
            menu_scheduler
            ;;
        7)
            cek_status_auto_refresh
            ;;
        0)
            echo -e "\e[32mTerima kasih! Sampai jumpa.\e[0m"
            exit 0
            ;;
        *)
            echo -e "\e[31mPilihan tidak valid, silakan coba lagi.\e[0m"
            pause
            ;;
    esac
done

#!/bin/bash

# =================================================================
# --- KONFIGURASI PATH ---
# =================================================================
FLAG_FILE="/opt/udp-proxy/log/enforcer.enabled"
STATE_FILE="/opt/udp-proxy/log/enforcer_state.log"
BANDWIDTH_LOG="/opt/udp-proxy/log/bandwidth_usage.log"
DAILY_LEDGER="/opt/udp-proxy/log/bw_daily_ledger.log"
DAILY_LOG_DIR="/opt/udp-proxy/log/daily"
IPTABLES_CHAIN="UDP_ENFORCER"
BANDWIDTH_CHAIN="UDP_BW_COUNTER"
SERVICE_NAME="udp-enforcer.service"

# Warna untuk tampilan
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
CYAN='\033[0;36m'
NC='\033[0m' # No Color
BOLD='\033[1m'

# =================================================================
# --- FUNGSI HELPER ---
# =================================================================
get_status_icon() {
    if sudo systemctl is-active --quiet "$SERVICE_NAME" 2>/dev/null; then
        echo -e "${GREEN}[AKTIF]${NC}"
    else
        echo -e "${RED}[NONAKTIF]${NC}"
    fi
}

get_blocked_count() {
    # COLUMNS=9999 mencegah wrapping, dan hilangkan || echo "0" karena grep -c sudah default 0
    COLUMNS=9999 sudo iptables -L "$IPTABLES_CHAIN" -n -v 2>/dev/null | grep -c "DROP"
}

get_online_count() {
    if [ -f "$STATE_FILE" ]; then
        grep -c ":" "$STATE_FILE" 2>/dev/null || echo "0"
    else
        echo "0"
    fi
}

# =================================================================
# --- FUNGSI MENU ---
# =================================================================
show_menu() {
    clear
    local status_icon=$(get_status_icon)
    local blocked_count=$(get_blocked_count)
    local online_count=$(get_online_count)
    
    echo -e "${CYAN}==========================================${NC}"
    echo -e "${BOLD}     PANEL PENJAGA UDP - HUNTER${NC}"
    echo -e "${CYAN}==========================================${NC}"
    echo -e " Status Layanan : $status_icon"
    echo -e " User Online    : ${GREEN}$online_count${NC}"
    echo -e " IP Diblokir    : ${RED}$blocked_count${NC}"
    echo -e "${CYAN}==========================================${NC}"
    echo " 1. Aktifkan Penjaga UDP"
    echo " 2. Nonaktifkan Penjaga UDP"
    echo " 3. Restart Penjaga UDP"
    echo -e "${CYAN}------------------------------------------${NC}"
    echo " 4. Lihat Status Detail"
    echo " 5. Pantau Log Aktivitas (Live)"
    echo -e "${CYAN}------------------------------------------${NC}"
    echo " 6. Lihat User Sedang Online"
    echo " 7. Lihat IP yang Diblokir (+User)"
    echo " 8. Bersihkan Semua IP Diblokir"
    echo -e "${CYAN}------------------------------------------${NC}"
    echo " 9. Monitor Bandwidth (5 mnt & 1 Jam)"
    echo "10. Daftar User Hari Ini (24 Jam)"
    echo "11. Lihat Riwayat Laporan"
    echo -e "${CYAN}------------------------------------------${NC}"
    echo "12. Reset Data Bandwidth Hari Ini"
    echo "13. Keluar"
    echo -e "${CYAN}==========================================${NC}"
}

# =================================================================
# --- MENU 1: AKTIFKAN ---
# =================================================================
menu_enable() {
    echo -e "${YELLOW}>> Mengaktifkan Penjaga...${NC}"
    
    if [ -f "/opt/udp-proxy/panel/enable.sh" ]; then
        sudo /opt/udp-proxy/panel/enable.sh
    else
        mkdir -p "$(dirname "$FLAG_FILE")"
        touch "$FLAG_FILE"
        sudo systemctl start "$SERVICE_NAME" 2>/dev/null
        sudo systemctl enable "$SERVICE_NAME" 2>/dev/null
        echo -e "${GREEN}>> Penjaga telah diaktifkan (manual)${NC}"
    fi
    
    sleep 1
}

# =================================================================
# --- MENU 2: NONAKTIFKAN ---
# =================================================================
menu_disable() {
    echo -e "${YELLOW}>> Menonaktifkan Penjaga...${NC}"
    
    if [ -f "/opt/udp-proxy/panel/disable.sh" ]; then
        sudo /opt/udp-proxy/panel/disable.sh
    else
        rm -f "$FLAG_FILE"
        sudo systemctl stop "$SERVICE_NAME" 2>/dev/null
        echo -e "${GREEN}>> Penjaga telah dinonaktifkan (manual)${NC}"
    fi
    
    sleep 1
}

# =================================================================
# --- MENU 3: RESTART ---
# =================================================================
menu_restart() {
    echo -e "${YELLOW}>> Merestart Penjaga...${NC}"
    sudo systemctl restart "$SERVICE_NAME"
    sleep 2
    if sudo systemctl is-active --quiet "$SERVICE_NAME"; then
        echo -e "${GREEN}>> Penjaga berhasil direstart${NC}"
    else
        echo -e "${RED}>> Gagal merestart penjaga! Cek log untuk detail.${NC}"
    fi
}

# =================================================================
# --- MENU 4: STATUS DETAIL ---
# =================================================================
menu_status() {
    echo -e "${CYAN}>> Status Detail Layanan Penjaga:${NC}"
    echo "=============================================="
    sudo systemctl status "$SERVICE_NAME" --no-pager -l
    echo ""
    echo "=============================================="
    echo -e "${CYAN}>> Status Chain IPTables:${NC}"
    echo "--- Chain UDP_ENFORCER (Blokir) ---"
    # Gunakan COLUMNS=9999 agar komentar tidak terputus ke baris baru
    COLUMNS=9999 sudo iptables -L "$IPTABLES_CHAIN" -n -v --line-numbers 2>/dev/null | head -20
    echo ""
    echo "--- Chain UDP_BW_COUNTER (Bandwidth) ---"
    local bw_rules=$(sudo iptables -L "$BANDWIDTH_CHAIN" -n 2>/dev/null | grep -c "user:")
    echo "Jumlah aturan counter: $bw_rules"
    echo "=============================================="
}

# =================================================================
# --- MENU 5: LIVE LOG ---
# =================================================================
menu_live_log() {
    echo -e "${CYAN}>> Menampilkan log aktivitas (tekan Ctrl+C untuk kembali)${NC}"
    echo "=============================================="
    sudo journalctl -u "$SERVICE_NAME" -f --no-tail
}

# =================================================================
# --- MENU 6: USER ONLINE ---
# =================================================================
menu_online_users() {
    echo -e "${CYAN}>> Daftar User Sedang Online:${NC}"
    echo "=============================================="
    
    if [ -f "$STATE_FILE" ] && [ -s "$STATE_FILE" ]; then
        local current_time=$(date +%s)
        printf "%-15s %-18s %15s %12s\n" "USER" "IP" "DURASI" "STATUS"
        echo "----------------------------------------------"
        
        while IFS=':' read -r user ip last_seen violation_time; do
            if [[ -z "$user" || -z "$ip" || -z "$last_seen" ]]; then continue; fi
            
            local elapsed=$((current_time - last_seen))
            local hours=$((elapsed / 3600))
            local mins=$(((elapsed % 3600) / 60))
            local duration=""
            
            if [ "$hours" -gt 0 ]; then
                duration="${hours}j ${mins}m"
            else
                duration="${mins}m"
            fi
            
            local status="Normal"
            if [[ -n "$violation_time" && "$violation_time" =~ ^[0-9]+$ ]]; then
                status="${RED}Pelanggaran${NC}"
            fi
            
            printf "%-15s %-18s %15s %12s\n" "$user" "$ip" "$duration" "$status"
        done < "$STATE_FILE"
    else
        echo -e "${YELLOW}>> Tidak ada user yang sedang online.${NC}"
    fi
    echo "=============================================="
}

# =================================================================
# --- MENU 7: IP DIBLOKIR (+ USER) ---
# =================================================================
menu_blocked_ips() {
    echo -e "${CYAN}>> Daftar IP yang Saat Ini Diblokir:${NC}"
    echo "=============================================================="
    
    # COLUMNS=9999 mencegah komentar iptables terputus ke baris baru
    local rules=$(COLUMNS=9999 sudo iptables -L "$IPTABLES_CHAIN" -n -v --line-numbers 2>/dev/null)
    
    if [ -z "$rules" ]; then
        echo -e "${YELLOW}>> Chain iptables tidak ditemukan atau kosong.${NC}"
        echo "=============================================================="
        return
    fi
    
    # Hapus || echo "0" karena grep -c sudah otomatis mengembalikan 0
    # Hapus tanda ^ pada "^DROP" karena baris dimulai dari angka urutan
    local drop_count=$(echo "$rules" | grep -c "DROP")
    
    if [ "$drop_count" -eq 0 ]; then
        echo -e "${GREEN}>> Tidak ada IP yang diblokir saat ini.${NC}"
    else
        printf "%-6s %-18s %-15s %-30s\n" "NO" "IP" "PACKETS" "KOMENTAR"
        echo "--------------------------------------------------------------"
        echo "$rules" | grep "DROP" | while read -r num pkts bytes target prot opt in out source dest extra; do
            local ip="$source"
            local comment=$(echo "$extra" | grep -oP 'Kicked \K[^ ]+' 2>/dev/null || echo "-")
            local waktu=$(echo "$extra" | grep -oP 'on \K[0-9-]+-[0-9:]+' 2>/dev/null || echo "-")
            printf "%-6s %-18s %-15s User: %-15s Waktu: %s\n" "$num" "$ip" "$pkts" "$comment" "$waktu"
        done
    fi
    echo "=============================================================="
    echo -e "Total: ${RED}$drop_count IP${NC} diblokir"
}

# =================================================================
# --- MENU 8: BERSIHKAN BLOKIR ---
# =================================================================
menu_clear_blocked() {
    local drop_count=$(get_blocked_count)
    
    if [ "$drop_count" -eq 0 ]; then
        echo -e "${GREEN}>> Tidak ada IP yang perlu dibersihkan.${NC}"
        return
    fi
    
    echo -e "${RED}==============================================${NC}"
    echo -e "${RED}PERINGATAN!${NC}"
    echo -e "Anda akan membersihkan ${BOLD}$drop_count IP${NC} yang diblokir."
    echo -e "Ini juga akan membersihkan file state dan lock."
    echo -e "${RED}==============================================${NC}"
    
    read -p "Yakin ingin melanjutkan? (y/N): " confirm
    
    if [[ "$confirm" =~ ^[Yy]$ ]]; then
        echo -e "${YELLOW}>> Membersihkan aturan iptables...${NC}"
        sudo iptables -F "$IPTABLES_CHAIN" 2>/dev/null
        
        echo -e "${YELLOW}>> Membersihkan state file...${NC}"
        > "$STATE_FILE"
        
        echo -e "${YELLOW}>> Membersihkan lock files...${NC}"
        rm -f /opt/udp-proxy/log/*.lock
        
        echo -e "${GREEN}>> Berhasil! Semua IP telah dibebaskan.${NC}"
    else
        echo -e "${YELLOW}>> Dibatalkan.${NC}"
    fi
}

# =================================================================
# --- MENU 9: MONITOR BANDWIDTH ---
# =================================================================
menu_bandwidth() {
    echo -e "${CYAN}>> Laporan Bandwidth Akumulasi Per User:${NC}"
    echo "--------------------------------------------------------------"
    
    if [ -f "$BANDWIDTH_LOG" ] && [ -s "$BANDWIDTH_LOG" ]; then
        cat "$BANDWIDTH_LOG"
        echo "--------------------------------------------------------------"
        echo -e "${YELLOW}Catatan:${NC}"
        echo "- Data 5 menit: Pemakaian dalam 5 menit terakhir"
        echo "- Data 1 jam: Pemakaian dalam 1 jam terakhir"
        echo "- Avg Mbps: Rata-rata kecepatan dari data 5 menit"
        echo "- User offline > 1 jam akan otomatis hilang dari sini"
        echo ""
        echo -e "${GREEN}Gunakan Menu 10 untuk melihat total harian yang TIDAK PERNAH HILANG.${NC}"
    else
        echo -e "${YELLOW}>> File log bandwidth belum tersedia atau kosong.${NC}"
        echo ">> Tunggu beberapa menit setelah user online."
    fi
    echo "--------------------------------------------------------------"
}

# =================================================================
# --- MENU 10: DAFTAR USER HARI INI ---
# =================================================================
menu_daily_users() {
    echo -e "${CYAN}>> Daftar Total Pemakaian Semua User (Hari Ini)${NC}"
    echo "--------------------------------------------------------------"
    
    if [ -f "$DAILY_LEDGER" ] && [ -s "$DAILY_LEDGER" ]; then
        local total_bytes=$(awk -F':' '{sum+=$2} END {print sum}' "$DAILY_LEDGER")
        local total_mb=$(echo "scale=2; $total_bytes / 1048576" | bc 2>/dev/null || echo "0.00")
        local user_count=$(wc -l < "$DAILY_LEDGER")
        
        printf "%-15s %18s %12s\n" "USER" "TOTAL HARI INI (MB)" "PERSEN"
        echo "--------------------------------------------------------------"
        awk -F':' -v total="$total_bytes" '{
            mb = $2/1048576;
            pct = ($2/total)*100;
            printf "%-15s %18.2f MB %10.1f%%\n", $1, mb, pct
        }' "$DAILY_LEDGER" | sort -k2 -rn
        echo "--------------------------------------------------------------"
        echo -e "Total: ${BOLD}$total_mb MB${NC} dari ${BOLD}$user_count user${NC}"
        echo "--------------------------------------------------------------"
        echo -e "${YELLOW}* Semua user yang pernah terhubung hari ini tetap ada${NC}"
        echo -e "${YELLOW}* Daftar ini direset otomatis tepat jam 00:00 malam${NC}"
    else
        echo -e "${YELLOW}>> Belum ada user yang terhubung hari ini.${NC}"
    fi
}

# =================================================================
# --- MENU 11: RIWAYAT LAPORAN ---
# =================================================================
menu_history() {
    echo -e "${CYAN}>> Riwayat Laporan Harian${NC}"
    echo "--------------------------------------------------------------"
    
    if [ -d "$DAILY_LOG_DIR" ] && ls "$DAILY_LOG_DIR"/*.log 1>/dev/null 2>&1; then
        echo "Tanggal yang tersedia:"
        echo "--------------------------------------------------------------"
        ls -1 "$DAILY_LOG_DIR"/*.log 2>/dev/null | while read -r file; do
            local date_name=$(basename "$file" .log)
            local size=$(du -h "$file" | cut -f1)
            local user_count=$(wc -l < "$file")
            echo "  - $date_name ($size, $user_count user)"
        done
        echo "--------------------------------------------------------------"
    else
        echo -e "${YELLOW}>> Belum ada riwayat laporan tersimpan.${NC}"
        return
    fi
    
    echo ""
    read -p "Masukkan tanggal (Format: YYYY-MM-DD) atau tekan Enter untuk batal: " req_date
    
    if [[ "$req_date" =~ ^[0-9]{4}-[0-9]{2}-[0-9]{2}$ ]]; then
        local hist_file="$DAILY_LOG_DIR/${req_date}.log"
        
        if [ -f "$hist_file" ] && [ -s "$hist_file" ]; then
            local total_bytes=$(awk -F':' '{sum+=$2} END {print sum}' "$hist_file")
            local total_mb=$(echo "scale=2; $total_bytes / 1048576" | bc 2>/dev/null || echo "0.00")
            
            echo ""
            echo -e "${CYAN}>> Laporan Total Semua User Pada Tanggal: $req_date${NC}"
            echo "--------------------------------------------------------------"
            printf "%-15s %18s %12s\n" "USER" "TOTAL (MB)" "PERSEN"
            echo "--------------------------------------------------------------"
            awk -F':' -v total="$total_bytes" '{
                mb = $2/1048576;
                pct = ($2/total)*100;
                printf "%-15s %18.2f MB %10.1f%%\n", $1, mb, pct
            }' "$hist_file" | sort -k2 -rn
            echo "--------------------------------------------------------------"
            echo -e "Total hari itu: ${BOLD}$total_mb MB${NC}"
            echo "--------------------------------------------------------------"
        else
            echo -e "${RED}>> Tidak ditemukan data untuk tanggal $req_date.${NC}"
        fi
    else
        echo -e "${YELLOW}>> Batal melihat riwayat.${NC}"
    fi
}

# =================================================================
# --- MENU 12: RESET DATA BANDWIDTH ---
# =================================================================
menu_reset_bandwidth() {
    echo -e "${RED}==============================================${NC}"
    echo -e "${RED}PERINGATAN!${NC}"
    echo "Anda akan mereset SEMUA data bandwidth hari ini:"
    echo "  - Data akumulasi 5 menit & 1 jam"
    echo "  - Data buku harian (Menu 10)"
    echo ""
    echo -e "${YELLOW}Data riwayat hari sebelumnya TIDAK terpengaruh.${NC}"
    echo -e "${RED}==============================================${NC}"
    
    read -p "Yakin ingin mereset? (y/N): " confirm
    
    if [[ "$confirm" =~ ^[Yy]$ ]]; then
        echo -e "${YELLOW}>> Mereset data bandwidth...${NC}"
        
        > "$BANDWIDTH_LOG"
        > "/opt/udp-proxy/log/bw_accumulator.log"
        > "/opt/udp-proxy/log/bw_history_1h.log"
        > "$DAILY_LEDGER"
        > "/opt/udp-proxy/log/bw_flush_queue.log"
        
        sudo iptables -Z "$BANDWIDTH_CHAIN" 2>/dev/null
        
        echo -e "${GREEN}>> Data bandwidth hari ini berhasil direset.${NC}"
    else
        echo -e "${YELLOW}>> Dibatalkan.${NC}"
    fi
}

# =================================================================
# --- LOOP UTAMA MENU ---
# =================================================================
while true; do
    show_menu
    read -p "Masukkan pilihan Anda [1-13]: " choice

    case $choice in
        1)  menu_enable ;;
        2)  menu_disable ;;
        3)  menu_restart ;;
        4)  menu_status ;;
        5)  menu_live_log ;;
        6)  menu_online_users ;;
        7)  menu_blocked_ips ;;
        8)  menu_clear_blocked ;;
        9)  menu_bandwidth ;;
        10) menu_daily_users ;;
        11) menu_history ;;
        12) menu_reset_bandwidth ;;
        13)
            echo -e "${CYAN}>> Keluar dari menu Hunter. Sampai jumpa!${NC}"
            exit 0
            ;;
        *)
            echo -e "${RED}>> Pilihan tidak valid, silakan coba lagi.${NC}"
            ;;
    esac

    echo ""
    read -p "Tekan [Enter] untuk melanjutkan..."
done