use crate::{ config::{Config, PrfItem}, core::*, logging, logging_error, module::lightweight::{self, auto_lightweight_mode_init}, process::AsyncHandler, utils::{init, logging::Type, server}, wrap_err, }; use anyhow::{bail, Result}; use once_cell::sync::OnceCell; use parking_lot::{Mutex, RwLock}; use percent_encoding::percent_decode_str; use scopeguard; use std::time::{Duration, Instant}; use tauri::{AppHandle, Manager}; use tauri::Url; //#[cfg(not(target_os = "linux"))] // use window_shadows::set_shadow; pub static VERSION: OnceCell = OnceCell::new(); // 定义默认窗口尺寸常量 const DEFAULT_WIDTH: u32 = 940; const DEFAULT_HEIGHT: u32 = 700; // 添加全局UI准备就绪标志 static UI_READY: OnceCell> = OnceCell::new(); // 窗口创建锁,防止并发创建窗口 static WINDOW_CREATING: OnceCell> = OnceCell::new(); // UI就绪阶段状态枚举 #[derive(Debug, Clone, Copy, PartialEq)] pub enum UiReadyStage { NotStarted, Loading, DomReady, ResourcesLoaded, Ready, } // UI就绪详细状态 #[derive(Debug)] struct UiReadyState { stage: RwLock, } impl Default for UiReadyState { fn default() -> Self { Self { stage: RwLock::new(UiReadyStage::NotStarted), } } } // 获取UI就绪状态细节 static UI_READY_STATE: OnceCell = OnceCell::new(); fn get_window_creating_lock() -> &'static Mutex<(bool, Instant)> { WINDOW_CREATING.get_or_init(|| Mutex::new((false, Instant::now()))) } fn get_ui_ready() -> &'static RwLock { UI_READY.get_or_init(|| RwLock::new(false)) } fn get_ui_ready_state() -> &'static UiReadyState { UI_READY_STATE.get_or_init(UiReadyState::default) } // 更新UI准备阶段 pub fn update_ui_ready_stage(stage: UiReadyStage) { let state = get_ui_ready_state(); let mut stage_lock = state.stage.write(); *stage_lock = stage; // 如果是最终阶段,标记UI完全就绪 if stage == UiReadyStage::Ready { mark_ui_ready(); } } // 标记UI已准备就绪 pub fn mark_ui_ready() { let mut ready = get_ui_ready().write(); *ready = true; logging!(info, Type::Window, true, "UI已标记为完全就绪"); } // 重置UI就绪状态 pub fn reset_ui_ready() { { let mut ready = get_ui_ready().write(); *ready = false; } { let state = get_ui_ready_state(); let mut stage = state.stage.write(); *stage = UiReadyStage::NotStarted; } logging!(info, Type::Window, true, "UI就绪状态已重置"); } /// 异步方式处理启动后的额外任务 pub async fn resolve_setup_async(app_handle: &AppHandle) { let start_time = std::time::Instant::now(); logging!( info, Type::Setup, true, "开始执行异步设置任务... 线程ID: {:?}", std::thread::current().id() ); if VERSION.get().is_none() { let version = app_handle.package_info().version.to_string(); VERSION.get_or_init(|| { logging!(info, Type::Setup, true, "初始化版本信息: {}", version); version.clone() }); } logging_error!(Type::Setup, true, init::init_scheme()); logging_error!(Type::Setup, true, init::startup_script().await); logging!( info, Type::Config, true, "开始初始化配置... 线程ID: {:?}", std::thread::current().id() ); logging_error!(Type::Config, true, Config::init_config().await); logging!(info, Type::Config, true, "配置初始化完成"); logging!(trace, Type::Core, true, "启动核心管理器..."); logging_error!(Type::Core, true, CoreManager::global().init().await); log::trace!(target: "app", "启动内嵌服务器..."); server::embed_server(); logging!(trace, Type::Core, true, "启动 IPC 监控服务..."); logging_error!(Type::Tray, true, tray::Tray::global().init()); if let Some(app_handle) = handle::Handle::global().app_handle() { logging!(info, Type::Tray, true, "创建系统托盘..."); let result = tray::Tray::global() .create_tray_from_handle(&app_handle) .await; if result.is_ok() { logging!(info, Type::Tray, true, "系统托盘创建成功"); } else if let Err(e) = result { logging!(error, Type::Tray, true, "系统托盘创建失败: {}", e); } } else { logging!( error, Type::Tray, true, "无法创建系统托盘: app_handle不存在" ); } // 更新系统代理 logging_error!( Type::System, true, sysopt::Sysopt::global().update_sysproxy().await ); logging_error!( Type::System, true, sysopt::Sysopt::global().init_guard_sysproxy() ); // 创建窗口 let is_silent_start = { Config::verge().await.latest_ref().enable_silent_start }.unwrap_or(false); #[cfg(target_os = "macos")] { if is_silent_start { handle::Handle::global().set_activation_policy_accessory(); } } create_window(!is_silent_start).await; // 初始化定时器 logging_error!(Type::System, true, timer::Timer::global().init().await); // 自动进入轻量模式 auto_lightweight_mode_init().await; logging_error!(Type::Tray, true, tray::Tray::global().update_part().await); logging!(trace, Type::System, true, "初始化热键..."); logging_error!(Type::System, true, hotkey::Hotkey::global().init().await); let elapsed = start_time.elapsed(); logging!( info, Type::Setup, true, "异步设置任务完成,耗时: {:?}", elapsed ); // 如果初始化时间过长,记录警告 if elapsed.as_secs() > 10 { logging!( warn, Type::Setup, true, "异步设置任务耗时较长({:?})", elapsed ); } } /// reset system proxy (异步) pub async fn resolve_reset_async() { #[cfg(target_os = "macos")] logging!(info, Type::Tray, true, "Unsubscribing from traffic updates"); #[cfg(target_os = "macos")] tray::Tray::global().unsubscribe_traffic(); logging_error!( Type::System, true, sysopt::Sysopt::global().reset_sysproxy().await ); logging_error!(Type::Core, true, CoreManager::global().stop_core().await); #[cfg(target_os = "macos")] { logging!(info, Type::System, true, "Restoring system DNS settings"); restore_public_dns().await; } } /// Create the main window pub async fn create_window(is_show: bool) -> bool { logging!( info, Type::Window, true, "开始创建/显示主窗口, is_show={}", is_show ); if !is_show { logging!(info, Type::Window, true, "静默模式启动时不创建窗口"); lightweight::set_lightweight_mode(true).await; handle::Handle::notify_startup_completed(); return false; } if let Some(app_handle) = handle::Handle::global().app_handle() { if let Some(window) = app_handle.get_webview_window("main") { logging!(info, Type::Window, true, "主窗口已存在,将显示现有窗口"); if is_show { if window.is_minimized().unwrap_or(false) { logging!(info, Type::Window, true, "窗口已最小化,正在取消最小化"); let _ = window.unminimize(); } let _ = window.show(); let _ = window.set_focus(); #[cfg(target_os = "macos")] handle::Handle::global().set_activation_policy_regular(); } return true; } } let creating_lock = get_window_creating_lock(); let mut creating = creating_lock.lock(); let (is_creating, last_time) = *creating; let elapsed = last_time.elapsed(); if is_creating && elapsed < Duration::from_secs(2) { logging!( info, Type::Window, true, "窗口创建请求被忽略,因为最近创建过 ({:?}ms)", elapsed.as_millis() ); return false; } *creating = (true, Instant::now()); // ScopeGuard 确保创建状态重置,防止 webview 卡死 let _guard = scopeguard::guard(creating, |mut creating_guard| { *creating_guard = (false, Instant::now()); logging!(debug, Type::Window, true, "[ScopeGuard] 窗口创建状态已重置"); }); let app_handle = match handle::Handle::global().app_handle() { Some(handle) => handle, None => { logging!( error, Type::Window, true, "无法获取app_handle,窗口创建失败" ); return false; } }; match tauri::WebviewWindowBuilder::new( &app_handle, "main", /* the unique window label */ tauri::WebviewUrl::App("index.html".into()), ) .title("Clash Verge") .center() .decorations(true) .fullscreen(false) .inner_size(DEFAULT_WIDTH as f64, DEFAULT_HEIGHT as f64) .min_inner_size(520.0, 520.0) .visible(true) // 立即显示窗口,避免用户等待 .initialization_script( r#" console.log('[Tauri] 窗口初始化脚本开始执行'); function createLoadingOverlay() { if (document.getElementById('initial-loading-overlay')) { console.log('[Tauri] 加载指示器已存在'); return; } console.log('[Tauri] 创建加载指示器'); const loadingDiv = document.createElement('div'); loadingDiv.id = 'initial-loading-overlay'; loadingDiv.innerHTML = `
Loading Clash Verge...
`; if (document.body) { document.body.appendChild(loadingDiv); } else { document.addEventListener('DOMContentLoaded', () => { if (document.body && !document.getElementById('initial-loading-overlay')) { document.body.appendChild(loadingDiv); } }); } } createLoadingOverlay(); if (document.readyState === 'loading') { document.addEventListener('DOMContentLoaded', createLoadingOverlay); } else { createLoadingOverlay(); } console.log('[Tauri] 窗口初始化脚本执行完成'); "#, ) .build() { Ok(newly_created_window) => { logging!(debug, Type::Window, true, "主窗口实例创建成功"); update_ui_ready_stage(UiReadyStage::NotStarted); AsyncHandler::spawn(move || async move { handle::Handle::global().mark_startup_completed(); logging!( debug, Type::Window, true, "异步窗口任务开始 (启动已标记完成)" ); // 先运行轻量模式检测 lightweight::run_once_auto_lightweight().await; // 发送启动完成事件,触发前端开始加载 logging!( debug, Type::Window, true, "发送 verge://startup-completed 事件" ); handle::Handle::notify_startup_completed(); if is_show { let window_clone = newly_created_window.clone(); // 立即显示窗口 let _ = window_clone.show(); let _ = window_clone.set_focus(); logging!(info, Type::Window, true, "窗口已立即显示"); #[cfg(target_os = "macos")] handle::Handle::global().set_activation_policy_regular(); let timeout_seconds = if crate::module::lightweight::is_in_lightweight_mode() { 3 } else { 8 }; logging!( info, Type::Window, true, "开始监控UI加载状态 (最多{}秒)...", timeout_seconds ); // 异步监控UI状态,使用try_read避免死锁 AsyncHandler::spawn(move || async move { logging!( debug, Type::Window, true, "启动UI状态监控线程,超时{}秒", timeout_seconds ); let ui_ready_checker = || async { let (mut check_count, mut consecutive_failures) = (0, 0); loop { let is_ready = get_ui_ready() .try_read() .map(|guard| *guard) .unwrap_or_else(|| { consecutive_failures += 1; if consecutive_failures > 50 { logging!( warn, Type::Window, true, "UI状态监控连续{}次无法获取读锁,可能存在死锁", consecutive_failures ); consecutive_failures = 0; } false }); if is_ready { logging!( debug, Type::Window, true, "UI状态监控检测到就绪信号,退出监控" ); return; } consecutive_failures = 0; tokio::time::sleep(Duration::from_millis(100)).await; check_count += 1; if check_count % 20 == 0 { logging!( debug, Type::Window, true, "UI加载状态检查... ({}秒)", check_count / 10 ); } } }; let wait_result = tokio::time::timeout( Duration::from_secs(timeout_seconds), ui_ready_checker(), ) .await; match wait_result { Ok(_) => { logging!(info, Type::Window, true, "UI已完全加载就绪"); handle::Handle::global() .get_window() .map(|window| window.eval(r" const overlay = document.getElementById('initial-loading-overlay'); if (overlay) { overlay.style.opacity = '0'; setTimeout(() => overlay.remove(), 300); } ")); } Err(_) => { logging!( warn, Type::Window, true, "UI加载监控超时({}秒),但窗口已正常显示", timeout_seconds ); get_ui_ready() .try_write() .map(|mut guard| { *guard = true; logging!( info, Type::Window, true, "超时后成功设置UI就绪状态" ); }) .unwrap_or_else(|| { logging!( error, Type::Window, true, "超时后无法获取UI状态写锁,可能存在严重死锁" ); }); } } }); logging!(info, Type::Window, true, "窗口显示流程完成"); } else { logging!( debug, Type::Window, true, "is_show为false,窗口保持隐藏状态" ); } }); true } Err(e) => { logging!(error, Type::Window, true, "主窗口构建失败: {}", e); false } } } pub async fn resolve_scheme(param: String) -> Result<()> { log::info!(target:"app", "received deep link: {param}"); let param_str = if param.starts_with("[") && param.len() > 4 { param .get(2..param.len() - 2) .ok_or_else(|| anyhow::anyhow!("Invalid string slice boundaries"))? } else { param.as_str() }; // 解析 URL let link_parsed = match Url::parse(param_str) { Ok(url) => url, Err(e) => { bail!("failed to parse deep link: {:?}, param: {:?}", e, param); } }; if link_parsed.scheme() == "clash" || link_parsed.scheme() == "clash-verge" { let name = link_parsed .query_pairs() .find(|(key, _)| key == "name") .map(|(_, value)| value.into_owned()); let url_param = if let Some(query) = link_parsed.query() { let prefix = "url="; if let Some(pos) = query.find(prefix) { let raw_url = &query[pos + prefix.len()..]; Some(percent_decode_str(raw_url).decode_utf8_lossy().to_string()) } else { None } } else { None }; match url_param { Some(url) => { log::info!(target:"app", "decoded subscription url: {url}"); create_window(false).await; match PrfItem::from_url(url.as_ref(), name, None, None).await { Ok(item) => { let uid = match item.uid.clone() { Some(uid) => uid, None => { logging!(error, Type::Config, true, "Profile item missing UID"); handle::Handle::notice_message( "import_sub_url::error", "Profile item missing UID".to_string(), ); return Ok(()); } }; let result = crate::config::profiles::profiles_append_item_safe(item).await; let _ = wrap_err!(result); handle::Handle::notice_message("import_sub_url::ok", uid); } Err(e) => { handle::Handle::notice_message("import_sub_url::error", e.to_string()); } } } None => bail!("failed to get profile url"), } } Ok(()) } #[cfg(target_os = "macos")] pub async fn set_public_dns(dns_server: String) { use crate::{core::handle, utils::dirs}; use tauri_plugin_shell::ShellExt; let app_handle = match handle::Handle::global().app_handle() { Some(handle) => handle, None => { log::error!(target: "app", "app_handle not available for DNS configuration"); return; } }; log::info!(target: "app", "try to set system dns"); let resource_dir = match dirs::app_resources_dir() { Ok(dir) => dir, Err(e) => { log::error!(target: "app", "Failed to get resource directory: {}", e); return; } }; let script = resource_dir.join("set_dns.sh"); if !script.exists() { log::error!(target: "app", "set_dns.sh not found"); return; } let script = script.to_string_lossy().into_owned(); match app_handle .shell() .command("bash") .args([script, dns_server]) .current_dir(resource_dir) .status() .await { Ok(status) => { if status.success() { log::info!(target: "app", "set system dns successfully"); } else { let code = status.code().unwrap_or(-1); log::error!(target: "app", "set system dns failed: {code}"); } } Err(err) => { log::error!(target: "app", "set system dns failed: {err}"); } } } #[cfg(target_os = "macos")] pub async fn restore_public_dns() { use crate::{core::handle, utils::dirs}; use tauri_plugin_shell::ShellExt; let app_handle = match handle::Handle::global().app_handle() { Some(handle) => handle, None => { log::error!(target: "app", "app_handle not available for DNS restoration"); return; } }; log::info!(target: "app", "try to unset system dns"); let resource_dir = match dirs::app_resources_dir() { Ok(dir) => dir, Err(e) => { log::error!(target: "app", "Failed to get resource directory: {}", e); return; } }; let script = resource_dir.join("unset_dns.sh"); if !script.exists() { log::error!(target: "app", "unset_dns.sh not found"); return; } let script = script.to_string_lossy().into_owned(); match app_handle .shell() .command("bash") .args([script]) .current_dir(resource_dir) .status() .await { Ok(status) => { if status.success() { log::info!(target: "app", "unset system dns successfully"); } else { let code = status.code().unwrap_or(-1); log::error!(target: "app", "unset system dns failed: {code}"); } } Err(err) => { log::error!(target: "app", "unset system dns failed: {err}"); } } }