lrlex: Allow stdin as input file

softdevteam · Aug 22, 2024 · 06ea4d6 · 06ea4d6
1 parent 67e1adc
commit 06ea4d6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lrlex/src/main.rs b/lrlex/src/main.rs
@@ -2,7 +2,7 @@ use getopts::Options;
 use std::{
     env,
     fs::File,
-    io::{stderr, Read, Write},
+    io::{stdin, stderr, Read, Write},
     path::Path,
     process,
     str::FromStr,
@@ -26,14 +26,18 @@ fn usage(prog: &str, msg: &str) {
 }
 
 fn read_file(path: &str) -> String {
+    let mut s = String::new();
+    if path == "-" {
+        stdin().read_to_string(&mut s).unwrap();
+        return s;
+    }
     let mut f = match File::open(path) {
         Ok(r) => r,
         Err(e) => {
             writeln!(stderr(), "Can't open file {}: {}", path, e).ok();
             process::exit(1);
         }
     };
-    let mut s = String::new();
     f.read_to_string(&mut s).unwrap();
     s
 }