Merge pull request #60 from quadrupleslap-forks-things/master

Use atty to check for color support

Author: KodrAus

Cargo.toml

@@ -18,6 +18,7 @@ log = { version = "0.4.0", features = ["std"] }
 regex = { version = "0.2", optional = true }
 termcolor = "0.3"
 chrono = "0.4"
+atty = "0.2"
 
 [[test]]
 name = "regexp_filter"

src/fmt.rs

@@ -42,6 +42,7 @@ use std::cell::RefCell;
 use termcolor::{ColorSpec, ColorChoice, Buffer, BufferWriter, WriteColor};
 use chrono::{DateTime, Utc};
 use chrono::format::Item;
+use atty;
 
 pub use termcolor::Color;
 
@@ -232,7 +233,16 @@ impl Builder {
     /// Build a terminal writer.
     pub fn build(&mut self) -> Writer {
         let color_choice = match self.write_style {
-            WriteStyle::Auto => ColorChoice::Auto,
+            WriteStyle::Auto => {
+                if atty::is(match self.target {
+                    Target::Stderr => atty::Stream::Stderr,
+                    Target::Stdout => atty::Stream::Stdout,
+                }) {
+                    ColorChoice::Auto
+                } else {
+                    ColorChoice::Never
+                }
+            },
             WriteStyle::Always => ColorChoice::Always,
             WriteStyle::Never => ColorChoice::Never,
         };

src/lib.rs

@@ -155,6 +155,7 @@
 extern crate log;
 extern crate termcolor;
 extern crate chrono;
+extern crate atty;
 
 use std::env;
 use std::borrow::Cow;