// NetworkDlg.cpp - 网络配置对话框实现

#include "stdafx.h"
#include "NetworkDlg.h"
#include "afxdialogex.h"
#include "2015Remote.h"

// 外部函数：刷新缓存的限流配置
extern void ReloadBanConfig();
extern void ReloadDllRateLimitConfig();

CNetworkDlg::CNetworkDlg(CWnd* pParent /*=nullptr*/)
    : CDialogLangEx(IDD_DIALOG_NETWORK, pParent)
    , m_nDllLimitSeconds(3600)
    , m_nDllLimitCount(4)
    , m_nBanWindow(60)
    , m_nBanMaxConn(15)
    , m_nBanDuration(3600)
{
}

CNetworkDlg::~CNetworkDlg()
{
}

void CNetworkDlg::DoDataExchange(CDataExchange* pDX)
{
    __super::DoDataExchange(pDX);
    DDX_Control(pDX, IDC_LIST_WHITELIST, m_ListWhitelist);
    DDX_Control(pDX, IDC_LIST_BLACKLIST, m_ListBlacklist);
    DDX_Control(pDX, IDC_EDIT_NETWORK_IP, m_EditIP);
    DDX_Text(pDX, IDC_EDIT_DLL_LIMIT_SECONDS, m_nDllLimitSeconds);
    DDX_Text(pDX, IDC_EDIT_DLL_LIMIT_COUNT, m_nDllLimitCount);
    DDX_Text(pDX, IDC_EDIT_BAN_WINDOW, m_nBanWindow);
    DDX_Text(pDX, IDC_EDIT_BAN_MAX_CONN, m_nBanMaxConn);
    DDX_Text(pDX, IDC_EDIT_BAN_DURATION, m_nBanDuration);
}

BEGIN_MESSAGE_MAP(CNetworkDlg, CDialogLangEx)
    ON_BN_CLICKED(IDC_BTN_ADD_WHITELIST, &CNetworkDlg::OnBnClickedAddWhitelist)
    ON_BN_CLICKED(IDC_BTN_DEL_WHITELIST, &CNetworkDlg::OnBnClickedDelWhitelist)
    ON_BN_CLICKED(IDC_BTN_ADD_BLACKLIST, &CNetworkDlg::OnBnClickedAddBlacklist)
    ON_BN_CLICKED(IDC_BTN_DEL_BLACKLIST, &CNetworkDlg::OnBnClickedDelBlacklist)
    ON_BN_CLICKED(IDOK, &CNetworkDlg::OnBnClickedOk)
END_MESSAGE_MAP()

BOOL CNetworkDlg::OnInitDialog()
{
    __super::OnInitDialog();

    // 多语言翻译
    SetWindowText(_TR("网络配置"));
    SetDlgItemText(IDC_STATIC_WHITELIST, _TR("白名单 (不限流):"));
    SetDlgItemText(IDC_STATIC_BLACKLIST, _TR("黑名单 (拒绝连接):"));
    SetDlgItemText(IDC_STATIC_IP_INPUT, _TR("IP 地址:"));
    SetDlgItemText(IDC_BTN_ADD_WHITELIST, _TR("添加 >>"));
    SetDlgItemText(IDC_BTN_DEL_WHITELIST, _TR("删除"));
    SetDlgItemText(IDC_BTN_ADD_BLACKLIST, _TR("添加 >>"));
    SetDlgItemText(IDC_BTN_DEL_BLACKLIST, _TR("删除"));
    SetDlgItemText(IDC_STATIC_NETWORK_HINT, _TR("说明:"));
    SetDlgItemText(IDC_STATIC_NETWORK_HINT1, _TR("- 白名单IP: 不受限流"));
    SetDlgItemText(IDC_STATIC_NETWORK_HINT2, _TR("- 黑名单IP: 拒绝请求"));
    SetDlgItemText(IDC_STATIC_NETWORK_HINT3, _TR("- 127.0.0.1 自动白名单"));
    SetDlgItemText(IDC_STATIC_NETWORK_HINT4, _TR("- 配置即时生效并保存"));
    SetDlgItemText(IDOK, _TR("确定"));
    SetDlgItemText(IDCANCEL, _TR("取消"));
    SetDlgItemText(IDC_STATIC_DLL_LIMIT, _TR("DLL限流设置:"));
    SetDlgItemText(IDC_STATIC_DLL_WINDOW, _TR("时间窗口(秒):"));
    SetDlgItemText(IDC_STATIC_DLL_COUNT, _TR("最大请求数:"));
    SetDlgItemText(IDC_STATIC_BAN_SETTINGS, _TR("IP封禁设置:"));
    SetDlgItemText(IDC_STATIC_BAN_WINDOW, _TR("统计窗口(秒):"));
    SetDlgItemText(IDC_STATIC_BAN_MAXCONN, _TR("最大连接数:"));
    SetDlgItemText(IDC_STATIC_BAN_DURATION, _TR("封禁时长(秒):"));

    // 加载现有列表和阈值
    LoadLists();
    LoadThresholds();

    return TRUE;
}

