diff --git a/contrib/epee/include/console_handler.h b/contrib/epee/include/console_handler.h index 6832f2ea..e780ad4d 100644 --- a/contrib/epee/include/console_handler.h +++ b/contrib/epee/include/console_handler.h @@ -374,6 +374,9 @@ namespace epee } else { +#ifdef HAVE_READLINE + rdln::suspend_readline pause_readline; +#endif std::cout << "unknown command: " << command << std::endl; std::cout << usage; } @@ -477,6 +480,9 @@ namespace epee lookup::mapped_type & vt = m_command_handlers[cmd]; vt.first = hndlr; vt.second = usage; +#ifdef HAVE_READLINE + rdln::readline_buffer::add_completion(cmd); +#endif } bool process_command_vec(const std::vector& cmd) diff --git a/contrib/epee/include/readline_buffer.h b/contrib/epee/include/readline_buffer.h index 916d14f0..8dd082a7 100644 --- a/contrib/epee/include/readline_buffer.h +++ b/contrib/epee/include/readline_buffer.h @@ -3,6 +3,8 @@ #include #include #include +#include +#include namespace rdln { @@ -19,12 +21,23 @@ namespace rdln } void get_line(std::string& line) const; void set_prompt(const std::string& prompt); + static void add_completion(const std::string& command) + { + if(std::find(completion_commands.begin(), completion_commands.end(), command) != completion_commands.end()) + return; + completion_commands.push_back(command); + } + static const std::vector& get_completions() + { + return completion_commands; + } protected: virtual int sync(); private: std::streambuf* m_cout_buf; + static std::vector completion_commands; }; class suspend_readline diff --git a/contrib/epee/src/readline_buffer.cpp b/contrib/epee/src/readline_buffer.cpp index 53261132..549f49d4 100644 --- a/contrib/epee/src/readline_buffer.cpp +++ b/contrib/epee/src/readline_buffer.cpp @@ -6,6 +6,7 @@ #include #include #include +#include static int process_input(); static void install_line_handler(); @@ -16,6 +17,8 @@ static std::string last_prompt; std::mutex line_mutex, sync_mutex, process_mutex; std::condition_variable have_line; +std::vector rdln::readline_buffer::completion_commands = {"exit"}; + namespace { rdln::readline_buffer* current = NULL; @@ -79,6 +82,7 @@ void rdln::readline_buffer::set_prompt(const std::string& prompt) last_prompt = prompt; if(m_cout_buf == NULL) return; + std::lock_guard lock(sync_mutex); rl_set_prompt(last_prompt.c_str()); rl_redisplay(); } @@ -113,8 +117,7 @@ int rdln::readline_buffer::sync() do { - char x = this->sgetc(); - m_cout_buf->sputc(x); + m_cout_buf->sputc( this->sgetc() ); } while ( this->snextc() != EOF ); @@ -149,43 +152,44 @@ static int process_input() static void handle_line(char* line) { - if (line != NULL) + if(last_line == "exit" || last_line == "q") { - std::lock_guard lock(sync_mutex); - rl_set_prompt(last_prompt.c_str()); - rl_already_prompted = 1; return; } - rl_set_prompt(""); - rl_replace_line("", 0); - rl_redisplay(); + std::lock_guard lock(sync_mutex); rl_set_prompt(last_prompt.c_str()); + rl_already_prompted = 1; } static int handle_enter(int x, int y) { std::lock_guard lock(sync_mutex); char* line = NULL; - + line = rl_copy_text(0, rl_end); + rl_crlf(); + rl_on_new_line(); rl_set_prompt(""); rl_replace_line("", 1); rl_redisplay(); - if (strcmp(line, "") != 0) + std::string test_line = line; + boost::trim_right(test_line); + if (test_line.length() > 0) { - last_line = line; - add_history(line); + last_line = test_line; + add_history(test_line.c_str()); have_line.notify_one(); } free(line); - - if(last_line != "exit") + + if(last_line != "exit" && last_line != "q") { rl_set_prompt(last_prompt.c_str()); + rl_on_new_line_with_prompt(); rl_redisplay(); } - + rl_done = 1; return 0; } @@ -197,15 +201,50 @@ static int startup_hook() return 0; } +static char* completion_matches(const char* text, int state) +{ + static size_t list_index; + static size_t len; + + if(state == 0) + { + list_index = 0; + len = strlen(text); + } + + const std::vector& completions = rdln::readline_buffer::get_completions(); + for(; list_index