use super::{clash_api, logger::Logger}; use crate::log_err; use crate::{config::*, utils::dirs}; use anyhow::{bail, Context, Result}; use once_cell::sync::OnceCell; use parking_lot::Mutex; use std::{fs, io::Write, sync::Arc, time::Duration}; use sysinfo::{Pid, PidExt, ProcessExt, System, SystemExt}; use tauri::api::process::{Command, CommandChild, CommandEvent}; use tokio::time::sleep; #[derive(Debug)] pub struct CoreManager { sidecar: Arc>>, #[allow(unused)] use_service_mode: Arc>, } impl CoreManager { pub fn global() -> &'static CoreManager { static CORE_MANAGER: OnceCell = OnceCell::new(); CORE_MANAGER.get_or_init(|| CoreManager { sidecar: Arc::new(Mutex::new(None)), use_service_mode: Arc::new(Mutex::new(false)), }) } pub fn init(&self) -> Result<()> { // kill old clash process let _ = dirs::clash_pid_path() .and_then(|path| fs::read(path).map(|p| p.to_vec()).context("")) .and_then(|pid| String::from_utf8_lossy(&pid).parse().context("")) .map(|pid| { let mut system = System::new(); system.refresh_all(); system.process(Pid::from_u32(pid)).map(|proc| { if proc.name().contains("clash") { proc.kill(); } }); }); tauri::async_runtime::spawn(async { // 启动clash log_err!(Self::global().run_core().await); }); Ok(()) } /// 检查配置是否正确 pub fn check_config(&self) -> Result<()> { let config_path = Config::generate_file(ConfigType::Check)?; let config_path = dirs::path_to_str(&config_path)?; let clash_core = { Config::verge().latest().clash_core.clone() }; let clash_core = clash_core.unwrap_or("clash".into()); let output = Command::new_sidecar(clash_core)? .args(["-t", "-f", config_path]) .output()?; if !output.status.success() { Logger::global().set_log(output.stdout.clone()); bail!("{}", output.stdout); // 过滤掉终端颜色值 } Ok(()) } /// 启动核心 pub async fn run_core(&self) -> Result<()> { let config_path = Config::generate_file(ConfigType::Run)?; #[cfg(target_os = "windows")] { use super::win_service; // 服务模式 let enable = { let enable = Config::verge().data().enable_service_mode.clone(); enable.unwrap_or(false) }; *self.use_service_mode.lock() = enable; if enable { // 服务模式启动失败就直接运行sidecar match { win_service::check_service().await?; win_service::run_core_by_service().await } { Ok(_) => return Ok(()), Err(err) => { // 修改这个值,免得stop出错 *self.use_service_mode.lock() = false; log::error!(target: "app", "{err}"); } } } } let mut sidecar = self.sidecar.lock(); if let Some(child) = sidecar.take() { let _ = child.kill(); } let app_dir = dirs::app_home_dir()?; let app_dir = dirs::path_to_str(&app_dir)?; let clash_core = { Config::verge().latest().clash_core.clone() }; let clash_core = clash_core.unwrap_or("clash".into()); let config_path = dirs::path_to_str(&config_path)?; // fix #212 let args = match clash_core.as_str() { "clash-meta" => vec!["-m", "-d", app_dir, "-f", config_path], _ => vec!["-d", app_dir, "-f", config_path], }; let cmd = Command::new_sidecar(clash_core)?; let (mut rx, cmd_child) = cmd.args(args).spawn()?; // 将pid写入文件中 crate::log_err!({ let pid = cmd_child.pid(); let path = dirs::clash_pid_path()?; fs::File::create(path) .context("failed to create the pid file")? .write(format!("{pid}").as_bytes()) .context("failed to write pid to the file")?; >::Ok(()) }); *sidecar = Some(cmd_child); tauri::async_runtime::spawn(async move { while let Some(event) = rx.recv().await { match event { CommandEvent::Stdout(line) => { let can_short = line.starts_with("time=") && line.len() > 33; let stdout = if can_short { &line[33..] } else { &line }; log::info!(target: "app" ,"[clash]: {}", stdout); Logger::global().set_log(line); } CommandEvent::Stderr(err) => { log::error!(target: "app" ,"[clash]: {err}"); Logger::global().set_log(err); } CommandEvent::Error(err) => { log::error!(target: "app" ,"[clash]: {err}"); Logger::global().set_log(err); } CommandEvent::Terminated(_) => { log::info!(target: "app" ,"clash core terminated"); break; } _ => {} } } }); Ok(()) } /// 停止核心运行 pub fn stop_core(&self) -> Result<()> { #[cfg(target_os = "windows")] if *self.use_service_mode.lock() { tauri::async_runtime::block_on(async move { log_err!(super::win_service::stop_core_by_service().await); }); return Ok(()); } let mut sidecar = self.sidecar.lock(); if let Some(child) = sidecar.take() { let _ = child.kill(); } Ok(()) } /// 切换核心 pub async fn change_core(&self, clash_core: Option) -> Result<()> { let clash_core = clash_core.ok_or(anyhow::anyhow!("clash core is null"))?; if &clash_core != "clash" && &clash_core != "clash-meta" { bail!("invalid clash core name \"{clash_core}\""); } Config::verge().draft().clash_core = Some(clash_core); // 清掉旧日志 Logger::global().clear_log(); match self.run_core().await { Ok(_) => { Config::verge().apply(); Config::runtime().apply(); log_err!(Config::verge().latest().save_file()); Ok(()) } Err(err) => { Config::verge().discard(); Config::runtime().discard(); Err(err) } } } /// 更新proxies那些 /// 如果涉及端口和外部控制则需要重启 pub async fn update_config(&self) -> Result<()> { // 更新配置 Config::generate()?; // 检查配置是否正常 self.check_config()?; // 更新运行时配置 let path = Config::generate_file(ConfigType::Run)?; let path = dirs::path_to_str(&path)?; // 发送请求 发送5次 for i in 0..5 { match clash_api::put_configs(path).await { Ok(_) => break, Err(err) => { if i < 4 { log::error!(target: "app", "{err}"); } else { bail!(err); } } } sleep(Duration::from_millis(250)).await; } Ok(()) } }