1 files changed, 13 insertions(+), 1 deletions(-)

M src/icon.rs

@@ 1,5 1,17 @@
 use std::io::{Error, Write};
 
+use crate::color;
+
 pub fn icon<W: Write>(mut w: W) -> Result<(), Error> {
-    write!(w, "❯ ")
+    match std::env::args()
+        .nth(1)
+        .and_then(|s| s.parse::<usize>().ok())
+    {
+        Some(exit_code) if exit_code != 0 => {
+            color::red(&mut w)?;
+            write!(w, "❯ ")?;
+            color::reset(&mut w)
+        }
+        _ => write!(w, "❯ "),
+    }
 }