grand refactor

1 files changed, 14 insertions, 0 deletions

diff --git a/src/prompt_char.rs b/src/prompt_char.rs
new file mode 100644
index 0000000..1df4c5d
--- /dev/null
+++ b/src/prompt_char.rs
@@ -0,0 +1,14 @@
+use std::env;
+use colored::*;
+
+pub fn prompt_char() -> colored::ColoredString {
+    let user_char = env::var("PROMPT_CHAR").unwrap_or("$ ".into());
+    let root_char = env::var("PROMPT_CHAR_ROOT").unwrap_or("# ".into());
+
+    let euid = unsafe { libc::geteuid() };
+    match euid {
+        0 => return root_char.red(),
+        _ => return user_char.green()
+    }
+}
+