97 lines
2.4 KiB
Rust
97 lines
2.4 KiB
Rust
|
use super::{use_filter, use_valid_fields};
|
||
|
|
use anyhow::Result;
|
||
|
|
use serde_yaml::{self, Mapping};
|
||
|
|
|
||
|
|
pub fn use_script(
|
||
|
|
script: String,
|
||
|
|
config: Mapping,
|
||
|
|
valid: Vec<String>,
|
||
|
|
) -> Result<(Mapping, Vec<(String, String)>)> {
|
||
|
|
use rquickjs::{Context, Func, Runtime};
|
||
|
|
use std::sync::{Arc, Mutex};
|
||
|
|
|
||
|
|
let runtime = Runtime::new().unwrap();
|
||
|
|
let context = Context::full(&runtime).unwrap();
|
||
|
|
let outputs = Arc::new(Mutex::new(vec![]));
|
||
|
|
|
||
|
|
let copy_outputs = outputs.clone();
|
||
|
|
let result = context.with(|ctx| -> Result<Mapping> {
|
||
|
|
ctx.globals().set(
|
||
|
|
"__verge_log__",
|
||
|
|
Func::from(move |level: String, data: String| {
|
||
|
|
let mut out = copy_outputs.lock().unwrap();
|
||
|
|
out.push((level, data));
|
||
|
|
}),
|
||
|
|
)?;
|
||
|
|
|
||
|
|
ctx.eval(
|
||
|
|
r#"var console = Object.freeze({
|
||
|
|
log(data){__verge_log__("log",JSON.stringify(data))},
|
||
|
|
info(data){__verge_log__("info",JSON.stringify(data))},
|
||
|
|
error(data){__verge_log__("error",JSON.stringify(data))},
|
||
|
|
debug(data){__verge_log__("debug",JSON.stringify(data))},
|
||
|
|
});"#,
|
||
|
|
)?;
|
||
|
|
|
||
|
|
let config_str = serde_json::to_string(&config)?;
|
||
|
|
|
||
|
|
let code = format!("\n{script}\n;\nJSON.stringify(main({config_str})||'')");
|
||
|
|
let result: String = ctx.eval(code.as_str())?;
|
||
|
|
if result == "\"\"" {
|
||
|
|
anyhow::bail!("main function should return object");
|
||
|
|
}
|
||
|
|
Ok(serde_json::from_str::<Mapping>(result.as_str())?)
|
||
|
|
});
|
||
|
|
|
||
|
|
let mut out = outputs.lock().unwrap();
|
||
|
|
match result {
|
||
|
|
Ok(config) => {
|
||
|
|
let valid = use_valid_fields(valid);
|
||
|
|
let config = use_filter(config, valid);
|
||
|
|
Ok((config, out.to_vec()))
|
||
|
|
}
|
||
|
|
Err(err) => {
|
||
|
|
out.push(("error".into(), err.to_string()));
|
||
|
|
Ok((config, out.to_vec()))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn test_script() {
|
||
|
|
let script = r#"
|
||
|
|
function main(config) {
|
||
|
|
if (Array.isArray(config.rules)) {
|
||
|
|
config.rules = [...config.rules, "add"];
|
||
|
|
}
|
||
|
|
console.log(config);
|
||
|
|
config.proxies = ["111"];
|
||
|
|
return config;
|
||
|
|
}
|
||
|
|
"#;
|
||
|
|
|
||
|
|
let config = r#"
|
||
|
|
rules:
|
||
|
|
- 111
|
||
|
|
- 222
|
||
|
|
tun:
|
||
|
|
enable: false
|
||
|
|
dns:
|
||
|
|
enable: false
|
||
|
|
"#;
|
||
|
|
|
||
|
|
let config = serde_yaml::from_str(config).unwrap();
|
||
|
|
let (config, results) = use_script(
|
||
|
|
script.into(),
|
||
|
|
config,
|
||
|
|
vec!["tun"].iter().map(|s| s.to_string()).collect(),
|
||
|
|
)
|
||
|
|
.unwrap();
|
||
|
|
|
||
|
|
let config_str = serde_yaml::to_string(&config).unwrap();
|
||
|
|
|
||
|
|
println!("{config_str}");
|
||
|
|
|
||
|
|
dbg!(results);
|
||
|
|
}
|