91 lines
2.4 KiB
Rust
91 lines
2.4 KiB
Rust
|
extern crate reqwest;
|
||
|
|
|
||
|
|
use crate::config::verge::{ProfileData, ProfileUserInfo};
|
||
|
|
use crate::init::{app_home_dir, read_verge_config, save_verge_config};
|
||
|
|
use std::default::Default;
|
||
|
|
use std::fs::File;
|
||
|
|
use std::io::Write;
|
||
|
|
use std::time::{SystemTime, UNIX_EPOCH};
|
||
|
|
|
||
|
|
/// Todo: log
|
||
|
|
/// Import the Profile from url
|
||
|
|
/// save to the `verge.yaml` file
|
||
|
|
pub async fn import_profile(profile_url: &str) -> Result<(), reqwest::Error> {
|
||
|
|
let resp = reqwest::get(profile_url).await?;
|
||
|
|
let header = resp.headers().clone();
|
||
|
|
let value = header
|
||
|
|
.get("Subscription-Userinfo")
|
||
|
|
.unwrap()
|
||
|
|
.to_str()
|
||
|
|
.unwrap();
|
||
|
|
let value: Vec<&str> = value.clone().split(';').collect();
|
||
|
|
|
||
|
|
let mut user_info = ProfileUserInfo::default();
|
||
|
|
|
||
|
|
for each in value.iter() {
|
||
|
|
let each = each.clone().trim();
|
||
|
|
if let Some(val) = each.strip_prefix("upload=") {
|
||
|
|
user_info.upload = val.parse().unwrap_or(0u64);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
if let Some(val) = each.strip_prefix("download=") {
|
||
|
|
user_info.download = val.parse().unwrap_or(0u64);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
if let Some(val) = each.strip_prefix("total=") {
|
||
|
|
user_info.total = val.parse().unwrap_or(0u64);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
if let Some(val) = each.strip_prefix("expire=") {
|
||
|
|
user_info.expire = val.parse().unwrap_or(0u64);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// save file
|
||
|
|
let file_data = resp.text_with_charset("utf-8").await?;
|
||
|
|
let cur_time = SystemTime::now()
|
||
|
|
.duration_since(UNIX_EPOCH)
|
||
|
|
.unwrap()
|
||
|
|
.as_secs();
|
||
|
|
let file_name = format!("{}.yaml", cur_time);
|
||
|
|
let file_path = app_home_dir().join("profiles").join(&file_name);
|
||
|
|
|
||
|
|
File::create(file_path)
|
||
|
|
.unwrap()
|
||
|
|
.write(file_data.as_bytes())
|
||
|
|
.unwrap();
|
||
|
|
|
||
|
|
let mut verge = read_verge_config();
|
||
|
|
|
||
|
|
let mut profiles = if verge.profiles.is_some() {
|
||
|
|
verge.profiles.unwrap()
|
||
|
|
} else {
|
||
|
|
vec![]
|
||
|
|
};
|
||
|
|
|
||
|
|
let profile = ProfileData {
|
||
|
|
name: Some(file_name.clone()),
|
||
|
|
file: Some(file_name.clone()),
|
||
|
|
mode: Some(String::from("rule")),
|
||
|
|
url: Some(String::from(profile_url)),
|
||
|
|
selected: Some(vec![]),
|
||
|
|
user_info: Some(user_info),
|
||
|
|
};
|
||
|
|
|
||
|
|
let target_index = profiles
|
||
|
|
.iter()
|
||
|
|
.position(|x| x.name.is_some() && x.name.as_ref().unwrap().as_str() == file_name.as_str());
|
||
|
|
|
||
|
|
if target_index.is_none() {
|
||
|
|
profiles.push(profile)
|
||
|
|
} else {
|
||
|
|
profiles[target_index.unwrap()] = profile;
|
||
|
|
}
|
||
|
|
|
||
|
|
verge.profiles = Some(profiles);
|
||
|
|
save_verge_config(&verge);
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
}
|