diff --git a/src/main.rs b/src/main.rs index 76ad636..61a3ba2 100644 --- a/src/main.rs +++ b/src/main.rs @@ -40,8 +40,13 @@ fn main() -> io::Result<()> { process::exit(1); } - let in_file = fs::File::open(&args[1])?; - let mut in_reader = io::BufReader::new(in_file); + let mut in_reader:Box; + if args[1] == "-" { + in_reader = Box::new(io::stdin()); + } else { + let in_file = fs::File::open(&args[1])?; + in_reader = Box::new(io::BufReader::new(in_file)); + } let mut out_writer:Box; if args.len() == 3 {