use crate::{APP_HANDLE, singleton, constants::timing}; use parking_lot::RwLock; use std::{sync::Arc, thread}; use tauri::{AppHandle, Manager, WebviewWindow}; use tauri_plugin_mihomo::{Mihomo, MihomoExt}; use tokio::sync::RwLockReadGuard; use super::notification::{ErrorMessage, FrontendEvent, NotificationSystem}; #[derive(Debug, Clone)] pub struct Handle { is_exiting: Arc>, startup_errors: Arc>>, startup_completed: Arc>, pub(crate) notification_system: Arc>>, } impl Default for Handle { fn default() -> Self { Self { is_exiting: Arc::new(RwLock::new(false)), startup_errors: Arc::new(RwLock::new(Vec::new())), startup_completed: Arc::new(RwLock::new(false)), notification_system: Arc::new(RwLock::new(Some(NotificationSystem::new()))), } } } singleton!(Handle, HANDLE); impl Handle { pub fn new() -> Self { Self::default() } pub fn init(&self) { if self.is_exiting() { return; } let mut system_opt = self.notification_system.write(); if let Some(system) = system_opt.as_mut() { if !system.is_running { system.start(); } } } pub fn app_handle() -> &'static AppHandle { APP_HANDLE.get().expect("App handle not initialized") } pub async fn mihomo() -> RwLockReadGuard<'static, Mihomo> { Self::app_handle().mihomo().read().await } pub fn get_window() -> Option { Self::app_handle().get_webview_window("main") } pub fn refresh_clash() { let handle = Self::global(); if handle.is_exiting() { return; } let system_opt = handle.notification_system.read(); if let Some(system) = system_opt.as_ref() { system.send_event(FrontendEvent::RefreshClash); } } pub fn refresh_verge() { let handle = Self::global(); if handle.is_exiting() { return; } let system_opt = handle.notification_system.read(); if let Some(system) = system_opt.as_ref() { system.send_event(FrontendEvent::RefreshVerge); } } pub fn notify_profile_changed(profile_id: String) { Self::send_event(FrontendEvent::ProfileChanged { current_profile_id: profile_id, }); } pub fn notify_timer_updated(profile_index: String) { Self::send_event(FrontendEvent::TimerUpdated { profile_index }); } pub fn notify_profile_update_started(uid: String) { Self::send_event(FrontendEvent::ProfileUpdateStarted { uid }); } pub fn notify_profile_update_completed(uid: String) { Self::send_event(FrontendEvent::ProfileUpdateCompleted { uid }); } pub fn notice_message, M: Into>(status: S, msg: M) { let handle = Self::global(); let status_str = status.into(); let msg_str = msg.into(); if !*handle.startup_completed.read() { let mut errors = handle.startup_errors.write(); errors.push(ErrorMessage { status: status_str, message: msg_str, }); return; } if handle.is_exiting() { return; } Self::send_event(FrontendEvent::NoticeMessage { status: status_str, message: msg_str, }); } fn send_event(event: FrontendEvent) { let handle = Self::global(); if handle.is_exiting() { return; } let system_opt = handle.notification_system.read(); if let Some(system) = system_opt.as_ref() { system.send_event(event); } } pub fn mark_startup_completed(&self) { *self.startup_completed.write() = true; self.send_startup_errors(); } fn send_startup_errors(&self) { let errors = { let mut errors = self.startup_errors.write(); std::mem::take(&mut *errors) }; if errors.is_empty() { return; } let _ = thread::Builder::new() .name("startup-errors-sender".into()) .spawn(move || { thread::sleep(timing::STARTUP_ERROR_DELAY); let handle = Handle::global(); if handle.is_exiting() { return; } let system_opt = handle.notification_system.read(); if let Some(system) = system_opt.as_ref() { for error in errors { if handle.is_exiting() { break; } system.send_event(FrontendEvent::NoticeMessage { status: error.status, message: error.message, }); thread::sleep(timing::ERROR_BATCH_DELAY); } } }); } pub fn set_is_exiting(&self) { *self.is_exiting.write() = true; let mut system_opt = self.notification_system.write(); if let Some(system) = system_opt.as_mut() { system.shutdown(); } } pub fn is_exiting(&self) -> bool { *self.is_exiting.read() } } #[cfg(target_os = "macos")] impl Handle { pub fn set_activation_policy(&self, policy: tauri::ActivationPolicy) -> Result<(), String> { Self::app_handle() .set_activation_policy(policy) .map_err(|e| e.to_string()) } pub fn set_activation_policy_regular(&self) { let _ = self.set_activation_policy(tauri::ActivationPolicy::Regular); } pub fn set_activation_policy_accessory(&self) { let _ = self.set_activation_policy(tauri::ActivationPolicy::Accessory); } }