Added support for receiving from a pipe

PatrickTulskie · Jan 27, 2023 · 14ba428 · 14ba428
1 parent 75f9cfc
commit 14ba428
Show file tree

Hide file tree

Showing 5 changed files with 80 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/dfang/Cargo.toml b/dfang/Cargo.toml
@@ -14,3 +14,4 @@ edition = "2021"
 [dependencies]
 regex = "1"
 lazy_static = "1"
+atty = "0.2.*"
diff --git a/dfang/src/main.rs b/dfang/src/main.rs
@@ -1,6 +1,9 @@
 use lazy_static::lazy_static;
 use regex::Regex;
 use std::env;
+use std::io;
+use std::io::Read;
+extern crate atty;
 
 lazy_static! {
     // Replacers
@@ -20,7 +23,16 @@ fn main() {
     let args: Vec<String> = env::args().collect();
 
     if args.len() < 2 {
-        help();
+        let mut input = String::new();
+        if !atty::is(atty::Stream::Stdin) {
+            // read input from pipe
+            io::stdin().read_to_string(&mut input).unwrap();
+            for line in input.lines() {
+                println!("{}", defang(line));
+            }
+        } else {
+            help();
+        }
     } else {
         for i in 1..args.len() {
             println!("{}", defang(&args[i]));

diff --git a/rfang/Cargo.toml b/rfang/Cargo.toml
@@ -14,3 +14,4 @@ edition = "2021"
 [dependencies]
 regex = "1"
 lazy_static = "1"
+atty = "0.2.*"
diff --git a/rfang/src/main.rs b/rfang/src/main.rs
@@ -1,6 +1,9 @@
 use lazy_static::lazy_static;
 use regex::Regex;
 use std::env;
+use std::io;
+use std::io::Read;
+extern crate atty;
 
 lazy_static! {
     // Replacers
@@ -15,7 +18,16 @@ fn main() {
     let args: Vec<String> = env::args().collect();
 
     if args.len() < 2 {
-        help();
+        let mut input = String::new();
+        if !atty::is(atty::Stream::Stdin) {
+            // read input from pipe
+            io::stdin().read_to_string(&mut input).unwrap();
+            for line in input.lines() {
+                println!("{}", refang(line));
+            }
+        } else {
+            help();
+        }
     } else {
         for i in 1..args.len() {
             println!("{}", refang(&args[i]));