shadowfacts/grass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use dynamic dispatch to simplify file io a bit

Browse Source
This commit is contained in:
Connor Skees 2020-07-16 00:07:59 -04:00
parent d5d527ab70
commit fb724b8bee
1 changed files with 17 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/main.rs
View File

@ -192,20 +192,17 @@ fn main() -> std::io::Result<()> {
.unicode_error_messages(!matches.is_present("NO_UNICODE"))
.allows_charset(!matches.is_present("NO_CHARSET"));
if let Some(name) = matches.value_of("INPUT") {
if let Some(path) = matches.value_of("OUTPUT") {
let mut buf = BufWriter::new(File::open(path).unwrap_or(File::create(path)?));
buf.write_all(
from_path(name, &options)
.unwrap_or_else(|e| {
eprintln!("{}", e);
std::process::exit(1)
})
.as_bytes(),
)?;
let (mut stdout_write, mut file_write);
let buf_out: &mut dyn Write = if let Some(path) = matches.value_of("OUTPUT") {
file_write = BufWriter::new(File::open(path).unwrap_or(File::create(path)?));
&mut file_write
} else {
let mut stdout = BufWriter::new(stdout());
stdout.write_all(
stdout_write = BufWriter::new(stdout());
&mut stdout_write
};
if let Some(name) = matches.value_of("INPUT") {
buf_out.write_all(
from_path(name, &options)
.unwrap_or_else(|e| {
eprintln!("{}", e);
@ -214,6 +211,5 @@ fn main() -> std::io::Result<()> {
.as_bytes(),
)?;
}
}
Ok(())
}