add config opts for prompt chars

1 files changed, 7 insertions, 5 deletions

diff --git a/src/prompt_char.rs b/src/prompt_char.rs
index 1df4c5d..a95681e 100644
--- a/src/prompt_char.rs
+++ b/src/prompt_char.rs
@@ -1,14 +1,16 @@
 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());
+pub fn get_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 user_char_color = env::var("PROMPT_CHAR_COLOR").unwrap_or("green ".into());
+    let root_char_color = env::var("PROMPT_CHAR_ROOT_COLOR").unwrap_or("red".into());
 
     let euid = unsafe { libc::geteuid() };
     match euid {
-        0 => return root_char.red(),
-        _ => return user_char.green()
+        0 => return root_char.color(root_char_color),
+        _ => return user_char.color(user_char_color)
     }
 }