1061 lines
35 KiB
Rust
1061 lines
35 KiB
Rust
use once_cell::sync::OnceCell;
|
||
use tauri::Emitter;
|
||
use tauri::tray::TrayIconBuilder;
|
||
#[cfg(target_os = "macos")]
|
||
pub mod speed_rate;
|
||
use crate::module::lightweight;
|
||
use crate::process::AsyncHandler;
|
||
use crate::utils::window_manager::WindowManager;
|
||
use crate::{
|
||
Type, cmd,
|
||
config::Config,
|
||
feat, logging,
|
||
module::lightweight::is_in_lightweight_mode,
|
||
singleton_lazy,
|
||
utils::{dirs::find_target_icons, i18n::t},
|
||
};
|
||
|
||
use super::handle;
|
||
use anyhow::Result;
|
||
use futures::future::join_all;
|
||
use parking_lot::Mutex;
|
||
use std::collections::HashMap;
|
||
use std::{
|
||
fs,
|
||
sync::atomic::{AtomicBool, Ordering},
|
||
time::{Duration, Instant},
|
||
};
|
||
use tauri::{
|
||
AppHandle, Wry,
|
||
menu::{CheckMenuItem, IsMenuItem, MenuEvent, MenuItem, PredefinedMenuItem, Submenu},
|
||
tray::{MouseButton, MouseButtonState, TrayIconEvent},
|
||
};
|
||
|
||
// TODO:
|
||
// - 定时更新代理菜单(5s)
|
||
// - 前端选择节点/代理组测速后，发送 event 事件，Tray 监听此事件
|
||
// - 是否需要将可变菜单抽离存储起来，后续直接更新对应菜单实例，无需重新创建菜单(待考虑)
|
||
|
||
#[derive(Clone)]
|
||
struct TrayState {}
|
||
|
||
// 托盘点击防抖机制
|
||
static TRAY_CLICK_DEBOUNCE: OnceCell<Mutex<Instant>> = OnceCell::new();
|
||
const TRAY_CLICK_DEBOUNCE_MS: u64 = 300;
|
||
|
||
fn get_tray_click_debounce() -> &'static Mutex<Instant> {
|
||
TRAY_CLICK_DEBOUNCE.get_or_init(|| Mutex::new(Instant::now() - Duration::from_secs(1)))
|
||
}
|
||
|
||
fn should_handle_tray_click() -> bool {
|
||
let debounce_lock = get_tray_click_debounce();
|
||
let mut last_click = debounce_lock.lock();
|
||
let now = Instant::now();
|
||
|
||
if now.duration_since(*last_click) >= Duration::from_millis(TRAY_CLICK_DEBOUNCE_MS) {
|
||
*last_click = now;
|
||
true
|
||
} else {
|
||
log::debug!(target: "app", "托盘点击被防抖机制忽略，距离上次点击 {:?}ms",
|
||
now.duration_since(*last_click).as_millis());
|
||
false
|
||
}
|
||
}
|
||
|
||
#[cfg(target_os = "macos")]
|
||
pub struct Tray {
|
||
last_menu_update: Mutex<Option<Instant>>,
|
||
menu_updating: AtomicBool,
|
||
}
|
||
|
||
#[cfg(not(target_os = "macos"))]
|
||
pub struct Tray {
|
||
last_menu_update: Mutex<Option<Instant>>,
|
||
menu_updating: AtomicBool,
|
||
}
|
||
|
||
impl TrayState {
|
||
pub async fn get_common_tray_icon() -> (bool, Vec<u8>) {
|
||
let verge = Config::verge().await.latest_ref().clone();
|
||
let is_common_tray_icon = verge.common_tray_icon.unwrap_or(false);
|
||
if is_common_tray_icon
|
||
&& let Ok(Some(common_icon_path)) = find_target_icons("common")
|
||
&& let Ok(icon_data) = fs::read(common_icon_path)
|
||
{
|
||
return (true, icon_data);
|
||
}
|
||
#[cfg(target_os = "macos")]
|
||
{
|
||
let tray_icon_colorful = verge.tray_icon.unwrap_or("monochrome".to_string());
|
||
if tray_icon_colorful == "monochrome" {
|
||
(
|
||
false,
|
||
include_bytes!("../../../icons/tray-icon-mono.ico").to_vec(),
|
||
)
|
||
} else {
|
||
(
|
||
false,
|
||
include_bytes!("../../../icons/tray-icon.ico").to_vec(),
|
||
)
|
||
}
|
||
}
|
||
|
||
#[cfg(not(target_os = "macos"))]
|
||
{
|
||
(
|
||
false,
|
||
include_bytes!("../../../icons/tray-icon.ico").to_vec(),
|
||
)
|
||
}
|
||
}
|
||
|
||
pub async fn get_sysproxy_tray_icon() -> (bool, Vec<u8>) {
|
||
let verge = Config::verge().await.latest_ref().clone();
|
||
let is_sysproxy_tray_icon = verge.sysproxy_tray_icon.unwrap_or(false);
|
||
if is_sysproxy_tray_icon
|
||
&& let Ok(Some(sysproxy_icon_path)) = find_target_icons("sysproxy")
|
||
&& let Ok(icon_data) = fs::read(sysproxy_icon_path)
|
||
{
|
||
return (true, icon_data);
|
||
}
|
||
#[cfg(target_os = "macos")]
|
||
{
|
||
let tray_icon_colorful = verge.tray_icon.clone().unwrap_or("monochrome".to_string());
|
||
if tray_icon_colorful == "monochrome" {
|
||
(
|
||
false,
|
||
include_bytes!("../../../icons/tray-icon-sys-mono-new.ico").to_vec(),
|
||
)
|
||
} else {
|
||
(
|
||
false,
|
||
include_bytes!("../../../icons/tray-icon-sys.ico").to_vec(),
|
||
)
|
||
}
|
||
}
|
||
|
||
#[cfg(not(target_os = "macos"))]
|
||
{
|
||
(
|
||
false,
|
||
include_bytes!("../../../icons/tray-icon-sys.ico").to_vec(),
|
||
)
|
||
}
|
||
}
|
||
|
||
pub async fn get_tun_tray_icon() -> (bool, Vec<u8>) {
|
||
let verge = Config::verge().await.latest_ref().clone();
|
||
let is_tun_tray_icon = verge.tun_tray_icon.unwrap_or(false);
|
||
if is_tun_tray_icon
|
||
&& let Ok(Some(tun_icon_path)) = find_target_icons("tun")
|
||
&& let Ok(icon_data) = fs::read(tun_icon_path)
|
||
{
|
||
return (true, icon_data);
|
||
}
|
||
#[cfg(target_os = "macos")]
|
||
{
|
||
let tray_icon_colorful = verge.tray_icon.clone().unwrap_or("monochrome".to_string());
|
||
if tray_icon_colorful == "monochrome" {
|
||
(
|
||
false,
|
||
include_bytes!("../../../icons/tray-icon-tun-mono-new.ico").to_vec(),
|
||
)
|
||
} else {
|
||
(
|
||
false,
|
||
include_bytes!("../../../icons/tray-icon-tun.ico").to_vec(),
|
||
)
|
||
}
|
||
}
|
||
#[cfg(not(target_os = "macos"))]
|
||
{
|
||
(
|
||
false,
|
||
include_bytes!("../../../icons/tray-icon-tun.ico").to_vec(),
|
||
)
|
||
}
|
||
}
|
||
}
|
||
|
||
impl Default for Tray {
|
||
fn default() -> Self {
|
||
Tray {
|
||
last_menu_update: Mutex::new(None),
|
||
menu_updating: AtomicBool::new(false),
|
||
}
|
||
}
|
||
}
|
||
|
||
// Use simplified singleton_lazy macro
|
||
singleton_lazy!(Tray, TRAY, Tray::default);
|
||
|
||
impl Tray {
|
||
pub async fn init(&self) -> Result<()> {
|
||
let app_handle = handle::Handle::app_handle();
|
||
self.create_tray_from_handle(app_handle).await?;
|
||
Ok(())
|
||
}
|
||
|
||
/// 更新托盘点击行为
|
||
pub async fn update_click_behavior(&self) -> Result<()> {
|
||
let app_handle = handle::Handle::app_handle();
|
||
let tray_event = { Config::verge().await.latest_ref().tray_event.clone() };
|
||
let tray_event: String = tray_event.unwrap_or("main_window".into());
|
||
let tray = app_handle
|
||
.tray_by_id("main")
|
||
.ok_or_else(|| anyhow::anyhow!("Failed to get main tray"))?;
|
||
match tray_event.as_str() {
|
||
"tray_menu" => tray.set_show_menu_on_left_click(true)?,
|
||
_ => tray.set_show_menu_on_left_click(false)?,
|
||
}
|
||
Ok(())
|
||
}
|
||
|
||
/// 更新托盘菜单
|
||
pub async fn update_menu(&self) -> Result<()> {
|
||
// 调整最小更新间隔，确保状态及时刷新
|
||
const MIN_UPDATE_INTERVAL: Duration = Duration::from_millis(100);
|
||
|
||
// 检查是否正在更新
|
||
if self.menu_updating.load(Ordering::Acquire) {
|
||
return Ok(());
|
||
}
|
||
|
||
// 检查更新频率，但允许重要事件跳过频率限制
|
||
let should_force_update = match std::thread::current().name() {
|
||
Some("main") => true,
|
||
_ => {
|
||
let last_update = self.last_menu_update.lock();
|
||
if let Some(last_time) = *last_update {
|
||
last_time.elapsed() >= MIN_UPDATE_INTERVAL
|
||
} else {
|
||
true
|
||
}
|
||
}
|
||
};
|
||
|
||
if !should_force_update {
|
||
return Ok(());
|
||
}
|
||
|
||
let app_handle = handle::Handle::app_handle();
|
||
|
||
// 设置更新状态
|
||
self.menu_updating.store(true, Ordering::Release);
|
||
|
||
let result = self.update_menu_internal(app_handle).await;
|
||
|
||
{
|
||
let mut last_update = self.last_menu_update.lock();
|
||
*last_update = Some(Instant::now());
|
||
}
|
||
self.menu_updating.store(false, Ordering::Release);
|
||
|
||
result
|
||
}
|
||
|
||
async fn update_menu_internal(&self, app_handle: &AppHandle) -> Result<()> {
|
||
let verge = Config::verge().await.latest_ref().clone();
|
||
let system_proxy = verge.enable_system_proxy.as_ref().unwrap_or(&false);
|
||
let tun_mode = verge.enable_tun_mode.as_ref().unwrap_or(&false);
|
||
let mode = {
|
||
Config::clash()
|
||
.await
|
||
.latest_ref()
|
||
.0
|
||
.get("mode")
|
||
.map(|val| val.as_str().unwrap_or("rule"))
|
||
.unwrap_or("rule")
|
||
.to_owned()
|
||
};
|
||
let profile_uid_and_name = Config::profiles()
|
||
.await
|
||
.data_mut()
|
||
.all_profile_uid_and_name()
|
||
.unwrap_or_default();
|
||
let is_lightweight_mode = is_in_lightweight_mode();
|
||
|
||
match app_handle.tray_by_id("main") {
|
||
Some(tray) => {
|
||
let _ = tray.set_menu(Some(
|
||
create_tray_menu(
|
||
app_handle,
|
||
Some(mode.as_str()),
|
||
*system_proxy,
|
||
*tun_mode,
|
||
profile_uid_and_name,
|
||
is_lightweight_mode,
|
||
)
|
||
.await?,
|
||
));
|
||
log::debug!(target: "app", "托盘菜单更新成功");
|
||
Ok(())
|
||
}
|
||
None => {
|
||
log::warn!(target: "app", "更新托盘菜单失败: 托盘不存在");
|
||
Ok(())
|
||
}
|
||
}
|
||
}
|
||
|
||
/// 更新托盘图标
|
||
#[cfg(target_os = "macos")]
|
||
pub async fn update_icon(&self, _rate: Option<Rate>) -> Result<()> {
|
||
let app_handle = match handle::Handle::app_handle() {
|
||
Some(handle) => handle,
|
||
None => {
|
||
log::warn!(target: "app", "更新托盘图标失败: app_handle不存在");
|
||
return Ok(());
|
||
}
|
||
};
|
||
|
||
let tray = match app_handle.tray_by_id("main") {
|
||
Some(tray) => tray,
|
||
None => {
|
||
log::warn!(target: "app", "更新托盘图标失败: 托盘不存在");
|
||
return Ok(());
|
||
}
|
||
};
|
||
|
||
let verge = Config::verge().await.latest_ref().clone();
|
||
let system_mode = verge.enable_system_proxy.as_ref().unwrap_or(&false);
|
||
let tun_mode = verge.enable_tun_mode.as_ref().unwrap_or(&false);
|
||
|
||
let (_is_custom_icon, icon_bytes) = match (*system_mode, *tun_mode) {
|
||
(true, true) => TrayState::get_tun_tray_icon().await,
|
||
(true, false) => TrayState::get_sysproxy_tray_icon().await,
|
||
(false, true) => TrayState::get_tun_tray_icon().await,
|
||
(false, false) => TrayState::get_common_tray_icon().await,
|
||
};
|
||
|
||
let colorful = verge.tray_icon.clone().unwrap_or("monochrome".to_string());
|
||
let is_colorful = colorful == "colorful";
|
||
|
||
let _ = tray.set_icon(Some(tauri::image::Image::from_bytes(&icon_bytes)?));
|
||
let _ = tray.set_icon_as_template(!is_colorful);
|
||
Ok(())
|
||
}
|
||
|
||
#[cfg(not(target_os = "macos"))]
|
||
pub async fn update_icon(&self) -> Result<()> {
|
||
let app_handle = handle::Handle::app_handle();
|
||
|
||
let tray = match app_handle.tray_by_id("main") {
|
||
Some(tray) => tray,
|
||
None => {
|
||
log::warn!(target: "app", "更新托盘图标失败: 托盘不存在");
|
||
return Ok(());
|
||
}
|
||
};
|
||
|
||
let verge = Config::verge().await.latest_ref().clone();
|
||
let system_mode = verge.enable_system_proxy.as_ref().unwrap_or(&false);
|
||
let tun_mode = verge.enable_tun_mode.as_ref().unwrap_or(&false);
|
||
|
||
let (_is_custom_icon, icon_bytes) = match (*system_mode, *tun_mode) {
|
||
(true, true) => TrayState::get_tun_tray_icon().await,
|
||
(true, false) => TrayState::get_sysproxy_tray_icon().await,
|
||
(false, true) => TrayState::get_tun_tray_icon().await,
|
||
(false, false) => TrayState::get_common_tray_icon().await,
|
||
};
|
||
|
||
let _ = tray.set_icon(Some(tauri::image::Image::from_bytes(&icon_bytes)?));
|
||
Ok(())
|
||
}
|
||
|
||
/// 更新托盘显示状态的函数
|
||
pub async fn update_tray_display(&self) -> Result<()> {
|
||
let app_handle = handle::Handle::app_handle();
|
||
let _tray = app_handle
|
||
.tray_by_id("main")
|
||
.ok_or_else(|| anyhow::anyhow!("Failed to get main tray"))?;
|
||
|
||
// 更新菜单
|
||
self.update_menu().await?;
|
||
|
||
Ok(())
|
||
}
|
||
|
||
/// 更新托盘提示
|
||
pub async fn update_tooltip(&self) -> Result<()> {
|
||
let app_handle = handle::Handle::app_handle();
|
||
|
||
let verge = Config::verge().await.latest_ref().clone();
|
||
let system_proxy = verge.enable_system_proxy.as_ref().unwrap_or(&false);
|
||
let tun_mode = verge.enable_tun_mode.as_ref().unwrap_or(&false);
|
||
|
||
let switch_map = {
|
||
let mut map = std::collections::HashMap::new();
|
||
map.insert(true, "on");
|
||
map.insert(false, "off");
|
||
map
|
||
};
|
||
|
||
let mut current_profile_name = "None".to_string();
|
||
{
|
||
let profiles = Config::profiles().await;
|
||
let profiles = profiles.latest_ref();
|
||
if let Some(current_profile_uid) = profiles.get_current()
|
||
&& let Ok(profile) = profiles.get_item(¤t_profile_uid)
|
||
{
|
||
current_profile_name = match &profile.name {
|
||
Some(profile_name) => profile_name.to_string(),
|
||
None => current_profile_name,
|
||
};
|
||
}
|
||
}
|
||
|
||
// Get localized strings before using them
|
||
let sys_proxy_text = t("SysProxy").await;
|
||
let tun_text = t("TUN").await;
|
||
let profile_text = t("Profile").await;
|
||
|
||
let version = env!("CARGO_PKG_VERSION");
|
||
if let Some(tray) = app_handle.tray_by_id("main") {
|
||
let _ = tray.set_tooltip(Some(&format!(
|
||
"Clash Verge {version}\n{}: {}\n{}: {}\n{}: {}",
|
||
sys_proxy_text,
|
||
switch_map[system_proxy],
|
||
tun_text,
|
||
switch_map[tun_mode],
|
||
profile_text,
|
||
current_profile_name
|
||
)));
|
||
} else {
|
||
log::warn!(target: "app", "更新托盘提示失败: 托盘不存在");
|
||
}
|
||
|
||
Ok(())
|
||
}
|
||
|
||
pub async fn update_part(&self) -> Result<()> {
|
||
// self.update_menu().await?;
|
||
// 更新轻量模式显示状态
|
||
self.update_tray_display().await?;
|
||
self.update_icon().await?;
|
||
self.update_tooltip().await?;
|
||
Ok(())
|
||
}
|
||
|
||
pub async fn create_tray_from_handle(&self, app_handle: &AppHandle) -> Result<()> {
|
||
log::info!(target: "app", "正在从AppHandle创建系统托盘");
|
||
|
||
// 获取图标
|
||
let icon_bytes = TrayState::get_common_tray_icon().await.1;
|
||
let icon = tauri::image::Image::from_bytes(&icon_bytes)?;
|
||
|
||
#[cfg(target_os = "linux")]
|
||
let builder = TrayIconBuilder::with_id("main")
|
||
.icon(icon)
|
||
.icon_as_template(false);
|
||
|
||
#[cfg(any(target_os = "macos", target_os = "windows"))]
|
||
let show_menu_on_left_click = {
|
||
let tray_event = { Config::verge().await.latest_ref().tray_event.clone() };
|
||
let tray_event: String = tray_event.unwrap_or("main_window".into());
|
||
tray_event.as_str() == "tray_menu"
|
||
};
|
||
|
||
#[cfg(not(target_os = "linux"))]
|
||
let mut builder = TrayIconBuilder::with_id("main")
|
||
.icon(icon)
|
||
.icon_as_template(false);
|
||
|
||
#[cfg(any(target_os = "macos", target_os = "windows"))]
|
||
{
|
||
if !show_menu_on_left_click {
|
||
builder = builder.show_menu_on_left_click(false);
|
||
}
|
||
}
|
||
|
||
let tray = builder.build(app_handle)?;
|
||
|
||
tray.on_tray_icon_event(|_app_handle, event| {
|
||
AsyncHandler::spawn(|| async move {
|
||
let tray_event = { Config::verge().await.latest_ref().tray_event.clone() };
|
||
let tray_event: String = tray_event.unwrap_or("main_window".into());
|
||
log::debug!(target: "app", "tray event: {tray_event:?}");
|
||
|
||
if let TrayIconEvent::Click {
|
||
button: MouseButton::Left,
|
||
button_state: MouseButtonState::Down,
|
||
..
|
||
} = event
|
||
{
|
||
// 添加防抖检查，防止快速连击
|
||
if !should_handle_tray_click() {
|
||
return;
|
||
}
|
||
|
||
use std::future::Future;
|
||
use std::pin::Pin;
|
||
|
||
let fut: Pin<Box<dyn Future<Output = ()> + Send>> = match tray_event.as_str() {
|
||
"system_proxy" => Box::pin(async move {
|
||
feat::toggle_system_proxy().await;
|
||
}),
|
||
"tun_mode" => Box::pin(async move {
|
||
feat::toggle_tun_mode(None).await;
|
||
}),
|
||
"main_window" => Box::pin(async move {
|
||
if !lightweight::exit_lightweight_mode().await {
|
||
WindowManager::show_main_window().await;
|
||
};
|
||
}),
|
||
_ => Box::pin(async move {}),
|
||
};
|
||
fut.await;
|
||
}
|
||
});
|
||
});
|
||
tray.on_menu_event(on_menu_event);
|
||
log::info!(target: "app", "系统托盘创建成功");
|
||
Ok(())
|
||
}
|
||
|
||
// 托盘统一的状态更新函数
|
||
pub async fn update_all_states(&self) -> Result<()> {
|
||
// 确保所有状态更新完成
|
||
self.update_tray_display().await?;
|
||
// self.update_menu().await?;
|
||
self.update_icon().await?;
|
||
self.update_tooltip().await?;
|
||
|
||
Ok(())
|
||
}
|
||
}
|
||
|
||
async fn create_tray_menu(
|
||
app_handle: &AppHandle,
|
||
mode: Option<&str>,
|
||
system_proxy_enabled: bool,
|
||
tun_mode_enabled: bool,
|
||
profile_uid_and_name: Vec<(String, String)>,
|
||
is_lightweight_mode: bool,
|
||
) -> Result<tauri::menu::Menu<Wry>> {
|
||
let mode = mode.unwrap_or("");
|
||
|
||
// 获取当前配置文件的选中代理组信息
|
||
let current_profile_selected = {
|
||
let profiles_config = Config::profiles().await;
|
||
let profiles_ref = profiles_config.latest_ref();
|
||
profiles_ref
|
||
.get_current()
|
||
.and_then(|uid| profiles_ref.get_item(&uid).ok())
|
||
.and_then(|profile| profile.selected.clone())
|
||
.unwrap_or_default()
|
||
};
|
||
|
||
let proxy_nodes_data = handle::Handle::mihomo().await.get_proxies().await?;
|
||
|
||
let version = env!("CARGO_PKG_VERSION");
|
||
|
||
let hotkeys = Config::verge()
|
||
.await
|
||
.latest_ref()
|
||
.hotkeys
|
||
.as_ref()
|
||
.map(|h| {
|
||
h.iter()
|
||
.filter_map(|item| {
|
||
let mut parts = item.split(',');
|
||
match (parts.next(), parts.next()) {
|
||
(Some(func), Some(key)) => Some((func.to_string(), key.to_string())),
|
||
_ => None,
|
||
}
|
||
})
|
||
.collect::<std::collections::HashMap<String, String>>()
|
||
})
|
||
.unwrap_or_default();
|
||
|
||
let profile_menu_items: Vec<CheckMenuItem<Wry>> = {
|
||
let futures = profile_uid_and_name
|
||
.iter()
|
||
.map(|(profile_uid, profile_name)| {
|
||
let app_handle = app_handle.clone();
|
||
let profile_uid = profile_uid.clone();
|
||
let profile_name = profile_name.clone();
|
||
async move {
|
||
let is_current_profile = Config::profiles()
|
||
.await
|
||
.data_mut()
|
||
.is_current_profile_index(profile_uid.to_string());
|
||
CheckMenuItem::with_id(
|
||
&app_handle,
|
||
format!("profiles_{profile_uid}"),
|
||
t(&profile_name).await,
|
||
true,
|
||
is_current_profile,
|
||
None::<&str>,
|
||
)
|
||
}
|
||
});
|
||
let results = join_all(futures).await;
|
||
results.into_iter().collect::<Result<Vec<_>, _>>()?
|
||
};
|
||
|
||
// 代理组子菜单
|
||
let proxy_submenus: Vec<Submenu<Wry>> = {
|
||
let mut submenus = Vec::new();
|
||
let mut group_name_submenus_hash = HashMap::new();
|
||
|
||
for (group_name, group_data) in proxy_nodes_data.proxies.iter() {
|
||
// Filter groups based on mode
|
||
let should_show = match mode {
|
||
"global" => group_name == "GLOBAL",
|
||
_ => group_name != "GLOBAL",
|
||
} &&
|
||
// Check if the group is hidden
|
||
!group_data.hidden.unwrap_or_default();
|
||
|
||
if !should_show {
|
||
continue;
|
||
}
|
||
|
||
let Some(all_proxies) = group_data.all.as_ref() else {
|
||
continue;
|
||
};
|
||
|
||
let now_proxy = group_data.now.as_deref().unwrap_or_default();
|
||
|
||
// Create proxy items
|
||
let group_items: Vec<CheckMenuItem<Wry>> = all_proxies
|
||
.iter()
|
||
.filter_map(|proxy_str| {
|
||
let is_selected = *proxy_str == now_proxy;
|
||
let item_id = format!("proxy_{}_{}", group_name, proxy_str);
|
||
|
||
// Get delay for display
|
||
let delay_text = proxy_nodes_data
|
||
.proxies
|
||
.get(proxy_str)
|
||
.and_then(|h| h.history.last())
|
||
.map(|h| match h.delay {
|
||
0 => "-ms".to_string(),
|
||
delay if delay >= 10000 => "-ms".to_string(),
|
||
_ => format!("{}ms", h.delay),
|
||
})
|
||
.unwrap_or_else(|| "-ms".to_string());
|
||
|
||
let display_text = format!("{} | {}", proxy_str, delay_text);
|
||
|
||
CheckMenuItem::with_id(
|
||
app_handle,
|
||
item_id,
|
||
display_text,
|
||
true,
|
||
is_selected,
|
||
None::<&str>,
|
||
)
|
||
.map_err(|e| log::warn!(target: "app", "创建代理菜单项失败: {}", e))
|
||
.ok()
|
||
})
|
||
.collect();
|
||
|
||
if group_items.is_empty() {
|
||
continue;
|
||
}
|
||
|
||
// Determine if group is active
|
||
let is_group_active = match mode {
|
||
"global" => group_name == "GLOBAL" && !now_proxy.is_empty(),
|
||
"direct" => false,
|
||
_ => {
|
||
current_profile_selected
|
||
.iter()
|
||
.any(|s| s.name.as_deref() == Some(group_name))
|
||
&& !now_proxy.is_empty()
|
||
}
|
||
};
|
||
|
||
let group_display_name = if is_group_active {
|
||
format!("✓ {}", group_name)
|
||
} else {
|
||
group_name.to_string()
|
||
};
|
||
|
||
let group_items_refs: Vec<&dyn IsMenuItem<Wry>> = group_items
|
||
.iter()
|
||
.map(|item| item as &dyn IsMenuItem<Wry>)
|
||
.collect();
|
||
|
||
if let Ok(submenu) = Submenu::with_id_and_items(
|
||
app_handle,
|
||
format!("proxy_group_{}", group_name),
|
||
group_display_name,
|
||
true,
|
||
&group_items_refs,
|
||
) {
|
||
group_name_submenus_hash.insert(group_name.to_string(), submenu);
|
||
} else {
|
||
log::warn!(target: "app", "创建代理组子菜单失败: {}", group_name);
|
||
}
|
||
}
|
||
|
||
// 获取运行时代理组配置
|
||
let runtime_proxy_groups_config = cmd::get_runtime_config()
|
||
.await
|
||
.map_err(|e| {
|
||
logging!(
|
||
error,
|
||
Type::Cmd,
|
||
"Failed to fetch runtime proxy groups for tray menu: {e}"
|
||
);
|
||
})
|
||
.ok()
|
||
.flatten()
|
||
.map(|config| {
|
||
config
|
||
.get("proxy-groups")
|
||
.and_then(|groups| groups.as_sequence())
|
||
.map(|groups| {
|
||
groups
|
||
.iter()
|
||
.filter_map(|group| group.get("name"))
|
||
.filter_map(|name| name.as_str())
|
||
.map(|name| name.to_string())
|
||
.collect::<Vec<String>>()
|
||
})
|
||
.unwrap_or_default()
|
||
});
|
||
|
||
if let Some(runtime_proxy_groups_config) = runtime_proxy_groups_config {
|
||
for group_name in runtime_proxy_groups_config {
|
||
if let Some(submenu) = group_name_submenus_hash.get(&group_name) {
|
||
submenus.push(submenu.clone());
|
||
}
|
||
}
|
||
} else {
|
||
for (_, submenu) in group_name_submenus_hash {
|
||
submenus.push(submenu);
|
||
}
|
||
}
|
||
|
||
submenus
|
||
};
|
||
|
||
// Pre-fetch all localized strings
|
||
let dashboard_text = t("Dashboard").await;
|
||
let rule_mode_text = t("Rule Mode").await;
|
||
let global_mode_text = t("Global Mode").await;
|
||
let direct_mode_text = t("Direct Mode").await;
|
||
let profiles_text = t("Profiles").await;
|
||
let proxies_text = t("Proxies").await;
|
||
let system_proxy_text = t("System Proxy").await;
|
||
let tun_mode_text = t("TUN Mode").await;
|
||
let lightweight_mode_text = t("LightWeight Mode").await;
|
||
let copy_env_text = t("Copy Env").await;
|
||
let conf_dir_text = t("Conf Dir").await;
|
||
let core_dir_text = t("Core Dir").await;
|
||
let logs_dir_text = t("Logs Dir").await;
|
||
let open_dir_text = t("Open Dir").await;
|
||
let restart_clash_text = t("Restart Clash Core").await;
|
||
let restart_app_text = t("Restart App").await;
|
||
let verge_version_text = t("Verge Version").await;
|
||
let more_text = t("More").await;
|
||
let exit_text = t("Exit").await;
|
||
|
||
// Convert to references only when needed
|
||
let profile_menu_items_refs: Vec<&dyn IsMenuItem<Wry>> = profile_menu_items
|
||
.iter()
|
||
.map(|item| item as &dyn IsMenuItem<Wry>)
|
||
.collect();
|
||
|
||
let open_window = &MenuItem::with_id(
|
||
app_handle,
|
||
"open_window",
|
||
dashboard_text,
|
||
true,
|
||
hotkeys.get("open_or_close_dashboard").map(|s| s.as_str()),
|
||
)?;
|
||
|
||
let rule_mode = &CheckMenuItem::with_id(
|
||
app_handle,
|
||
"rule_mode",
|
||
rule_mode_text,
|
||
true,
|
||
mode == "rule",
|
||
hotkeys.get("clash_mode_rule").map(|s| s.as_str()),
|
||
)?;
|
||
|
||
let global_mode = &CheckMenuItem::with_id(
|
||
app_handle,
|
||
"global_mode",
|
||
global_mode_text,
|
||
true,
|
||
mode == "global",
|
||
hotkeys.get("clash_mode_global").map(|s| s.as_str()),
|
||
)?;
|
||
|
||
let direct_mode = &CheckMenuItem::with_id(
|
||
app_handle,
|
||
"direct_mode",
|
||
direct_mode_text,
|
||
true,
|
||
mode == "direct",
|
||
hotkeys.get("clash_mode_direct").map(|s| s.as_str()),
|
||
)?;
|
||
|
||
let profiles = &Submenu::with_id_and_items(
|
||
app_handle,
|
||
"profiles",
|
||
profiles_text,
|
||
true,
|
||
&profile_menu_items_refs,
|
||
)?;
|
||
|
||
// 创建代理主菜单
|
||
let proxies_submenu = if !proxy_submenus.is_empty() {
|
||
let proxy_submenu_refs: Vec<&dyn IsMenuItem<Wry>> = proxy_submenus
|
||
.iter()
|
||
.map(|submenu| submenu as &dyn IsMenuItem<Wry>)
|
||
.collect();
|
||
|
||
Some(Submenu::with_id_and_items(
|
||
app_handle,
|
||
"proxies",
|
||
proxies_text,
|
||
true,
|
||
&proxy_submenu_refs,
|
||
)?)
|
||
} else {
|
||
None
|
||
};
|
||
|
||
let system_proxy = &CheckMenuItem::with_id(
|
||
app_handle,
|
||
"system_proxy",
|
||
system_proxy_text,
|
||
true,
|
||
system_proxy_enabled,
|
||
hotkeys.get("toggle_system_proxy").map(|s| s.as_str()),
|
||
)?;
|
||
|
||
let tun_mode = &CheckMenuItem::with_id(
|
||
app_handle,
|
||
"tun_mode",
|
||
tun_mode_text,
|
||
true,
|
||
tun_mode_enabled,
|
||
hotkeys.get("toggle_tun_mode").map(|s| s.as_str()),
|
||
)?;
|
||
|
||
let lighteweight_mode = &CheckMenuItem::with_id(
|
||
app_handle,
|
||
"entry_lightweight_mode",
|
||
lightweight_mode_text,
|
||
true,
|
||
is_lightweight_mode,
|
||
hotkeys.get("entry_lightweight_mode").map(|s| s.as_str()),
|
||
)?;
|
||
|
||
let copy_env = &MenuItem::with_id(app_handle, "copy_env", copy_env_text, true, None::<&str>)?;
|
||
|
||
let open_app_dir = &MenuItem::with_id(
|
||
app_handle,
|
||
"open_app_dir",
|
||
conf_dir_text,
|
||
true,
|
||
None::<&str>,
|
||
)?;
|
||
|
||
let open_core_dir = &MenuItem::with_id(
|
||
app_handle,
|
||
"open_core_dir",
|
||
core_dir_text,
|
||
true,
|
||
None::<&str>,
|
||
)?;
|
||
|
||
let open_logs_dir = &MenuItem::with_id(
|
||
app_handle,
|
||
"open_logs_dir",
|
||
logs_dir_text,
|
||
true,
|
||
None::<&str>,
|
||
)?;
|
||
|
||
let open_dir = &Submenu::with_id_and_items(
|
||
app_handle,
|
||
"open_dir",
|
||
open_dir_text,
|
||
true,
|
||
&[open_app_dir, open_core_dir, open_logs_dir],
|
||
)?;
|
||
|
||
let restart_clash = &MenuItem::with_id(
|
||
app_handle,
|
||
"restart_clash",
|
||
restart_clash_text,
|
||
true,
|
||
None::<&str>,
|
||
)?;
|
||
|
||
let restart_app = &MenuItem::with_id(
|
||
app_handle,
|
||
"restart_app",
|
||
restart_app_text,
|
||
true,
|
||
None::<&str>,
|
||
)?;
|
||
|
||
let app_version = &MenuItem::with_id(
|
||
app_handle,
|
||
"app_version",
|
||
format!("{} {version}", verge_version_text),
|
||
true,
|
||
None::<&str>,
|
||
)?;
|
||
|
||
let more = &Submenu::with_id_and_items(
|
||
app_handle,
|
||
"more",
|
||
more_text,
|
||
true,
|
||
&[restart_clash, restart_app, app_version],
|
||
)?;
|
||
|
||
let quit = &MenuItem::with_id(app_handle, "quit", exit_text, true, Some("CmdOrControl+Q"))?;
|
||
|
||
let separator = &PredefinedMenuItem::separator(app_handle)?;
|
||
|
||
// 动态构建菜单项
|
||
let mut menu_items: Vec<&dyn IsMenuItem<Wry>> = vec![
|
||
open_window,
|
||
separator,
|
||
rule_mode,
|
||
global_mode,
|
||
direct_mode,
|
||
separator,
|
||
profiles,
|
||
];
|
||
|
||
// 如果有代理节点，添加代理节点菜单
|
||
if let Some(ref proxies_menu) = proxies_submenu {
|
||
menu_items.push(proxies_menu);
|
||
}
|
||
|
||
menu_items.extend_from_slice(&[
|
||
separator,
|
||
system_proxy as &dyn IsMenuItem<Wry>,
|
||
tun_mode as &dyn IsMenuItem<Wry>,
|
||
separator,
|
||
lighteweight_mode as &dyn IsMenuItem<Wry>,
|
||
copy_env as &dyn IsMenuItem<Wry>,
|
||
open_dir as &dyn IsMenuItem<Wry>,
|
||
more as &dyn IsMenuItem<Wry>,
|
||
separator,
|
||
quit as &dyn IsMenuItem<Wry>,
|
||
]);
|
||
|
||
let menu = tauri::menu::MenuBuilder::new(app_handle)
|
||
.items(&menu_items)
|
||
.build()?;
|
||
Ok(menu)
|
||
}
|
||
|
||
fn on_menu_event(_: &AppHandle, event: MenuEvent) {
|
||
AsyncHandler::spawn(|| async move {
|
||
match event.id.as_ref() {
|
||
mode @ ("rule_mode" | "global_mode" | "direct_mode") => {
|
||
let mode = &mode[0..mode.len() - 5]; // Removing the "_mode" suffix
|
||
logging!(
|
||
info,
|
||
Type::ProxyMode,
|
||
true,
|
||
"Switch Proxy Mode To: {}",
|
||
mode
|
||
);
|
||
feat::change_clash_mode(mode.into()).await;
|
||
}
|
||
"open_window" => {
|
||
log::info!(target: "app", "托盘菜单点击: 打开窗口");
|
||
|
||
if !should_handle_tray_click() {
|
||
return;
|
||
}
|
||
if !lightweight::exit_lightweight_mode().await {
|
||
WindowManager::show_main_window().await;
|
||
};
|
||
}
|
||
"system_proxy" => {
|
||
feat::toggle_system_proxy().await;
|
||
}
|
||
"tun_mode" => {
|
||
feat::toggle_tun_mode(None).await;
|
||
}
|
||
"copy_env" => feat::copy_clash_env().await,
|
||
"open_app_dir" => {
|
||
let _ = cmd::open_app_dir().await;
|
||
}
|
||
"open_core_dir" => {
|
||
let _ = cmd::open_core_dir().await;
|
||
}
|
||
"open_logs_dir" => {
|
||
let _ = cmd::open_logs_dir().await;
|
||
}
|
||
"restart_clash" => feat::restart_clash_core().await,
|
||
"restart_app" => feat::restart_app().await,
|
||
"entry_lightweight_mode" => {
|
||
if !should_handle_tray_click() {
|
||
return;
|
||
}
|
||
if !is_in_lightweight_mode() {
|
||
lightweight::entry_lightweight_mode().await; // Await async function
|
||
} else {
|
||
lightweight::exit_lightweight_mode().await; // Await async function
|
||
}
|
||
}
|
||
"quit" => {
|
||
feat::quit().await;
|
||
}
|
||
id if id.starts_with("profiles_") => {
|
||
let profile_index = &id["profiles_".len()..];
|
||
feat::toggle_proxy_profile(profile_index.into()).await;
|
||
}
|
||
id if id.starts_with("proxy_") => {
|
||
// proxy_{group_name}_{proxy_name}
|
||
let parts: Vec<&str> = id.splitn(3, '_').collect();
|
||
|
||
if parts.len() == 3 && parts[0] == "proxy" {
|
||
let group_name = parts[1];
|
||
let proxy_name = parts[2];
|
||
|
||
match cmd::proxy::update_proxy_and_sync(
|
||
group_name.to_string(),
|
||
proxy_name.to_string(),
|
||
)
|
||
.await
|
||
{
|
||
Ok(_) => {
|
||
log::info!(target: "app", "切换代理成功: {} -> {}", group_name, proxy_name);
|
||
}
|
||
Err(e) => {
|
||
log::error!(target: "app", "切换代理失败: {} -> {}, 错误: {:?}", group_name, proxy_name, e);
|
||
|
||
// Fallback to IPC update
|
||
if (handle::Handle::mihomo()
|
||
.await
|
||
.select_node_for_group(group_name, proxy_name)
|
||
.await)
|
||
.is_ok()
|
||
{
|
||
log::info!(target: "app", "代理切换回退成功: {} -> {}", group_name, proxy_name);
|
||
|
||
let app_handle = handle::Handle::app_handle();
|
||
let _ = app_handle.emit("verge://force-refresh-proxies", ());
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
_ => {}
|
||
}
|
||
|
||
// Ensure tray state update is awaited and properly handled
|
||
if let Err(e) = Tray::global().update_all_states().await {
|
||
log::warn!(target: "app", "更新托盘状态失败: {e}");
|
||
}
|
||
});
|
||
}
|