mroot/yazi
1
0
Fork 0
mirror of https://github.com/sxyazi/yazi.git synced 2026-07-25 08:41:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Update command.rs

Browse source
This commit is contained in:
mnv 2024-10-26 13:59:08 +05:30 • committed by sxyazi
parent 9babeef6d9
commit bb887f9230
No known key found for this signature in database
1 changed files with 12 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
yazi-plugin/src/process/command.rs
View file

@ -1,4 +1,5 @@
use std::process::Stdio;
use std::ffi::OsString;
use mlua::{AnyUserData, ExternalError, IntoLuaMulti, Lua, Table, UserData, Value};
use tokio::process::{ChildStderr, ChildStdin, ChildStdout};
use super::{Child, output::Output};
@ -63,27 +64,32 @@ impl UserData for Command {
}
methods.add_function_mut("arg", |_, (ud, arg): (AnyUserData, mlua::String)| {
ud.borrow_mut::<Self>()?.inner.arg(arg.to_string_lossy());
let arg = OsString::from(arg.to_str()?);
ud.borrow_mut::<Self>()?.inner.arg(arg);
Ok(ud)
});
methods.add_function_mut("args", |_, (ud, args): (AnyUserData, Vec<mlua::String>)| {
let args: Vec<_> = args.iter().map(|arg| arg.to_string_lossy()).collect();
let args: Vec<OsString> = args
.iter()
.map(|arg| OsString::from(arg.to_str().unwrap()))
.collect();
ud.borrow_mut::<Self>()?.inner.args(&args);
Ok(ud)
});
methods.add_function_mut("cwd", |_, (ud, dir): (AnyUserData, mlua::String)| {
ud.borrow_mut::<Self>()?.inner.current_dir(dir.to_str()?);
let dir = OsString::from(dir.to_str()?);
ud.borrow_mut::<Self>()?.inner.current_dir(dir);
Ok(ud)
});
methods.add_function_mut(
"env",
|_, (ud, key, value): (AnyUserData, mlua::String, mlua::String)| {
ud.borrow_mut::<Self>()?
.inner
.env(key.to_string_lossy(), value.to_string_lossy());
let key = OsString::from(key.to_str()?);
let value = OsString::from(value.to_str()?);
ud.borrow_mut::<Self>()?.inner.env(key, value);
Ok(ud)
},
);