feat: Replace emojis when NO_EMOJI env variable present

1 files changed, 34 insertions, 10 deletions

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()
+                );
+            }
+        }
     }};
 }