mirror of
https://github.com/ivabus/smurf
synced 2024-11-21 16:05:11 +03:00
0.3.0
- add read_file_bytes - rename read_file_to_str() -> read_file_str() - change return types of read_file_* to Result<T, std::io::Error> - unpub shell::run() - don't overwrite $PATH in shell::run() - remove unnecessary tests Signed-off-by: Ivan Bushchik <ivabus@ivabus.dev>
This commit is contained in:
parent
f96e9b8be6
commit
8449b97658
4 changed files with 36 additions and 56 deletions
|
@ -1,6 +1,6 @@
|
|||
[package]
|
||||
name = "smurf"
|
||||
version = "0.2.0"
|
||||
version = "0.3.0"
|
||||
edition = "2021"
|
||||
authors = ["Ivan Bushchik <ivabus@ivabus.dev>"]
|
||||
description = "SMall Useful Rust Functions"
|
||||
|
|
|
@ -7,18 +7,20 @@
|
|||
### IO
|
||||
|
||||
- macros input!(T) and input_vec!(T), that could be used to simplify stdin process
|
||||
- read_file_to_str() basically reads file to string (yes.)
|
||||
- read_file_str() basically reads file to string (yes.)
|
||||
- read_file_bytes() see read_file_str()
|
||||
|
||||
### Shell
|
||||
|
||||
- macros shell!()
|
||||
|
||||
## Usage
|
||||
|
||||
Add this to your `Cargo.toml`:
|
||||
|
||||
```toml
|
||||
[dependencies]
|
||||
smurf = "0.2"
|
||||
smurf = "0.3"
|
||||
```
|
||||
|
||||
and this to your crate root:
|
||||
|
@ -49,4 +51,4 @@ fn main() {
|
|||
|
||||
## License
|
||||
|
||||
[MIT](LICENSE-MIT)
|
||||
[MIT](./LICENSE)
|
||||
|
|
34
src/io.rs
34
src/io.rs
|
@ -1,40 +1,36 @@
|
|||
use std::fs::File;
|
||||
use std::io::{prelude::*, BufReader};
|
||||
|
||||
pub fn read_file_to_str(path: &std::path::PathBuf) -> Option<String> {
|
||||
let file = File::open(&path);
|
||||
let file = match file {
|
||||
Ok(data) => data,
|
||||
Err(_) => {
|
||||
eprintln!("Cannot open file \"{}\"", path.display());
|
||||
return None
|
||||
}
|
||||
};
|
||||
pub fn read_file_str(path: &std::path::PathBuf) -> Result<String, std::io::Error> {
|
||||
let file = File::open(&path)?;
|
||||
let mut buf_reader = BufReader::new(file);
|
||||
let mut content = String::new();
|
||||
buf_reader.read_to_string(&mut content).unwrap();
|
||||
Some(content)
|
||||
buf_reader.read_to_string(&mut content)?;
|
||||
Ok(content)
|
||||
}
|
||||
|
||||
pub fn read_file_bytes(path: &std::path::PathBuf) -> Result<Vec<u8>, std::io::Error> {
|
||||
let file = File::open(&path)?;
|
||||
let mut buf_reader = BufReader::new(file);
|
||||
let mut content: Vec<u8> = Vec::new();
|
||||
buf_reader.read_to_end(&mut content)?;
|
||||
Ok(content)
|
||||
}
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! input {
|
||||
($t:ty) => {
|
||||
{
|
||||
($t:ty) => {{
|
||||
let mut input = String::new();
|
||||
std::io::stdin().read_line(&mut input).unwrap();
|
||||
input.trim().parse::<$t>().unwrap()
|
||||
}
|
||||
}
|
||||
}};
|
||||
}
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! input_vec {
|
||||
($t:ty) => {
|
||||
{
|
||||
($t:ty) => {{
|
||||
let mut input = String::new();
|
||||
std::io::stdin().read_line(&mut input).unwrap();
|
||||
input.trim().split_whitespace().map(|x| x.parse::<$t>().unwrap()).collect::<Vec<$t>>()
|
||||
}
|
||||
}
|
||||
}};
|
||||
}
|
||||
|
|
30
src/shell.rs
30
src/shell.rs
|
@ -13,7 +13,8 @@ pub struct ShellResult {
|
|||
pub stderr: String,
|
||||
}
|
||||
|
||||
pub fn run(command: String) -> ShellResult {
|
||||
#[allow(dead_code)]
|
||||
fn run(command: String) -> ShellResult {
|
||||
let mut iter = command.split_whitespace();
|
||||
let mut current = iter.next();
|
||||
let mut words: Vec<&str> = vec![];
|
||||
|
@ -23,7 +24,6 @@ pub fn run(command: String) -> ShellResult {
|
|||
}
|
||||
let command = Command::new(&words[0])
|
||||
.args(&words[1..])
|
||||
.env("PATH", "/bin:/usr/bin") // TODO: Make this configurable
|
||||
.output()
|
||||
.expect(&format!("Failed to execute '{}'", command));
|
||||
return ShellResult {
|
||||
|
@ -35,6 +35,9 @@ pub fn run(command: String) -> ShellResult {
|
|||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
// Testing basic usage and only on unix
|
||||
|
||||
#[cfg(unix)]
|
||||
#[test]
|
||||
fn test_shell_macro() {
|
||||
let result = shell!("echo hello");
|
||||
|
@ -43,17 +46,7 @@ mod tests {
|
|||
assert_eq!(result.stderr, "");
|
||||
}
|
||||
|
||||
/* You literally can't use pipe using std::process::Command */
|
||||
// TODO: figure out how to test this
|
||||
#[ignore]
|
||||
#[test]
|
||||
fn test_shell_macro_with_stderr() {
|
||||
let result = shell!("echo hello 1>&2");
|
||||
assert_eq!(result.code, 0);
|
||||
assert_eq!(result.stdout, "");
|
||||
assert_eq!(result.stderr, "hello\n");
|
||||
}
|
||||
|
||||
#[cfg(unix)]
|
||||
#[test]
|
||||
fn test_shell_macro_with_args() {
|
||||
let result = shell!("echo {} {}", "hello", "world");
|
||||
|
@ -61,15 +54,4 @@ mod tests {
|
|||
assert_eq!(result.stdout, "hello world\n");
|
||||
assert_eq!(result.stderr, "");
|
||||
}
|
||||
|
||||
// TODO: figure out how to test this
|
||||
#[ignore]
|
||||
#[test]
|
||||
fn test_shell_macro_with_code() {
|
||||
let result = shell!("sh -c exit 1");
|
||||
assert_eq!(result.code, 1);
|
||||
assert_eq!(result.stdout, "");
|
||||
assert_eq!(result.stderr, "");
|
||||
}
|
||||
|
||||
}
|
Loading…
Reference in a new issue