use super::CmdResult; use super::StringifyErr; use crate::{ config::{ Config, IProfiles, PrfItem, PrfOption, profiles::{ profiles_append_item_with_filedata_safe, profiles_delete_item_safe, profiles_patch_item_safe, profiles_reorder_safe, profiles_save_file_safe, }, profiles_append_item_safe, }, core::{CoreManager, handle, timer::Timer, tray::Tray}, feat, logging, process::AsyncHandler, ret_err, utils::{dirs, help, logging::Type}, }; use smartstring::alias::String; use std::sync::atomic::{AtomicBool, AtomicU64, Ordering}; use std::time::Duration; // 全局请求序列号跟踪,用于避免队列化执行 static CURRENT_REQUEST_SEQUENCE: AtomicU64 = AtomicU64::new(0); static CURRENT_SWITCHING_PROFILE: AtomicBool = AtomicBool::new(false); #[tauri::command] pub async fn get_profiles() -> CmdResult { // 策略1: 尝试快速获取latest数据 let latest_result = tokio::time::timeout(Duration::from_millis(500), async { let profiles = Config::profiles().await; let latest = profiles.latest_ref(); IProfiles { current: latest.current.clone(), items: latest.items.clone(), } }) .await; match latest_result { Ok(profiles) => { logging!(info, Type::Cmd, "快速获取配置列表成功"); return Ok(profiles); } Err(_) => { logging!(warn, Type::Cmd, "快速获取配置超时(500ms)"); } } // 策略2: 如果快速获取失败,尝试获取data() let data_result = tokio::time::timeout(Duration::from_secs(2), async { let profiles = Config::profiles().await; let data = profiles.latest_ref(); IProfiles { current: data.current.clone(), items: data.items.clone(), } }) .await; match data_result { Ok(profiles) => { logging!(info, Type::Cmd, "获取draft配置列表成功"); return Ok(profiles); } Err(join_err) => { logging!( error, Type::Cmd, "获取draft配置任务失败或超时: {}", join_err ); } } // 策略3: fallback,尝试重新创建配置 logging!(warn, Type::Cmd, "所有获取配置策略都失败,尝试fallback"); Ok(IProfiles::new().await) } /// 增强配置文件 #[tauri::command] pub async fn enhance_profiles() -> CmdResult { match feat::enhance_profiles().await { Ok(_) => {} Err(e) => { logging!(error, Type::Cmd, "{}", e); return Err(e.to_string().into()); } } handle::Handle::refresh_clash(); Ok(()) } /// 导入配置文件 #[tauri::command] pub async fn import_profile(url: std::string::String, option: Option) -> CmdResult { logging!(info, Type::Cmd, "[导入订阅] 开始导入: {}", url); // 直接依赖 PrfItem::from_url 自身的超时/重试逻辑,不再使用 tokio::time::timeout 包裹 let item = &mut match PrfItem::from_url(&url, None, None, option.as_ref()).await { Ok(it) => { logging!(info, Type::Cmd, "[导入订阅] 下载完成,开始保存配置"); it } Err(e) => { logging!(error, Type::Cmd, "[导入订阅] 下载失败: {}", e); return Err(format!("导入订阅失败: {}", e).into()); } }; match profiles_append_item_safe(item).await { Ok(_) => match profiles_save_file_safe().await { Ok(_) => { logging!(info, Type::Cmd, "[导入订阅] 配置文件保存成功"); } Err(e) => { logging!(error, Type::Cmd, "[导入订阅] 保存配置文件失败: {}", e); } }, Err(e) => { logging!(error, Type::Cmd, "[导入订阅] 保存配置失败: {}", e); return Err(format!("导入订阅失败: {}", e).into()); } } // 立即发送配置变更通知 if let Some(uid) = &item.uid { logging!(info, Type::Cmd, "[导入订阅] 发送配置变更通知: {}", uid); handle::Handle::notify_profile_changed(uid.clone()); } // 异步保存配置文件并发送全局通知 let uid_clone = item.uid.clone(); if let Some(uid) = uid_clone { // 延迟发送,确保文件已完全写入 tokio::time::sleep(Duration::from_millis(100)).await; handle::Handle::notify_profile_changed(uid); } logging!(info, Type::Cmd, "[导入订阅] 导入完成: {}", url); Ok(()) } /// 调整profile的顺序 #[tauri::command] pub async fn reorder_profile(active_id: String, over_id: String) -> CmdResult { match profiles_reorder_safe(&active_id, &over_id).await { Ok(_) => { logging!(info, Type::Cmd, "重新排序配置文件"); Ok(()) } Err(err) => { logging!(error, Type::Cmd, "重新排序配置文件失败: {}", err); Err(format!("重新排序配置文件失败: {}", err).into()) } } } /// 创建新的profile /// 创建一个新的配置文件 #[tauri::command] pub async fn create_profile(item: PrfItem, file_data: Option) -> CmdResult { match profiles_append_item_with_filedata_safe(&item, file_data).await { Ok(_) => { // 发送配置变更通知 if let Some(uid) = &item.uid { logging!(info, Type::Cmd, "[创建订阅] 发送配置变更通知: {}", uid); handle::Handle::notify_profile_changed(uid.clone()); } Ok(()) } Err(err) => match err.to_string().as_str() { "the file already exists" => Err("the file already exists".into()), _ => Err(format!("add profile error: {err}").into()), }, } } /// 更新配置文件 #[tauri::command] pub async fn update_profile(index: String, option: Option) -> CmdResult { match feat::update_profile(&index, option.as_ref(), true, true).await { Ok(_) => Ok(()), Err(e) => { logging!(error, Type::Cmd, "{}", e); Err(e.to_string().into()) } } } /// 删除配置文件 #[tauri::command] pub async fn delete_profile(index: String) -> CmdResult { println!("delete_profile: {}", index); // 使用Send-safe helper函数 let should_update = profiles_delete_item_safe(&index).await.stringify_err()?; profiles_save_file_safe().await.stringify_err()?; if should_update { match CoreManager::global().update_config().await { Ok(_) => { handle::Handle::refresh_clash(); // 发送配置变更通知 logging!(info, Type::Cmd, "[删除订阅] 发送配置变更通知: {}", index); handle::Handle::notify_profile_changed(index); } Err(e) => { logging!(error, Type::Cmd, "{}", e); return Err(e.to_string().into()); } } } Ok(()) } /// 验证新配置文件的语法 async fn validate_new_profile(new_profile: &String) -> Result<(), ()> { logging!(info, Type::Cmd, "正在切换到新配置: {}", new_profile); // 获取目标配置文件路径 let config_file_result = { let profiles_config = Config::profiles().await; let profiles_data = profiles_config.latest_ref(); match profiles_data.get_item(new_profile) { Ok(item) => { if let Some(file) = &item.file { let path = dirs::app_profiles_dir().map(|dir| dir.join(file.as_str())); path.ok() } else { None } } Err(e) => { logging!(error, Type::Cmd, "获取目标配置信息失败: {}", e); None } } }; // 如果获取到文件路径,检查YAML语法 if let Some(file_path) = config_file_result { if !file_path.exists() { logging!( error, Type::Cmd, "目标配置文件不存在: {}", file_path.display() ); handle::Handle::notice_message( "config_validate::file_not_found", format!("{}", file_path.display()), ); return Err(()); } // 超时保护 let file_read_result = tokio::time::timeout( Duration::from_secs(5), tokio::fs::read_to_string(&file_path), ) .await; match file_read_result { Ok(Ok(content)) => { let yaml_parse_result = AsyncHandler::spawn_blocking(move || { serde_yaml_ng::from_str::(&content) }) .await; match yaml_parse_result { Ok(Ok(_)) => { logging!(info, Type::Cmd, "目标配置文件语法正确"); Ok(()) } Ok(Err(err)) => { let error_msg = format!(" {err}"); logging!( error, Type::Cmd, "目标配置文件存在YAML语法错误:{}", error_msg ); handle::Handle::notice_message( "config_validate::yaml_syntax_error", error_msg.clone(), ); Err(()) } Err(join_err) => { let error_msg = format!("YAML解析任务失败: {join_err}"); logging!(error, Type::Cmd, "{}", error_msg); handle::Handle::notice_message( "config_validate::yaml_parse_error", error_msg.clone(), ); Err(()) } } } Ok(Err(err)) => { let error_msg = format!("无法读取目标配置文件: {err}"); logging!(error, Type::Cmd, "{}", error_msg); handle::Handle::notice_message( "config_validate::file_read_error", error_msg.clone(), ); Err(()) } Err(_) => { let error_msg = "读取配置文件超时(5秒)".to_string(); logging!(error, Type::Cmd, "{}", error_msg); handle::Handle::notice_message( "config_validate::file_read_timeout", error_msg.clone(), ); Err(()) } } } else { Ok(()) } } /// 执行配置更新并处理结果 async fn restore_previous_profile(prev_profile: String) -> CmdResult<()> { logging!(info, Type::Cmd, "尝试恢复到之前的配置: {}", prev_profile); let restore_profiles = IProfiles { current: Some(prev_profile), items: None, }; Config::profiles() .await .draft_mut() .patch_config(restore_profiles) .stringify_err()?; Config::profiles().await.apply(); crate::process::AsyncHandler::spawn(|| async move { if let Err(e) = profiles_save_file_safe().await { logging!(warn, Type::Cmd, "Warning: 异步保存恢复配置文件失败: {e}"); } }); logging!(info, Type::Cmd, "成功恢复到之前的配置"); Ok(()) } async fn handle_success(current_sequence: u64, current_value: Option) -> CmdResult { let latest_sequence = CURRENT_REQUEST_SEQUENCE.load(Ordering::SeqCst); if current_sequence < latest_sequence { logging!( info, Type::Cmd, "内核操作后发现更新的请求 (序列号: {} < {}),忽略当前结果", current_sequence, latest_sequence ); Config::profiles().await.discard(); return Ok(false); } logging!( info, Type::Cmd, "配置更新成功,序列号: {}", current_sequence ); Config::profiles().await.apply(); handle::Handle::refresh_clash(); if let Err(e) = Tray::global().update_tooltip().await { logging!(warn, Type::Cmd, "Warning: 异步更新托盘提示失败: {e}"); } if let Err(e) = Tray::global().update_menu().await { logging!(warn, Type::Cmd, "Warning: 异步更新托盘菜单失败: {e}"); } if let Err(e) = profiles_save_file_safe().await { logging!(warn, Type::Cmd, "Warning: 异步保存配置文件失败: {e}"); } if let Some(current) = ¤t_value { logging!( info, Type::Cmd, "向前端发送配置变更事件: {}, 序列号: {}", current, current_sequence ); handle::Handle::notify_profile_changed(current.clone()); } CURRENT_SWITCHING_PROFILE.store(false, Ordering::SeqCst); Ok(true) } async fn handle_validation_failure( error_msg: String, current_profile: Option, ) -> CmdResult { logging!(warn, Type::Cmd, "配置验证失败: {}", error_msg); Config::profiles().await.discard(); if let Some(prev_profile) = current_profile { restore_previous_profile(prev_profile).await?; } handle::Handle::notice_message("config_validate::error", error_msg); CURRENT_SWITCHING_PROFILE.store(false, Ordering::SeqCst); Ok(false) } async fn handle_update_error(e: E, current_sequence: u64) -> CmdResult { logging!( warn, Type::Cmd, "更新过程发生错误: {}, 序列号: {}", e, current_sequence ); Config::profiles().await.discard(); handle::Handle::notice_message("config_validate::boot_error", e.to_string()); CURRENT_SWITCHING_PROFILE.store(false, Ordering::SeqCst); Ok(false) } async fn handle_timeout(current_profile: Option, current_sequence: u64) -> CmdResult { let timeout_msg = "配置更新超时(30秒),可能是配置验证或核心通信阻塞"; logging!( error, Type::Cmd, "{}, 序列号: {}", timeout_msg, current_sequence ); Config::profiles().await.discard(); if let Some(prev_profile) = current_profile { restore_previous_profile(prev_profile).await?; } handle::Handle::notice_message("config_validate::timeout", timeout_msg); CURRENT_SWITCHING_PROFILE.store(false, Ordering::SeqCst); Ok(false) } async fn perform_config_update( current_sequence: u64, current_value: Option, current_profile: Option, ) -> CmdResult { logging!( info, Type::Cmd, "开始内核配置更新,序列号: {}", current_sequence ); let update_result = tokio::time::timeout( Duration::from_secs(30), CoreManager::global().update_config(), ) .await; match update_result { Ok(Ok((true, _))) => handle_success(current_sequence, current_value).await, Ok(Ok((false, error_msg))) => handle_validation_failure(error_msg, current_profile).await, Ok(Err(e)) => handle_update_error(e, current_sequence).await, Err(_) => handle_timeout(current_profile, current_sequence).await, } } /// 修改profiles的配置 #[tauri::command] pub async fn patch_profiles_config(profiles: IProfiles) -> CmdResult { if CURRENT_SWITCHING_PROFILE.load(Ordering::SeqCst) { logging!(info, Type::Cmd, "当前正在切换配置,放弃请求"); return Ok(false); } CURRENT_SWITCHING_PROFILE.store(true, Ordering::SeqCst); // 为当前请求分配序列号 let current_sequence = CURRENT_REQUEST_SEQUENCE.fetch_add(1, Ordering::SeqCst) + 1; let target_profile = profiles.current.clone(); logging!( info, Type::Cmd, "开始修改配置文件,请求序列号: {}, 目标profile: {:?}", current_sequence, target_profile ); let latest_sequence = CURRENT_REQUEST_SEQUENCE.load(Ordering::SeqCst); if current_sequence < latest_sequence { logging!( info, Type::Cmd, "获取锁后发现更新的请求 (序列号: {} < {}),放弃当前请求", current_sequence, latest_sequence ); return Ok(false); } // 保存当前配置,以便在验证失败时恢复 let current_profile = Config::profiles().await.latest_ref().current.clone(); logging!(info, Type::Cmd, "当前配置: {:?}", current_profile); // 如果要切换配置,先检查目标配置文件是否有语法错误 if let Some(new_profile) = profiles.current.as_ref() && current_profile.as_ref() != Some(new_profile) && validate_new_profile(new_profile).await.is_err() { CURRENT_SWITCHING_PROFILE.store(false, Ordering::SeqCst); return Ok(false); } // 检查请求有效性 let latest_sequence = CURRENT_REQUEST_SEQUENCE.load(Ordering::SeqCst); if current_sequence < latest_sequence { logging!( info, Type::Cmd, "在核心操作前发现更新的请求 (序列号: {} < {}),放弃当前请求", current_sequence, latest_sequence ); return Ok(false); } // 更新profiles配置 logging!( info, Type::Cmd, "正在更新配置草稿,序列号: {}", current_sequence ); let current_value = profiles.current.clone(); let _ = Config::profiles().await.draft_mut().patch_config(profiles); // 在调用内核前再次验证请求有效性 let latest_sequence = CURRENT_REQUEST_SEQUENCE.load(Ordering::SeqCst); if current_sequence < latest_sequence { logging!( info, Type::Cmd, "在内核交互前发现更新的请求 (序列号: {} < {}),放弃当前请求", current_sequence, latest_sequence ); Config::profiles().await.discard(); return Ok(false); } perform_config_update(current_sequence, current_value, current_profile).await } /// 根据profile name修改profiles #[tauri::command] pub async fn patch_profiles_config_by_profile_index(profile_index: String) -> CmdResult { logging!(info, Type::Cmd, "切换配置到: {}", profile_index); let profiles = IProfiles { current: Some(profile_index), items: None, }; patch_profiles_config(profiles).await } /// 修改某个profile item的 #[tauri::command] pub async fn patch_profile(index: String, profile: PrfItem) -> CmdResult { // 保存修改前检查是否有更新 update_interval let profiles = Config::profiles().await; let should_refresh_timer = if let Ok(old_profile) = profiles.latest_ref().get_item(&index) { let old_interval = old_profile.option.as_ref().and_then(|o| o.update_interval); let new_interval = profile.option.as_ref().and_then(|o| o.update_interval); let old_allow_auto_update = old_profile .option .as_ref() .and_then(|o| o.allow_auto_update); let new_allow_auto_update = profile.option.as_ref().and_then(|o| o.allow_auto_update); (old_interval != new_interval) || (old_allow_auto_update != new_allow_auto_update) } else { false }; profiles_patch_item_safe(&index, &profile) .await .stringify_err()?; // 如果更新间隔或允许自动更新变更,异步刷新定时器 if should_refresh_timer { let index_clone = index.clone(); crate::process::AsyncHandler::spawn(move || async move { logging!(info, Type::Timer, "定时器更新间隔已变更,正在刷新定时器..."); if let Err(e) = crate::core::Timer::global().refresh().await { logging!(error, Type::Timer, "刷新定时器失败: {}", e); } else { // 刷新成功后发送自定义事件,不触发配置重载 crate::core::handle::Handle::notify_timer_updated(index_clone); } }); } Ok(()) } /// 查看配置文件 #[tauri::command] pub async fn view_profile(index: String) -> CmdResult { let profiles = Config::profiles().await; let profiles_ref = profiles.latest_ref(); let file = profiles_ref .get_item(&index) .stringify_err()? .file .clone() .ok_or("the file field is null")?; let path = dirs::app_profiles_dir() .stringify_err()? .join(file.as_str()); if !path.exists() { ret_err!("the file not found"); } help::open_file(path).stringify_err() } /// 读取配置文件内容 #[tauri::command] pub async fn read_profile_file(index: String) -> CmdResult { let item = { let profiles = Config::profiles().await; let profiles_ref = profiles.latest_ref(); PrfItem { file: profiles_ref.get_item(&index).stringify_err()?.file.clone(), ..Default::default() } }; let data = item.read_file().await.stringify_err()?; Ok(data) } /// 获取下一次更新时间 #[tauri::command] pub async fn get_next_update_time(uid: String) -> CmdResult> { let timer = Timer::global(); let next_time = timer.get_next_update_time(&uid).await; Ok(next_time) }