1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
use std::{process::{exit, Command, Stdio}, env, rc::Rc, io::Read};
use matrix::{vm::Vm, value::{Value, ValueMap}, unpack_args, Result, gc::Gc};
use matrix_macros::native_func;
use crate::{VmArgs, error};
#[native_func(1)]
fn sys_exit(_: VmArgs, args: Vec<Value>) -> Result<Value> {
let [value] = unpack_args!(args);
let Value::Int(i) = value else {
return error!("exit requires a int exit code")
};
exit(i as i32);
}
#[native_func(0)]
fn argv(_: VmArgs, _: Vec<Value>) -> Result<Value> {
Ok(Value::List(
Gc::new(
env::args()
.map(|a| Value::String(Rc::from(a.as_str())))
.collect()
)))
}
#[native_func(1)]
fn env(_: VmArgs, args: Vec<Value>) -> Result<Value> {
let [value] = unpack_args!(args);
let Value::String(value) = value else {
return error!("env requires a string name")
};
match std::env::var(value.as_ref()) {
Ok(v) => Ok(Value::String(v.into())),
Err(e) => error!("couldn't read env var: {e}")
}
}
#[native_func(2)]
fn exec(_: VmArgs, args: Vec<Value>) -> Result<Value> {
let [cmd, args] = unpack_args!(args);
let (cmd, args) = match (cmd, args) {
(Value::String(s), Value::List(l)) => (s, l.into_inner()),
_ => return error!("exec requires a string cmd and string argument list")
};
let mut sargs = Vec::new();
for arg in args {
let Value::String(arg) = arg else {
return error!("exec requires a string cmd and string argument list")
};
sargs.push(arg.to_string());
};
let cmd = Command::new(cmd.to_string())
.args(sargs)
.stdin(Stdio::piped())
.stdout(Stdio::piped())
.stderr(Stdio::piped())
.spawn();
let mut child = match cmd {
Ok(c) => c,
Err(e) => return error!("error executing command: {e}")
};
let status = match child.wait() {
Ok(s) => s,
Err(e) => return error!("error executing command: {e}")
};
let stdout = match child.stdout {
Some(ref mut out) => {
let mut buf = String::new();
let _ = out.read_to_string(&mut buf);
buf
},
None => String::new()
};
let stderr = match child.stderr {
Some(ref mut err) => {
let mut buf = String::new();
let _ = err.read_to_string(&mut buf);
buf
},
None => String::new()
};
let mut res = ValueMap::new();
res.insert(Value::from("success"), Value::Bool(status.success()))?;
res.insert(Value::from("code"), Value::Int(status.code().unwrap_or(0) as i64))?;
res.insert(Value::from("out"), Value::String(stdout.into()))?;
res.insert(Value::from("err"), Value::String(stderr.into()))?;
Ok(Value::Table(res.into()))
}
pub fn load(vm: &mut Vm) {
vm.load_global_fn(sys_exit(), "exit");
vm.load_global_fn(argv(), "argv");
vm.load_global_fn(exec(), "exec");
vm.load_global_fn(env(), "env");
}
|
