diff options
Diffstat (limited to 'src/readline_context.hh')
| -rw-r--r-- | src/readline_context.hh | 183 |
1 files changed, 183 insertions, 0 deletions
diff --git a/src/readline_context.hh b/src/readline_context.hh new file mode 100644 index 0000000..047cf56 --- /dev/null +++ b/src/readline_context.hh @@ -0,0 +1,183 @@ +/** + * Copyright (c) 2022, Timothy Stack + * + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * * Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * * Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * * Neither the name of Timothy Stack nor the names of its contributors + * may be used to endorse or promote products derived from this software + * without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ''AS IS'' AND ANY + * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY + * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES + * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; + * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND + * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS + * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + * + * @file readline_context.hh + */ + +#ifndef readline_context_hh +#define readline_context_hh + +#include <set> +#include <string> + +#include <readline/history.h> + +#include "base/lnav.console.hh" +#include "base/result.h" +#include "help_text.hh" + +class attr_line_t; +struct exec_context; + +typedef void (*readline_highlighter_t)(attr_line_t& line, int x); + +/** + * Container for information related to different readline contexts. Since + * lnav uses readline for different inputs, we need a way to keep things like + * history and tab-completions separate. + */ +class readline_context { +public: + typedef Result<std::string, lnav::console::user_message> (*command_func_t)( + exec_context& ec, std::string cmdline, std::vector<std::string>& args); + typedef std::string (*prompt_func_t)(exec_context& ec, + const std::string& cmdline); + typedef struct _command_t { + const char* c_name; + command_func_t c_func; + + struct help_text c_help; + prompt_func_t c_prompt{nullptr}; + + _command_t(const char* name, + command_func_t func, + help_text help = {}, + prompt_func_t prompt = nullptr) noexcept + : c_name(name), c_func(func), c_help(std::move(help)), + c_prompt(prompt) + { + } + + _command_t(command_func_t func) noexcept : c_name("anon"), c_func(func) + { + } + } command_t; + typedef std::map<std::string, command_t*> command_map_t; + + readline_context(std::string name, + command_map_t* commands = nullptr, + bool case_sensitive = true); + + const std::string& get_name() const { return this->rc_name; } + + void load(); + + void save(); + + void add_possibility(const std::string& type, const std::string& value) + { + this->rc_possibilities[type].insert(value); + } + + void rem_possibility(const std::string& type, const std::string& value) + { + this->rc_possibilities[type].erase(value); + } + + void clear_possibilities(const std::string& type) + { + this->rc_possibilities[type].clear(); + } + + bool is_case_sensitive() const { return this->rc_case_sensitive; } + + readline_context& set_append_character(int ch) + { + this->rc_append_character = ch; + + return *this; + } + + int get_append_character() const + { + return this->rc_append_character; + } + + readline_context& set_highlighter(readline_highlighter_t hl) + { + this->rc_highlighter = hl; + return *this; + } + + readline_context& set_quote_chars(const char* qc) + { + this->rc_quote_chars = qc; + + return *this; + } + + readline_context& with_readline_var(char** var, const char* val) + { + this->rc_vars.emplace_back(var, val); + + return *this; + } + + readline_highlighter_t get_highlighter() const + { + return this->rc_highlighter; + } + + static int command_complete(int, int); + + std::map<std::string, std::string> rc_prefixes; + +private: + static char** attempted_completion(const char* text, int start, int end); + static char* completion_generator(const char* text, int state); + + static readline_context* loaded_context; + static std::set<std::string>* arg_possibilities; + + struct readline_var { + readline_var(char** dst, const char* val) + { + this->rv_dst.ch = dst; + this->rv_val.ch = val; + } + + union { + char** ch; + } rv_dst; + union { + const char* ch; + } rv_val; + }; + + std::string rc_name; + HISTORY_STATE rc_history; + std::map<std::string, std::set<std::string>> rc_possibilities; + std::map<std::string, std::vector<std::string>> rc_prototypes; + bool rc_case_sensitive; + int rc_append_character; + const char* rc_quote_chars; + readline_highlighter_t rc_highlighter; + std::vector<readline_var> rc_vars; +}; + +#endif |