rl_helper: add support to manage prompt string

If we want to change prompt string, we need to manage it inside
rl_helper.

2 files changed, 10 insertions, 1 deletions

diff --git a/rl_helper.c b/rl_helper.c
index dc0af9d..2cc014f 100644
--- a/rl_helper.c
+++ b/rl_helper.c
@@ -44,6 +44,7 @@ char lnbuf[MAX_LINE_BUFFER]; /* buffer for our line editing */
 size_t pos = 0;
 char seq[MAX_SEQ]; /* sequence buffer (escape codes) */
 size_t seq_pos = 0;
+const char *prompt = "> ";
 
 typedef struct {
     char sequence[MAX_SEQ];
@@ -92,7 +93,7 @@ void rl_clear_line() {
 void rl_reprint_prompt() {
 
     rl_clear_line();
-    printf("> %s", lnbuf);
+    printf("%s%s", prompt, lnbuf);
     fflush(stdout);
 }
 
@@ -108,6 +109,12 @@ void rl_init(line_process_callback cb) {
     line_cb = cb;
 }
 
+void rl_set_prompt(const char *str) {
+
+    prompt = str;
+    rl_reprint_prompt();
+}
+
 void rl_quit() {
 
     rl_clear();
diff --git a/rl_helper.h b/rl_helper.h
index 0f23423..f1d6266 100644
--- a/rl_helper.h
+++ b/rl_helper.h
@@ -5,6 +5,8 @@ typedef void (*line_process_callback)(char *line);
 
 /* initializes buffers and set line process callback */
 void rl_init(line_process_callback cb);
+/* configure prompt string, (eg "> ") */
+void rl_set_prompt(const char *str);
 /* close resources */
 void rl_quit();
 /* add char to line buffer */