src/util/prompt.cc


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36


#include "util/prompt.h"

using namespace std;

namespace ov4
{

    bool emit_prompt = true; /* emit prompt (default) */

    void print_prompt()
    {
        if (emit_prompt) 
        {
            char hostname[MAXLINE];
            gethostname(hostname, MAXLINE);

	        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)
                << " " 
                << flush;
	    }
    }

    string get_prompt()
    {
        return (string)(safe_str(getlogin()) == "root" ? prompt_root : prompt_normal);
    }
}