use crate::utils::{dirs, help, tmpl}; use anyhow::{bail, Context, Result}; use reqwest::StatusCode; use serde::{Deserialize, Serialize}; use serde_yaml::Mapping; use std::fs; use sysproxy::Sysproxy; use super::Config; #[derive(Debug, Clone, Deserialize, Serialize)] pub struct PrfItem { pub uid: Option, /// profile item type /// enum value: remote | local | script | merge #[serde(rename = "type")] pub itype: Option, /// profile name pub name: Option, /// profile file pub file: Option, /// profile description #[serde(skip_serializing_if = "Option::is_none")] pub desc: Option, /// source url #[serde(skip_serializing_if = "Option::is_none")] pub url: Option, /// selected information #[serde(skip_serializing_if = "Option::is_none")] pub selected: Option>, /// subscription user info #[serde(skip_serializing_if = "Option::is_none")] pub extra: Option, /// updated time pub updated: Option, /// some options of the item #[serde(skip_serializing_if = "Option::is_none")] pub option: Option, /// the file data #[serde(skip)] pub file_data: Option, } #[derive(Default, Debug, Clone, Deserialize, Serialize)] pub struct PrfSelected { pub name: Option, pub now: Option, } #[derive(Default, Debug, Clone, Copy, Deserialize, Serialize)] pub struct PrfExtra { pub upload: usize, pub download: usize, pub total: usize, pub expire: usize, } #[derive(Default, Debug, Clone, Deserialize, Serialize, PartialEq, Eq)] pub struct PrfOption { /// for `remote` profile's http request /// see issue #13 #[serde(skip_serializing_if = "Option::is_none")] pub user_agent: Option, /// for `remote` profile /// use system proxy #[serde(skip_serializing_if = "Option::is_none")] pub with_proxy: Option, /// for `remote` profile /// use self proxy #[serde(skip_serializing_if = "Option::is_none")] pub self_proxy: Option, #[serde(skip_serializing_if = "Option::is_none")] pub update_interval: Option, } impl PrfOption { pub fn merge(one: Option, other: Option) -> Option { match (one, other) { (Some(mut a), Some(b)) => { a.user_agent = b.user_agent.or(a.user_agent); a.with_proxy = b.with_proxy.or(a.with_proxy); a.self_proxy = b.self_proxy.or(a.self_proxy); a.update_interval = b.update_interval.or(a.update_interval); Some(a) } t @ _ => t.0.or(t.1), } } } impl Default for PrfItem { fn default() -> Self { PrfItem { uid: None, itype: None, name: None, desc: None, file: None, url: None, selected: None, extra: None, updated: None, option: None, file_data: None, } } } impl PrfItem { /// From partial item /// must contain `itype` pub async fn from(item: PrfItem, file_data: Option) -> Result { if item.itype.is_none() { bail!("type should not be null"); } match item.itype.unwrap().as_str() { "remote" => { if item.url.is_none() { bail!("url should not be null"); } let url = item.url.as_ref().unwrap().as_str(); let name = item.name; let desc = item.desc; PrfItem::from_url(url, name, desc, item.option).await } "local" => { let name = item.name.unwrap_or("Local File".into()); let desc = item.desc.unwrap_or("".into()); PrfItem::from_local(name, desc, file_data) } "merge" => { let name = item.name.unwrap_or("Merge".into()); let desc = item.desc.unwrap_or("".into()); PrfItem::from_merge(name, desc) } "script" => { let name = item.name.unwrap_or("Script".into()); let desc = item.desc.unwrap_or("".into()); PrfItem::from_script(name, desc) } typ @ _ => bail!("invalid profile item type \"{typ}\""), } } /// ## Local type /// create a new item from name/desc pub fn from_local(name: String, desc: String, file_data: Option) -> Result { let uid = help::get_uid("l"); let file = format!("{uid}.yaml"); Ok(PrfItem { uid: Some(uid), itype: Some("local".into()), name: Some(name), desc: Some(desc), file: Some(file), url: None, selected: None, extra: None, option: None, updated: Some(chrono::Local::now().timestamp() as usize), file_data: Some(file_data.unwrap_or(tmpl::ITEM_LOCAL.into())), }) } /// ## Remote type /// create a new item from url pub async fn from_url( url: &str, name: Option, desc: Option, option: Option, ) -> Result { let opt_ref = option.as_ref(); let with_proxy = opt_ref.map_or(false, |o| o.with_proxy.unwrap_or(false)); let self_proxy = opt_ref.map_or(false, |o| o.self_proxy.unwrap_or(false)); let user_agent = opt_ref.map_or(None, |o| o.user_agent.clone()); let mut builder = reqwest::ClientBuilder::new().no_proxy(); // 使用软件自己的代理 if self_proxy { let port = Config::clash().data().get_info()?.port; let port = port.ok_or(anyhow::anyhow!("failed to get clash info port"))?; let proxy_scheme = format!("http://127.0.0.1:{port}"); if let Ok(proxy) = reqwest::Proxy::http(&proxy_scheme) { builder = builder.proxy(proxy); } if let Ok(proxy) = reqwest::Proxy::https(&proxy_scheme) { builder = builder.proxy(proxy); } if let Ok(proxy) = reqwest::Proxy::all(&proxy_scheme) { builder = builder.proxy(proxy); } } // 使用系统代理 else if with_proxy { match Sysproxy::get_system_proxy() { Ok(p @ Sysproxy { enable: true, .. }) => { let proxy_scheme = format!("http://{}:{}", p.host, p.port); if let Ok(proxy) = reqwest::Proxy::http(&proxy_scheme) { builder = builder.proxy(proxy); } if let Ok(proxy) = reqwest::Proxy::https(&proxy_scheme) { builder = builder.proxy(proxy); } if let Ok(proxy) = reqwest::Proxy::all(&proxy_scheme) { builder = builder.proxy(proxy); } } _ => {} }; } let version = unsafe { dirs::APP_VERSION }; let version = format!("clash-verge/{version}"); builder = builder.user_agent(user_agent.unwrap_or(version)); let resp = builder.build()?.get(url).send().await?; let status_code = resp.status(); if !StatusCode::is_success(&status_code) { bail!("failed to fetch remote profile with status {status_code}") } let header = resp.headers(); // parse the Subscription UserInfo let extra = match header.get("Subscription-Userinfo") { Some(value) => { let sub_info = value.to_str().unwrap_or(""); Some(PrfExtra { upload: help::parse_str(sub_info, "upload=").unwrap_or(0), download: help::parse_str(sub_info, "download=").unwrap_or(0), total: help::parse_str(sub_info, "total=").unwrap_or(0), expire: help::parse_str(sub_info, "expire=").unwrap_or(0), }) } None => None, }; // parse the Content-Disposition let filename = match header.get("Content-Disposition") { Some(value) => { let filename = value.to_str().unwrap_or(""); help::parse_str::(filename, "filename=") } None => None, }; // parse the profile-update-interval let option = match header.get("profile-update-interval") { Some(value) => match value.to_str().unwrap_or("").parse::() { Ok(val) => Some(PrfOption { update_interval: Some(val * 60), // hour -> min ..PrfOption::default() }), Err(_) => None, }, None => None, }; let uid = help::get_uid("r"); let file = format!("{uid}.yaml"); let name = name.unwrap_or(filename.unwrap_or("Remote File".into())); let data = resp.text_with_charset("utf-8").await?; // check the data whether the valid yaml format let yaml = serde_yaml::from_str::(&data) // .context("the remote profile data is invalid yaml")?; if !yaml.contains_key("proxies") && !yaml.contains_key("proxy-providers") { bail!("profile does not contain `proxies` or `proxy-providers`"); } Ok(PrfItem { uid: Some(uid), itype: Some("remote".into()), name: Some(name), desc, file: Some(file), url: Some(url.into()), selected: None, extra, option, updated: Some(chrono::Local::now().timestamp() as usize), file_data: Some(data), }) } /// ## Merge type (enhance) /// create the enhanced item by using `merge` rule pub fn from_merge(name: String, desc: String) -> Result { let uid = help::get_uid("m"); let file = format!("{uid}.yaml"); Ok(PrfItem { uid: Some(uid), itype: Some("merge".into()), name: Some(name), desc: Some(desc), file: Some(file), url: None, selected: None, extra: None, option: None, updated: Some(chrono::Local::now().timestamp() as usize), file_data: Some(tmpl::ITEM_MERGE.into()), }) } /// ## Script type (enhance) /// create the enhanced item by using javascript quick.js pub fn from_script(name: String, desc: String) -> Result { let uid = help::get_uid("s"); let file = format!("{uid}.js"); // js ext Ok(PrfItem { uid: Some(uid), itype: Some("script".into()), name: Some(name), desc: Some(desc), file: Some(file), url: None, selected: None, extra: None, option: None, updated: Some(chrono::Local::now().timestamp() as usize), file_data: Some(tmpl::ITEM_SCRIPT.into()), }) } /// get the file data pub fn read_file(&self) -> Result { if self.file.is_none() { bail!("could not find the file"); } let file = self.file.clone().unwrap(); let path = dirs::app_profiles_dir()?.join(file); fs::read_to_string(path).context("failed to read the file") } /// save the file data pub fn save_file(&self, data: String) -> Result<()> { if self.file.is_none() { bail!("could not find the file"); } let file = self.file.clone().unwrap(); let path = dirs::app_profiles_dir()?.join(file); fs::write(path, data.as_bytes()).context("failed to save the file") } /// get the data for enhanced mode pub fn to_enhance(&self) -> Option { let itype = self.itype.as_ref()?.as_str(); let file = self.file.clone()?; let uid = self.uid.clone().unwrap_or("".into()); let path = dirs::app_profiles_dir().ok()?.join(file); if !path.exists() { return None; } match itype { "script" => Some(ChainItem { uid, data: ChainType::Script(fs::read_to_string(path).ok()?), }), "merge" => Some(ChainItem { uid, data: ChainType::Merge(help::read_merge_mapping(&path).ok()?), }), _ => None, } } } #[derive(Debug, Clone)] pub struct ChainItem { pub uid: String, pub data: ChainType, } #[derive(Debug, Clone)] pub enum ChainType { Merge(Mapping), Script(String), }