diff options
-rw-r--r-- | contrib/epee/include/console_handler.h | 6 | ||||
-rw-r--r-- | contrib/epee/include/readline_buffer.h | 13 | ||||
-rw-r--r-- | contrib/epee/src/readline_buffer.cpp | 71 | ||||
-rw-r--r-- | src/simplewallet/simplewallet.cpp | 4 |
4 files changed, 76 insertions, 18 deletions
diff --git a/contrib/epee/include/console_handler.h b/contrib/epee/include/console_handler.h index 6832f2ea1..e780ad4de 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<std::string>& cmd) diff --git a/contrib/epee/include/readline_buffer.h b/contrib/epee/include/readline_buffer.h index 916d14f01..8dd082a70 100644 --- a/contrib/epee/include/readline_buffer.h +++ b/contrib/epee/include/readline_buffer.h @@ -3,6 +3,8 @@ #include <streambuf> #include <sstream> #include <iostream> +#include <vector> +#include <algorithm> 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<std::string>& get_completions() + { + return completion_commands; + } protected: virtual int sync(); private: std::streambuf* m_cout_buf; + static std::vector<std::string> completion_commands; }; class suspend_readline diff --git a/contrib/epee/src/readline_buffer.cpp b/contrib/epee/src/readline_buffer.cpp index 532611326..549f49d45 100644 --- a/contrib/epee/src/readline_buffer.cpp +++ b/contrib/epee/src/readline_buffer.cpp @@ -6,6 +6,7 @@ #include <mutex> #include <condition_variable> #include <boost/thread.hpp> +#include <boost/algorithm/string.hpp> 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<std::string> 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<std::mutex> 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<std::mutex> 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<std::mutex> 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<std::mutex> 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<std::string>& completions = rdln::readline_buffer::get_completions(); + for(; list_index<completions.size(); ) + { + const std::string& cmd = completions[list_index++]; + if(cmd.compare(0, len, text) == 0) + { + return strdup(cmd.c_str()); + } + } + + return NULL; +} + +static char** attempted_completion(const char* text, int start, int end) +{ + rl_attempted_completion_over = 1; + return rl_completion_matches(text, completion_matches); +} + static void install_line_handler() { rl_startup_hook = startup_hook; + rl_attempted_completion_function = attempted_completion; rl_callback_handler_install("", handle_line); } static void remove_line_handler() { + rl_replace_line("", 0); + rl_set_prompt(""); + rl_redisplay(); rl_unbind_key(RETURN); + rl_unbind_key(NEWLINE); rl_callback_handler_remove(); } diff --git a/src/simplewallet/simplewallet.cpp b/src/simplewallet/simplewallet.cpp index c1f2d3495..adf2fde80 100644 --- a/src/simplewallet/simplewallet.cpp +++ b/src/simplewallet/simplewallet.cpp @@ -65,8 +65,7 @@ #ifdef HAVE_READLINE #include "readline_buffer.h" #define PAUSE_READLINE() \ - rdln::suspend_readline pause_readline; \ - std::cout << std::endl + rdln::suspend_readline pause_readline; #else #define PAUSE_READLINE() #endif @@ -195,6 +194,7 @@ namespace } else { + PAUSE_READLINE(); set_console_color(m_color, m_bright); std::cout << m_oss.str(); reset_console_color(); |