From 8d62a9963708dbecd9312e8bcc4b47049c72d155 Mon Sep 17 00:00:00 2001
From: Matt Lebl <lebl.matt@gmail.com>
Date: Fri, 19 Mar 2021 02:16:07 -0700
Subject: feat: Replace emojis when NO_EMOJI env variable present

---
 src/ui.rs | 44 ++++++++++++++++++++++++++++++++++----------
 1 file changed, 34 insertions(+), 10 deletions(-)

(limited to 'src/ui.rs')

diff --git a/src/ui.rs b/src/ui.rs
index 38cbaa4..7ab8754 100644
--- a/src/ui.rs
+++ b/src/ui.rs
@@ -1,23 +1,47 @@
 macro_rules! warn {
     ($fmt:literal, $ex:expr) => {{
+        use std::env;
         use console::{style, Emoji};
         let formatstr = format!($fmt, $ex);
-        println!(
-            "{} {}",
-            style(Emoji("⚠️ ", "!")).red(),
-            style(formatstr).red()
-        );
+        match env::var("NO_EMOJI").is_ok() {
+            true => {
+                println!(
+                    "{} {}",
+                    style("!").red(),
+                    style(formatstr).red()
+                );
+            },
+            false => {
+                println!(
+                    "{} {}",
+                    style(Emoji("⚠️ ", "!")).red(),
+                    style(formatstr).red()
+                );
+            }
+        }
     }};
 }
 
 macro_rules! success {
     ($fmt:literal, $ex:expr) => {{
+        use std::env;
         use console::{style, Emoji};
         let formatstr = format!($fmt, $ex);
-        println!(
-            "{} {}",
-            style(Emoji("✅", "✓")).green(),
-            style(formatstr).green()
-        );
+        match env::var("NO_EMOJI").is_ok() {
+            true => {
+                println!(
+                    "{} {}",
+                    style("✓").green(),
+                    style(formatstr).green()
+                );
+            },
+            false => {
+                println!(
+                    "{} {}",
+                    style(Emoji("✅", "✓")).green(),
+                    style(formatstr).green()
+                );
+            }
+        }
     }};
 }
-- 
cgit v1.2.3