123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233 |
- use crate::{
- core::{ClashInfo, ProfileItem, ProfilesConfig, VergeConfig},
- states::{ClashState, ProfilesState, VergeState},
- utils::{
- config::{read_clash, save_clash},
- fetch::fetch_profile,
- sysopt::SysProxyConfig,
- },
- };
- use serde_yaml::Mapping;
- use tauri::State;
- /// get all profiles from `profiles.yaml`
- /// do not acquire the lock of ProfileLock
- #[tauri::command]
- pub fn get_profiles(profiles: State<'_, ProfilesState>) -> Result<ProfilesConfig, String> {
- match profiles.0.lock() {
- Ok(profiles) => Ok(profiles.clone()),
- Err(_) => Err("failed to get profiles lock".into()),
- }
- }
- /// synchronize data irregularly
- #[tauri::command]
- pub fn sync_profiles(profiles: State<'_, ProfilesState>) -> Result<(), String> {
- match profiles.0.lock() {
- Ok(mut profiles) => profiles.sync_file(),
- Err(_) => Err("failed to get profiles lock".into()),
- }
- }
- /// Import the profile from url
- /// and save to `profiles.yaml`
- #[tauri::command]
- pub async fn import_profile(url: String, profiles: State<'_, ProfilesState>) -> Result<(), String> {
- match fetch_profile(&url).await {
- Some(result) => {
- let mut profiles = profiles.0.lock().unwrap();
- profiles.import_from_url(url, result)
- }
- None => Err(format!("failed to fetch profile from `{}`", url)),
- }
- }
- /// Update the profile
- /// and save to `profiles.yaml`
- /// http request firstly
- /// then acquire the lock of `profiles.yaml`
- #[tauri::command]
- pub async fn update_profile(
- index: usize,
- clash: State<'_, ClashState>,
- profiles: State<'_, ProfilesState>,
- ) -> Result<(), String> {
- // maybe we can get the url from the web app directly
- let url = match profiles.0.lock() {
- Ok(mut profile) => {
- let items = profile.items.take().unwrap_or(vec![]);
- if index >= items.len() {
- return Err("the index out of bound".into());
- }
- let url = match &items[index].url {
- Some(u) => u.clone(),
- None => return Err("failed to update profile for `invalid url`".into()),
- };
- profile.items = Some(items);
- url
- }
- Err(_) => return Err("failed to get profiles lock".into()),
- };
- match fetch_profile(&url).await {
- Some(result) => match profiles.0.lock() {
- Ok(mut profiles) => {
- profiles.update_item(index, result)?;
- // reactivate the profile
- let current = profiles.current.clone().unwrap_or(0);
- if current == index {
- let clash = clash.0.lock().unwrap();
- profiles.activate(clash.info.clone())
- } else {
- Ok(())
- }
- }
- Err(_) => Err("failed to get profiles lock".into()),
- },
- None => Err(format!("failed to fetch profile from `{}`", url)),
- }
- }
- /// change the current profile
- #[tauri::command]
- pub fn select_profile(
- index: usize,
- clash: State<'_, ClashState>,
- profiles: State<'_, ProfilesState>,
- ) -> Result<(), String> {
- let mut profiles = profiles.0.lock().unwrap();
- match profiles.put_current(index) {
- Ok(()) => {
- let clash = clash.0.lock().unwrap();
- profiles.activate(clash.info.clone())
- }
- Err(err) => Err(err),
- }
- }
- /// delete profile item
- #[tauri::command]
- pub fn delete_profile(
- index: usize,
- clash_state: State<'_, ClashState>,
- profiles_state: State<'_, ProfilesState>,
- ) -> Result<(), String> {
- let mut profiles = profiles_state.0.lock().unwrap();
- match profiles.delete_item(index) {
- Ok(change) => match change {
- true => {
- let clash = clash_state.0.lock().unwrap();
- profiles.activate(clash.info.clone())
- }
- false => Ok(()),
- },
- Err(err) => Err(err),
- }
- }
- /// patch the profile config
- #[tauri::command]
- pub fn patch_profile(
- index: usize,
- profile: ProfileItem,
- profiles: State<'_, ProfilesState>,
- ) -> Result<(), String> {
- match profiles.0.lock() {
- Ok(mut profiles) => profiles.patch_item(index, profile),
- Err(_) => Err("can not get profiles lock".into()),
- }
- }
- /// restart the sidecar
- #[tauri::command]
- pub fn restart_sidecar(
- clash_state: State<'_, ClashState>,
- profiles_state: State<'_, ProfilesState>,
- ) -> Result<(), String> {
- let mut clash = clash_state.0.lock().unwrap();
- match clash.restart_sidecar() {
- Ok(_) => {
- let profiles = profiles_state.0.lock().unwrap();
- match profiles.activate(clash.info.clone()) {
- Ok(()) => Ok(()),
- Err(err) => {
- log::error!("{}", err);
- Err(err)
- }
- }
- }
- Err(err) => {
- log::error!("{}", err);
- Err(err)
- }
- }
- }
- /// get the clash core info from the state
- /// the caller can also get the infomation by clash's api
- #[tauri::command]
- pub fn get_clash_info(clash_state: State<'_, ClashState>) -> Result<ClashInfo, String> {
- match clash_state.0.lock() {
- Ok(arc) => Ok(arc.info.clone()),
- Err(_) => Err("failed to get clash lock".into()),
- }
- }
- /// todo: need refactor
- /// update the clash core config
- /// after putting the change to the clash core
- /// then we should save the latest config
- #[tauri::command]
- pub fn patch_clash_config(payload: Mapping) -> Result<(), String> {
- let mut config = read_clash();
- for (key, value) in payload.iter() {
- if config.contains_key(key) {
- config[key] = value.clone();
- } else {
- config.insert(key.clone(), value.clone());
- }
- }
- save_clash(&config)
- }
- /// get the system proxy
- #[tauri::command]
- pub fn get_sys_proxy() -> Result<SysProxyConfig, String> {
- match SysProxyConfig::get_sys() {
- Ok(value) => Ok(value),
- Err(err) => Err(err.to_string()),
- }
- }
- /// get the current proxy config
- /// which may not the same as system proxy
- #[tauri::command]
- pub fn get_cur_proxy(verge_state: State<'_, VergeState>) -> Result<Option<SysProxyConfig>, String> {
- match verge_state.0.lock() {
- Ok(verge) => Ok(verge.cur_sysproxy.clone()),
- Err(_) => Err("failed to get verge lock".into()),
- }
- }
- /// get the verge config
- #[tauri::command]
- pub fn get_verge_config(verge_state: State<'_, VergeState>) -> Result<VergeConfig, String> {
- match verge_state.0.lock() {
- Ok(arc) => Ok(arc.config.clone()),
- Err(_) => Err("failed to get verge lock".into()),
- }
- }
- /// patch the verge config
- /// this command only save the config and not responsible for other things
- #[tauri::command]
- pub async fn patch_verge_config(
- payload: VergeConfig,
- verge_state: State<'_, VergeState>,
- ) -> Result<(), String> {
- let mut verge = verge_state.0.lock().unwrap();
- verge.patch_config(payload)
- }
|