Kiyomichi Kosaka
76 lines
2.5 KiB
Rust
76 lines
2.5 KiB
Rust
use shlex::split;
|
|
use std::env;
|
|
use std::ffi::OsString;
|
|
use std::io;
|
|
use std::io::{Read, Write};
|
|
use std::process::{Command, Stdio};
|
|
|
|
pub fn call_cmd_with_input(cmd: &str, args: &Vec<String>, input: &str) -> io::Result<String> {
|
|
let mut cmd = Command::new(cmd).args(args).stdin(Stdio::piped()).stdout(Stdio::piped()).spawn()?;
|
|
let mut stdin = cmd.stdin.take().unwrap();
|
|
let stdout = cmd.stdout.as_mut().unwrap();
|
|
let in_data = input.to_string();
|
|
let write_handle = std::thread::spawn(move || stdin.write_all(in_data.as_bytes()));
|
|
let mut output = Vec::new();
|
|
stdout.read_to_end(&mut output)?;
|
|
match write_handle.join() {
|
|
Ok(_) => (),
|
|
Err(_) => return Err(io::Error::new(io::ErrorKind::Other, "Failed to run command")),
|
|
}
|
|
match String::from_utf8(output) {
|
|
Ok(x) => Ok(x),
|
|
Err(err) => Err(io::Error::new(io::ErrorKind::InvalidData, err.utf8_error())),
|
|
}
|
|
}
|
|
|
|
pub fn get_cmd_args_from_command(command: &str) -> io::Result<(String, Vec<String>)> {
|
|
let args = match split(command) {
|
|
Some(c) => c,
|
|
None => {
|
|
return Err(io::Error::new(
|
|
io::ErrorKind::InvalidData,
|
|
format!("Failed to parse the command: {:?}", command),
|
|
))
|
|
}
|
|
};
|
|
Ok((args[0].clone(), args[1..].to_vec()))
|
|
}
|
|
|
|
pub fn get_copy_command_from_env() -> (String, Vec<String>) {
|
|
let cmd_os_str = env::var_os("HEL_PB").unwrap_or_else(|| match env::consts::OS {
|
|
_ if env::var("TMUX").is_ok() => OsString::from("tmux load-buffer -"),
|
|
"macos" => OsString::from("pbcopy"),
|
|
"linux" => OsString::from("xclip"),
|
|
_ => OsString::from("cat"),
|
|
});
|
|
get_cmd_args_from_command(&cmd_os_str.to_string_lossy()).unwrap_or_else(|_| ("cat".to_string(), vec![]))
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn cmd_exec_test() {
|
|
assert_eq!(call_cmd_with_input("true", &vec![], "").unwrap(), "".to_string());
|
|
assert_eq!(call_cmd_with_input("cat", &vec![], "ok").unwrap(), "ok".to_string());
|
|
assert_eq!(
|
|
call_cmd_with_input(
|
|
"cat",
|
|
&vec![],
|
|
r###"line 1
|
|
line 2
|
|
line 3
|
|
line 4"###
|
|
)
|
|
.unwrap(),
|
|
"line 1\nline 2\nline 3\nline 4".to_string()
|
|
);
|
|
assert_ne!(call_cmd_with_input("cat", &vec![], "notok").unwrap(), "ok".to_string());
|
|
assert_eq!(
|
|
call_cmd_with_input("echo", &vec!["-n".to_string(), "test is ok".to_string()], "").unwrap(),
|
|
"test is ok".to_string()
|
|
);
|
|
}
|
|
}
|