void CNetworkDlg::LoadLists()
{
    // 加载白名单
    m_ListWhitelist.ResetContent();
    auto whiteIPs = IPWhitelist::getInstance().GetAll();
    for (const auto& ip : whiteIPs) {
        m_ListWhitelist.AddString(CString(ip.c_str()));
    }

    // 加载黑名单
    m_ListBlacklist.ResetContent();
    auto blackIPs = IPBlacklist::getInstance().GetAll();
    for (const auto& ip : blackIPs) {
        m_ListBlacklist.AddString(CString(ip.c_str()));
    }
}

void CNetworkDlg::SaveLists()
{
    // 保存白名单
    std::string whitelistStr = IPWhitelist::getInstance().Export();
    THIS_CFG.SetStr("settings", "IPWhitelist", whitelistStr.c_str());

    // 保存黑名单
    std::string blacklistStr = IPBlacklist::getInstance().Export();
    THIS_CFG.SetStr("settings", "IPBlacklist", blacklistStr.c_str());
}

void CNetworkDlg::LoadThresholds()
{
    // 加载 DLL 限流设置
    m_nDllLimitSeconds = THIS_CFG.GetInt("settings", "DllLimitSeconds", 3600);
    m_nDllLimitCount = THIS_CFG.GetInt("settings", "DllLimitCount", 4);

    // 加载 IP 封禁设置
    m_nBanWindow = THIS_CFG.GetInt("settings", "BanWindow", 60);
    m_nBanMaxConn = THIS_CFG.GetInt("settings", "BanMaxConn", 15);
    m_nBanDuration = THIS_CFG.GetInt("settings", "BanDuration", 3600);

    // 更新界面
    UpdateData(FALSE);
}

void CNetworkDlg::SaveThresholds()
{
    // 保存 DLL 限流设置
    THIS_CFG.SetInt("settings", "DllLimitSeconds", m_nDllLimitSeconds);
    THIS_CFG.SetInt("settings", "DllLimitCount", m_nDllLimitCount);

    // 保存 IP 封禁设置
    THIS_CFG.SetInt("settings", "BanWindow", m_nBanWindow);
    THIS_CFG.SetInt("settings", "BanMaxConn", m_nBanMaxConn);
    THIS_CFG.SetInt("settings", "BanDuration", m_nBanDuration);
}

bool CNetworkDlg::IsValidIP(const CString& ip)
{
    if (ip.IsEmpty()) return false;

    // 简单的 IP 格式检查: xxx.xxx.xxx.xxx
    int partCount = 0;
    int partValue = 0;
    bool hasDigit = false;

    for (int i = 0; i <= ip.GetLength(); i++) {
        TCHAR c = (i < ip.GetLength()) ? ip.GetAt(i) : '\0';

        if (c >= '0' && c <= '9') {
            partValue = partValue * 10 + (c - '0');
            hasDigit = true;
            if (partValue > 255) return false;
        }
        else if (c == '.' || c == '\0') {
            if (!hasDigit) return false;  // 空段
            partCount++;
            partValue = 0;
            hasDigit = false;
            if (c == '\0') break;
        }
        else {
            return false;  // 非法字符
        }
    }

    return (partCount == 4);
}

CString CNetworkDlg::GetInputIP()
{
    CString ip;
    m_EditIP.GetWindowText(ip);
    ip.Trim();
    return ip;
}

void CNetworkDlg::OnBnClickedAddWhitelist()
{
    CString ip = GetInputIP();

    if (!IsValidIP(ip)) {
        MessageBox(_TR("请输入有效的 IP 地址"), _TR("错误"), MB_OK | MB_ICONERROR);
        return;
    }

    CStringA ipA(ip);
    std::string ipStr(ipA.GetString());

    // 检查是否已在白名单中
    if (IPWhitelist::getInstance().IsWhitelisted(ipStr)) {
        MessageBox(_TR("该 IP 已在白名单中"), _TR("提示"), MB_OK | MB_ICONINFORMATION);
        return;
    }

    // 如果在黑名单中，先移除
    if (IPBlacklist::getInstance().IsBlacklisted(ipStr)) {
        IPBlacklist::getInstance().RemoveIP(ipStr);
        // 从黑名单列表中移除
        for (int i = 0; i < m_ListBlacklist.GetCount(); i++) {
            CString item;
            m_ListBlacklist.GetText(i, item);
            if (item == ip) {
                m_ListBlacklist.DeleteString(i);
                break;
            }
        }
    }

    // 添加到白名单
    IPWhitelist::getInstance().AddIP(ipStr);
    m_ListWhitelist.AddString(ip);
    m_EditIP.SetWindowText(_T(""));
}

