hostname support

1 files changed, 19 insertions, 1 deletions

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