122 lines
4.2 KiB
Rust
122 lines
4.2 KiB
Rust
use crate::enhance::field::use_keys;
|
||
use serde::{Deserialize, Serialize};
|
||
use serde_yaml_ng::{Mapping, Value};
|
||
use std::collections::HashMap;
|
||
#[derive(Default, Debug, Clone, Deserialize, Serialize)]
|
||
pub struct IRuntime {
|
||
pub config: Option<Mapping>,
|
||
// 记录在订阅中（包括merge和script生成的）出现过的keys
|
||
// 这些keys不一定都生效
|
||
pub exists_keys: Vec<String>,
|
||
pub chain_logs: HashMap<String, Vec<(String, String)>>,
|
||
}
|
||
|
||
impl IRuntime {
|
||
pub fn new() -> Self {
|
||
Self::default()
|
||
}
|
||
|
||
// 这里只更改 allow-lan | ipv6 | log-level | tun
|
||
pub fn patch_config(&mut self, patch: Mapping) {
|
||
if let Some(config) = self.config.as_mut() {
|
||
["allow-lan", "ipv6", "log-level", "unified-delay"]
|
||
.into_iter()
|
||
.for_each(|key| {
|
||
if let Some(value) = patch.get(key).to_owned() {
|
||
config.insert(key.into(), value.clone());
|
||
}
|
||
});
|
||
|
||
let patch_tun = patch.get("tun");
|
||
if patch_tun.is_some() {
|
||
let tun = config.get("tun");
|
||
let mut tun = tun.map_or(Mapping::new(), |val| {
|
||
val.as_mapping().cloned().unwrap_or(Mapping::new())
|
||
});
|
||
let patch_tun = patch_tun.map_or(Mapping::new(), |val| {
|
||
val.as_mapping().cloned().unwrap_or(Mapping::new())
|
||
});
|
||
use_keys(&patch_tun).into_iter().for_each(|key| {
|
||
if let Some(value) = patch_tun.get(&key).to_owned() {
|
||
tun.insert(key.into(), value.clone());
|
||
}
|
||
});
|
||
|
||
config.insert("tun".into(), Value::from(tun));
|
||
}
|
||
}
|
||
}
|
||
|
||
//跟新链式代理配置文件
|
||
/// {
|
||
/// "proxies":[
|
||
/// {
|
||
/// name : 入口节点,
|
||
/// type: xxx
|
||
/// server: xxx
|
||
/// port: xxx
|
||
/// ports: xxx
|
||
/// password: xxx
|
||
/// skip-cert-verify: xxx,
|
||
/// },
|
||
/// {
|
||
/// name : hop_node_1_xxxx,
|
||
/// type: xxx
|
||
/// server: xxx
|
||
/// port: xxx
|
||
/// ports: xxx
|
||
/// password: xxx
|
||
/// skip-cert-verify: xxx,
|
||
/// dialer-proxy : "入口节点"
|
||
/// },
|
||
/// {
|
||
/// name : 出口节点,
|
||
/// type: xxx
|
||
/// server: xxx
|
||
/// port: xxx
|
||
/// ports: xxx
|
||
/// password: xxx
|
||
/// skip-cert-verify: xxx,
|
||
/// dialer-proxy : "hop_node_1_xxxx"
|
||
/// }
|
||
/// ],
|
||
/// "proxy-groups" : [
|
||
/// {
|
||
/// name : "proxy_chain",
|
||
/// type: "select",
|
||
/// proxies ["出口节点"]
|
||
/// }
|
||
/// ]
|
||
/// }
|
||
///
|
||
/// 传入none 为删除
|
||
pub fn update_proxy_chain_config(&mut self, proxy_chain_config: Option<Value>) {
|
||
if let Some(config) = self.config.as_mut() {
|
||
if let Some(Value::Sequence(proxies)) = config.get_mut("proxies") {
|
||
proxies.iter_mut().for_each(|proxy| {
|
||
if let Some(proxy) = proxy.as_mapping_mut()
|
||
&& proxy.get("dialer-proxy").is_some()
|
||
{
|
||
proxy.remove("dialer-proxy");
|
||
}
|
||
});
|
||
}
|
||
|
||
if let Some(Value::Sequence(dialer_proxies)) = proxy_chain_config
|
||
&& let Some(Value::Sequence(proxies)) = config.get_mut("proxies")
|
||
{
|
||
for (i, dialer_proxy) in dialer_proxies.iter().enumerate() {
|
||
if let Some(Value::Mapping(proxy)) = proxies
|
||
.iter_mut()
|
||
.find(|proxy| proxy.get("name") == Some(dialer_proxy))
|
||
&& i != 0
|
||
&& let Some(dialer_proxy) = dialer_proxies.get(i - 1)
|
||
{
|
||
proxy.insert("dialer-proxy".into(), dialer_proxy.to_owned());
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|