void CNetworkDlg::OnBnClickedDelWhitelist()
{
    int sel = m_ListWhitelist.GetCurSel();
    if (sel == LB_ERR) {
        MessageBox(_TR("请先选择要删除的 IP"), _TR("提示"), MB_OK | MB_ICONINFORMATION);
        return;
    }

    CString ip;
    m_ListWhitelist.GetText(sel, ip);
    CStringA ipA(ip);
    std::string ipStr(ipA.GetString());

    IPWhitelist::getInstance().RemoveIP(ipStr);
    m_ListWhitelist.DeleteString(sel);
}

void CNetworkDlg::OnBnClickedAddBlacklist()
{
    CString ip = GetInputIP();

    if (!IsValidIP(ip)) {
        MessageBox(_TR("请输入有效的 IP 地址"), _TR("错误"), MB_OK | MB_ICONERROR);
        return;
    }

    CStringA ipA(ip);
    std::string ipStr(ipA.GetString());

    // 检查是否是本地地址
    if (ipStr == "127.0.0.1" || ipStr == "::1") {
        MessageBox(_TR("本地地址不能加入黑名单"), _TR("错误"), MB_OK | MB_ICONERROR);
        return;
    }

    // 检查是否已在黑名单中
    if (IPBlacklist::getInstance().IsBlacklisted(ipStr)) {
        MessageBox(_TR("该 IP 已在黑名单中"), _TR("提示"), MB_OK | MB_ICONINFORMATION);
        return;
    }

    // 如果在白名单中，先移除
    if (IPWhitelist::getInstance().IsWhitelisted(ipStr)) {
        IPWhitelist::getInstance().RemoveIP(ipStr);
        // 从白名单列表中移除
        for (int i = 0; i < m_ListWhitelist.GetCount(); i++) {
            CString item;
            m_ListWhitelist.GetText(i, item);
            if (item == ip) {
                m_ListWhitelist.DeleteString(i);
                break;
            }
        }
    }

    // 添加到黑名单
    IPBlacklist::getInstance().AddIP(ipStr);
    m_ListBlacklist.AddString(ip);
    m_EditIP.SetWindowText(_T(""));
}

void CNetworkDlg::OnBnClickedDelBlacklist()
{
    int sel = m_ListBlacklist.GetCurSel();
    if (sel == LB_ERR) {
        MessageBox(_TR("请先选择要删除的 IP"), _TR("提示"), MB_OK | MB_ICONINFORMATION);
        return;
    }

    CString ip;
    m_ListBlacklist.GetText(sel, ip);
    CStringA ipA(ip);
    std::string ipStr(ipA.GetString());

    IPBlacklist::getInstance().RemoveIP(ipStr);
    m_ListBlacklist.DeleteString(sel);
}

void CNetworkDlg::OnBnClickedOk()
{
    // 获取阈值输入
    UpdateData(TRUE);

    // 验证阈值范围
    if (m_nDllLimitSeconds < 60) m_nDllLimitSeconds = 60;
    if (m_nDllLimitSeconds > 86400) m_nDllLimitSeconds = 86400;
    if (m_nDllLimitCount < 1) m_nDllLimitCount = 1;
    if (m_nDllLimitCount > 100) m_nDllLimitCount = 100;
    if (m_nBanWindow < 10) m_nBanWindow = 10;
    if (m_nBanWindow > 3600) m_nBanWindow = 3600;
    if (m_nBanMaxConn < 1) m_nBanMaxConn = 1;
    if (m_nBanMaxConn > 1000) m_nBanMaxConn = 1000;
    if (m_nBanDuration < 60) m_nBanDuration = 60;
    if (m_nBanDuration > 86400) m_nBanDuration = 86400;

    // 保存配置
    SaveLists();
    SaveThresholds();

    // 刷新缓存
    ReloadBanConfig();
    ReloadDllRateLimitConfig();

    __super::OnOK();
}