diff options
Diffstat (limited to 'contrib/epee/include')
-rw-r--r-- | contrib/epee/include/console_handler.h | 6 | ||||
-rw-r--r-- | contrib/epee/include/readline_buffer.h | 13 |
2 files changed, 19 insertions, 0 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 |