性能优化

2026-08-06 10:33:16 +08:00
parent c7578a2e6b
commit e66c53e66d
105 changed files with 7273 additions and 5002 deletions
@@ -0,0 +1,92 @@
+//! 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(crate) 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(crate) 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)
+    }
+}