fix vt100 cursor pos race

1 files changed, 32 insertions, 11 deletions

diff --git a/src/util/prompt.cc b/src/util/prompt.cc
index 92ce2ce..0046b0f 100644
--- a/src/util/prompt.cc
+++ b/src/util/prompt.cc
@@ -7,29 +7,50 @@ namespace ov4
 
     bool emit_prompt = true; /* emit prompt (default) */
 
-    void print_prompt()
+    int print_prompt()
     {
         if (emit_prompt) 
         {
             char hostname[MAXLINE];
             gethostname(hostname, MAXLINE);
 
+            string _getlogin = safe_str(getlogin());
+            string _gethostname = safe_str(hostname);
+            string _get_current_dir = get_current_dir();
+            string _get_prompt = get_prompt_symbol();
+            string _at_symbol = "@";
+            string _space = " ";
+            string _semicolon = ":";
+
 	        cout 
                 << make_color((string)name, ov4::color::bright_green) 
-                << " "
-                << make_color(safe_str(getlogin()) , ov4::color::bright_blue)
-                << make_color("@", ov4::color::bright_blue)
-                << make_color(hostname, ov4::color::bright_blue)
-                << ":"
-                << make_color(get_current_dir(), ov4::color::bright_purple)
-                << " " 
-                << make_color(get_prompt(), ov4::color::bright_yellow)
-                << " " 
+                << _space
+                << make_color(_getlogin , ov4::color::bright_blue)
+                << make_color(_at_symbol, ov4::color::bright_blue)
+                << make_color(_gethostname, ov4::color::bright_blue)
+                << _semicolon
+                << make_color(_get_current_dir, ov4::color::bright_purple)
+                << _space
+                << make_color(_get_prompt, ov4::color::bright_yellow)
+                << _space
                 << flush;
+
+            return 
+                name.size() 
+                + _space.size()
+                + _getlogin.size() 
+                + _at_symbol.size() 
+                + _gethostname.size() 
+                + _semicolon.size()
+                + _get_current_dir.size()
+                + _space.size()
+                + _get_prompt.size() 
+                + _space.size();
 	    }
+        return 0;
     }
 
-    string get_prompt()
+    string get_prompt_symbol()
     {
         return (string)(safe_str(getlogin()) == "root" ? prompt_root : prompt_normal);
     }