flashrom_tester: Only print color when stdout isatty

Add the atty crate as a dependency. Print log and report in color only
when isatty is true.

BUG=b:246250254
BRANCH=None
TEST=ssh dut flashrom_tester # no color
TEST=ssh -t dut flashrom_tester # color

Change-Id: Ia3cc527fb98e53eda6773622340cf10764df2cba
Signed-off-by: Evan Benn <evanbenn@chromium.org>
Reviewed-on: https://review.coreboot.org/c/flashrom/+/69270
Tested-by: build bot (Jenkins) <no-reply@coreboot.org>
Reviewed-by: Angel Pons <th3fanbus@gmail.com>
Reviewed-by: Edward O'Callaghan <quasisec@chromium.org>

1 files changed, 29 insertions, 10 deletions

diff --git a/util/flashrom_tester/src/types.rs b/util/flashrom_tester/src/types.rs
index b22ded2b4..9cefb27e9 100644
--- a/util/flashrom_tester/src/types.rs
+++ b/util/flashrom_tester/src/types.rs
@@ -33,21 +33,40 @@
 // Software Foundation.
 //
 
-pub const BOLD: &str = "\x1b[1m";
+pub struct Color {
+    pub bold: &'static str,
+    pub reset: &'static str,
+    pub magenta: &'static str,
+    pub yellow: &'static str,
+    pub green: &'static str,
+    pub red: &'static str,
+}
+
+pub const COLOR: Color = Color {
+    bold: "\x1b[1m",
+    reset: "\x1b[0m",
+    magenta: "\x1b[35m",
+    yellow: "\x1b[33m",
+    green: "\x1b[92m",
+    red: "\x1b[31m",
+};
 
-pub const RESET: &str = "\x1b[0m";
-pub const MAGENTA: &str = "\x1b[35m";
-pub const YELLOW: &str = "\x1b[33m";
-pub const GREEN: &str = "\x1b[92m";
-pub const RED: &str = "\x1b[31m";
+pub const NOCOLOR: Color = Color {
+    bold: "",
+    reset: "",
+    magenta: "",
+    yellow: "",
+    green: "",
+    red: "",
+};
 
 macro_rules! style_dbg {
-    ($s: expr, $c: expr) => {
-        format!("{}{:?}{}", $c, $s, types::RESET)
+    ($s: expr, $c: expr, $col: expr) => {
+        format!("{}{:?}{}", $c, $s, $col.reset)
     };
 }
 macro_rules! style {
-    ($s: expr, $c: expr) => {
-        format!("{}{}{}", $c, $s, types::RESET)
+    ($s: expr, $c: expr, $col: expr) => {
+        format!("{}{}{}", $c, $s, $col.reset)
     };
 }