tyler.durden/icinga2
1
0
Fork 0
mirror of https://github.com/Icinga/icinga2.git synced 2025-07-26 23:24:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

Implement auto-completion support for 'icinga2 console'

Browse Source 
refs #8776
This commit is contained in:
Gunnar Beutner 2015-03-18 07:17:15 +01:00
parent d0ac1bdebf
commit cd9dfd5744
2 changed files with 80 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
lib/cli/consolecommand.cpp
View File

@ -22,6 +22,7 @@
#include "base/json.hpp" #include "base/json.hpp"
#include "base/console.hpp" #include "base/console.hpp"
#include "base/application.hpp" #include "base/application.hpp"
#include "base/objectlock.hpp"
#include "base/unixsocket.hpp" #include "base/unixsocket.hpp"
#include "base/utility.hpp" #include "base/utility.hpp"
#include "base/networkstream.hpp" #include "base/networkstream.hpp"
@ -33,6 +34,8 @@
using namespace icinga; using namespace icinga;
namespace po = boost::program_options; namespace po = boost::program_options;
static ScriptFrame l_ScriptFrame;
REGISTER_CLICOMMAND("console", ConsoleCommand); REGISTER_CLICOMMAND("console", ConsoleCommand);
String ConsoleCommand::GetDescription(void) const String ConsoleCommand::GetDescription(void) const
@ -58,6 +61,74 @@ void ConsoleCommand::InitParameters(boost::program_options::options_description&
; ;
} }
#ifdef HAVE_EDITLINE
static void AddSuggestion(std::vector<String>& matches, const String& word, const String& suggestion)
{
if (suggestion.Find(word) != 0)
return;
matches.push_back(suggestion);
}
static char *ConsoleCompleteHelper(const char *word, int state)
{
static std::vector<String> matches;
String aword = word;
if (state == 0) {
matches.clear();
AddSuggestion(matches, word, "object");
AddSuggestion(matches, word, "template");
AddSuggestion(matches, word, "include");
AddSuggestion(matches, word, "include_recursive");
AddSuggestion(matches, word, "library");
AddSuggestion(matches, word, "null");
AddSuggestion(matches, word, "true");
AddSuggestion(matches, word, "false");
AddSuggestion(matches, word, "const");
AddSuggestion(matches, word, "var");
AddSuggestion(matches, word, "this");
AddSuggestion(matches, word, "globals");
AddSuggestion(matches, word, "locals");
AddSuggestion(matches, word, "use");
AddSuggestion(matches, word, "apply");
AddSuggestion(matches, word, "to");
AddSuggestion(matches, word, "where");
AddSuggestion(matches, word, "import");
AddSuggestion(matches, word, "assign");
AddSuggestion(matches, word, "ignore");
AddSuggestion(matches, word, "function");
AddSuggestion(matches, word, "return");
AddSuggestion(matches, word, "break");
AddSuggestion(matches, word, "continue");
AddSuggestion(matches, word, "for");
AddSuggestion(matches, word, "if");
AddSuggestion(matches, word, "else");
AddSuggestion(matches, word, "while");
{
ObjectLock olock(l_ScriptFrame.Locals);
BOOST_FOREACH(const Dictionary::Pair& kv, l_ScriptFrame.Locals) {
AddSuggestion(matches, word, kv.first);
}
}
{
ObjectLock olock(ScriptGlobal::GetGlobals());
BOOST_FOREACH(const Dictionary::Pair& kv, ScriptGlobal::GetGlobals()) {
AddSuggestion(matches, word, kv.first);
}
}
}
if (state >= matches.size())
return NULL;
return strdup(matches[state].CStr());
}
#endif /* HAVE_EDITLINE */
/** /**
* The entry point for the "console" CLI command. * The entry point for the "console" CLI command.
* *
@ -65,10 +136,13 @@ void ConsoleCommand::InitParameters(boost::program_options::options_description&
*/ */
int ConsoleCommand::Run(const po::variables_map& vm, const std::vector<std::string>& ap) const int ConsoleCommand::Run(const po::variables_map& vm, const std::vector<std::string>& ap) const
{ {
ScriptFrame frame;
std::map<String, String> lines; std::map<String, String> lines;
int next_line = 1; int next_line = 1;
#ifdef HAVE_EDITLINE
rl_completion_entry_function = ConsoleCompleteHelper;
#endif /* HAVE_EDITLINE */
String addr, session; String addr, session;
if (vm.count("connect")) { if (vm.count("connect")) {
@ -139,7 +213,7 @@ incomplete:
expr = ConfigCompiler::CompileText(fileName, command); expr = ConfigCompiler::CompileText(fileName, command);
if (expr) { if (expr) {
Value result = expr->Evaluate(frame); Value result = expr->Evaluate(l_ScriptFrame);
std::cout << ConsoleColorTag(Console_ForegroundCyan); std::cout << ConsoleColorTag(Console_ForegroundCyan);
if (!result.IsObject() || result.IsObjectType<Array>() || result.IsObjectType<Dictionary>()) if (!result.IsObject() || result.IsObjectType<Array>() || result.IsObjectType<Dictionary>())
std::cout << JsonEncode(result); std::cout << JsonEncode(result);

4
lib/cli/editline.hpp
View File

@ -25,6 +25,10 @@ extern "C" {
char *readline(const char *prompt); char *readline(const char *prompt);
int add_history(const char *line); int add_history(const char *line);
typedef char *ELFunction(const char *, int);
extern ELFunction *rl_completion_entry_function;
} }
#endif /* EDITLINE_H */ #endif /* EDITLINE_H */