//! Windows 系统代理开关。

use super::MihomoManager;

// ===================== 系统代理（Windows） =====================

#[cfg(windows)]
pub(crate) fn set_system_proxy_windows(addr: &str) -> Result<(), String> {
    use winreg::enums::*;
    use winreg::RegKey;
    let hkcu = RegKey::predef(HKEY_CURRENT_USER);
    let (settings, _) = hkcu
        .create_subkey("Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings")
        .map_err(|e| e.to_string())?;
    settings
        .set_value("ProxyEnable", &1u32)
        .map_err(|e| e.to_string())?;
    settings
        .set_value("ProxyServer", &addr)
        .map_err(|e| e.to_string())?;
    notify_wininet();
    Ok(())
}

#[cfg(windows)]
pub(crate) fn clear_system_proxy_windows() -> Result<(), String> {
    use winreg::enums::*;
    use winreg::RegKey;
    let hkcu = RegKey::predef(HKEY_CURRENT_USER);
    let (settings, _) = hkcu
        .create_subkey("Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings")
        .map_err(|e| e.to_string())?;
    settings
        .set_value("ProxyEnable", &0u32)
        .map_err(|e| e.to_string())?;
    notify_wininet();
    Ok(())
}

#[cfg(windows)]
pub fn get_system_proxy_windows() -> bool {
    use winreg::enums::*;
    use winreg::RegKey;
    let hkcu = RegKey::predef(HKEY_CURRENT_USER);
    hkcu.open_subkey("Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings")
        .ok()
        .and_then(|s| s.get_value::<u32, _>("ProxyEnable").ok())
        .map(|v| v != 0)
        .unwrap_or(false)
}

#[cfg(windows)]
fn notify_wininet() {
    unsafe {
        use windows_sys::Win32::Networking::WinInet::*;
        InternetSetOptionW(std::ptr::null(), INTERNET_OPTION_SETTINGS_CHANGED, std::ptr::null(), 0);
        InternetSetOptionW(std::ptr::null(), INTERNET_OPTION_REFRESH, std::ptr::null(), 0);
    }
}

#[cfg(not(windows))]
pub(crate) fn set_system_proxy_windows(_addr: &str) -> Result<(), String> {
    Err("系统代理仅支持 Windows".into())
}
#[cfg(not(windows))]
pub(crate) fn clear_system_proxy_windows() -> Result<(), String> {
    Err("系统代理仅支持 Windows".into())
}
#[cfg(not(windows))]
pub fn get_system_proxy_windows() -> bool {
    false
}

impl MihomoManager {
    /// 开启系统代理（托盘菜单调用）
    pub fn enable_system_proxy(&self) -> Result<(), String> {
        let settings = self.load_settings();
        let addr = format!("127.0.0.1:{}", settings.mixed_port);
        set_system_proxy_windows(&addr)?;
        let mut s = settings;
        s.system_proxy = true;
        self.save_settings(&s)
    }

    /// 关闭系统代理（托盘菜单调用）
    pub fn disable_system_proxy(&self) -> Result<(), String> {
        clear_system_proxy_windows()?;
        let mut s = self.load_settings();
        s.system_proxy = false;
        self.save_settings(&s)
    }
}