#48 support feature isatty on stdout

sitkevij · Nov 16, 2020 · 94cee26 · 94cee26
1 parent 2e34704
commit 94cee26
Show file tree

Hide file tree

Showing 3 changed files with 63 additions and 56 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -30,6 +30,7 @@ opt-level = 3
 [dependencies]
 clap = "2.31.1"
 ansi_term = "0.11"
+atty = "0.2"
 
 [dev-dependencies]
 assert_cmd = "1.0.1"
diff --git a/src/lib.rs b/src/lib.rs
@@ -15,6 +15,7 @@
 extern crate ansi_term;
 extern crate clap;
 
+use atty::Stream;
 use clap::ArgMatches;
 use std::env;
 use std::error::Error;
@@ -276,6 +277,12 @@ pub fn run(matches: ArgMatches) -> Result<(), Box<dyn Error>> {
             }
         }
 
+        // prevent term color codes being sent to stdout
+        // test: cat Cargo.toml | target/debug/hx | more
+        if !atty::is(Stream::Stdout) {
+            colorize = false;
+        }
+
         // array output mode is mutually exclusive
         if let Some(array) = matches.value_of(ARG_ARR) {
             output_array(array, buf, truncate_len, column_width)?;