hostname support

2 files changed, 23 insertions, 2 deletions

diff --git a/include/util/prompt.h b/include/util/prompt.h
index 464156a..6cdcfb8 100644
--- a/include/util/prompt.h
+++ b/include/util/prompt.h
@@ -6,8 +6,11 @@
 
 namespace ov4
 {
-    constexpr std::string_view prompt = "$";
+    constexpr std::string_view prompt_normal = "$";
+    constexpr std::string_view prompt_root = "#";
+    constexpr size_t hostname_size = 128;
     extern bool emit_prompt;
 
     void print_prompt();
+    std::string get_prompt();
 }
diff --git a/src/util/prompt.cc b/src/util/prompt.cc
index 8f4e5e1..977cb0b 100644
--- a/src/util/prompt.cc
+++ b/src/util/prompt.cc
@@ -10,7 +10,25 @@ namespace ov4
     {
         if (emit_prompt) 
         {
-	        cout << name << " : " << get_current_dir() << " " << prompt << " " << flush;
+            char buffer[hostname_size];
+            gethostname(buffer, hostname_size);
+	        cout 
+                << name 
+                << " "
+                << getlogin()
+                << "@"
+                << buffer
+                << ":"
+                << get_current_dir() 
+                << " " 
+                << get_prompt() 
+                << " " 
+                << flush;
 	    }
     }
+
+    string get_prompt()
+    {
+        return (string)((string)(getlogin()) == "root" ? prompt_root : prompt_normal);
+    }
 }