committed by
Dan Gohman
parent
8fb681b86d
commit
90756a8a01
8
cranelift/src/clif-util.rs
Normal file → Executable file
8
cranelift/src/clif-util.rs
Normal file → Executable file
@@ -47,17 +47,17 @@ pub type CommandResult = Result<(), String>;
|
||||
|
||||
fn add_input_file_arg<'a>() -> clap::Arg<'a, 'a> {
|
||||
Arg::with_name("file")
|
||||
.required(true)
|
||||
.default_value("-")
|
||||
.multiple(true)
|
||||
.value_name("file")
|
||||
.help("Specify file(s) to be used for test")
|
||||
.help("Specify file(s) to be used for test. Defaults to reading from stdin.")
|
||||
}
|
||||
|
||||
fn add_single_input_file_arg<'a>() -> clap::Arg<'a, 'a> {
|
||||
Arg::with_name("single-file")
|
||||
.required(true)
|
||||
.default_value("-")
|
||||
.value_name("single-file")
|
||||
.help("Specify a file to be used")
|
||||
.help("Specify a file to be used. Defaults to reading from stdin.")
|
||||
}
|
||||
|
||||
fn add_pass_arg<'a>() -> clap::Arg<'a, 'a> {
|
||||
|
||||
@@ -12,17 +12,29 @@ use target_lexicon::Triple;
|
||||
|
||||
/// Read an entire file into a string.
|
||||
pub fn read_to_string<P: AsRef<Path>>(path: P) -> io::Result<String> {
|
||||
let mut file = File::open(path)?;
|
||||
let mut buffer = String::new();
|
||||
file.read_to_string(&mut buffer)?;
|
||||
if path.as_ref() == Path::new("-") {
|
||||
let stdin = io::stdin();
|
||||
let mut stdin = stdin.lock();
|
||||
stdin.read_to_string(&mut buffer)?;
|
||||
} else {
|
||||
let mut file = File::open(path)?;
|
||||
file.read_to_string(&mut buffer)?;
|
||||
}
|
||||
Ok(buffer)
|
||||
}
|
||||
|
||||
/// Read an entire file into a vector of bytes.
|
||||
pub fn read_to_end<P: AsRef<Path>>(path: P) -> io::Result<Vec<u8>> {
|
||||
let mut file = File::open(path)?;
|
||||
let mut buffer = Vec::new();
|
||||
file.read_to_end(&mut buffer)?;
|
||||
if path.as_ref() == Path::new("-") {
|
||||
let stdin = io::stdin();
|
||||
let mut stdin = stdin.lock();
|
||||
stdin.read_to_end(&mut buffer)?;
|
||||
} else {
|
||||
let mut file = File::open(path)?;
|
||||
file.read_to_end(&mut buffer)?;
|
||||
}
|
||||
Ok(buffer)
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user