mirror of https://github.com/Icinga/icinga2.git
349 lines
9.6 KiB
C++
349 lines
9.6 KiB
C++
/******************************************************************************
|
|
* Icinga 2 *
|
|
* Copyright (C) 2012-2014 Icinga Development Team (http://www.icinga.org) *
|
|
* *
|
|
* This program is free software; you can redistribute it and/or *
|
|
* modify it under the terms of the GNU General Public License *
|
|
* as published by the Free Software Foundation; either version 2 *
|
|
* of the License, or (at your option) any later version. *
|
|
* *
|
|
* This program is distributed in the hope that it will be useful, *
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
* GNU General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU General Public License *
|
|
* along with this program; if not, write to the Free Software Foundation *
|
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. *
|
|
******************************************************************************/
|
|
|
|
#include "cli/clicommand.hpp"
|
|
#include "base/logger.hpp"
|
|
#include "base/type.hpp"
|
|
#include "base/serializer.hpp"
|
|
#include <boost/algorithm/string/split.hpp>
|
|
#include <boost/algorithm/string/join.hpp>
|
|
#include <boost/algorithm/string/trim.hpp>
|
|
#include <boost/algorithm/string/classification.hpp>
|
|
#include <boost/foreach.hpp>
|
|
#include <boost/program_options.hpp>
|
|
#include <algorithm>
|
|
#include <iostream>
|
|
|
|
using namespace icinga;
|
|
namespace po = boost::program_options;
|
|
|
|
std::vector<String> icinga::GetBashCompletionSuggestions(const String& type, const String& word)
|
|
{
|
|
std::vector<String> result;
|
|
|
|
#ifndef _WIN32
|
|
String bashArg = "compgen -A " + Utility::EscapeShellArg(type) + " " + Utility::EscapeShellArg(word);
|
|
String cmd = "bash -c " + Utility::EscapeShellArg(bashArg);
|
|
|
|
FILE *fp = popen(cmd.CStr(), "r");
|
|
|
|
char line[4096];
|
|
while (fgets(line, sizeof(line), fp)) {
|
|
String wline = line;
|
|
boost::algorithm::trim_right_if(wline, boost::is_any_of("\r\n"));
|
|
result.push_back(wline);
|
|
}
|
|
fclose(fp);
|
|
|
|
/* Append a slash if there's only one suggestion and it's a directory */
|
|
if ((type == "file" || type == "directory") && result.size() == 1) {
|
|
String path = result[0];
|
|
|
|
struct stat statbuf;
|
|
if (lstat(path.CStr(), &statbuf) >= 0) {
|
|
if (S_ISDIR(statbuf.st_mode)) {
|
|
result.clear(),
|
|
result.push_back(path + "/");
|
|
}
|
|
}
|
|
}
|
|
#endif /* _WIN32 */
|
|
|
|
return result;
|
|
}
|
|
|
|
std::vector<String> icinga::GetFieldCompletionSuggestions(const Type *type, const String& word)
|
|
{
|
|
std::vector<String> result;
|
|
|
|
for (int i = 0; i < type->GetFieldCount(); i++) {
|
|
Field field = type->GetFieldInfo(i);
|
|
|
|
if (!(field.Attributes & FAConfig))
|
|
continue;
|
|
|
|
String fname = field.Name;
|
|
|
|
if (fname == "__name" || fname == "templates" || fname == "type")
|
|
continue;
|
|
|
|
String suggestion = fname + "=";
|
|
|
|
if (suggestion.Find(word) == 0)
|
|
result.push_back(suggestion);
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
boost::mutex& CLICommand::GetRegistryMutex(void)
|
|
{
|
|
static boost::mutex mtx;
|
|
return mtx;
|
|
}
|
|
|
|
std::map<std::vector<String>, CLICommand::Ptr>& CLICommand::GetRegistry(void)
|
|
{
|
|
static std::map<std::vector<String>, CLICommand::Ptr> registry;
|
|
return registry;
|
|
}
|
|
|
|
CLICommand::Ptr CLICommand::GetByName(const std::vector<String>& name)
|
|
{
|
|
boost::mutex::scoped_lock lock(GetRegistryMutex());
|
|
|
|
std::map<std::vector<String>, CLICommand::Ptr>::const_iterator it = GetRegistry().find(name);
|
|
|
|
if (it == GetRegistry().end())
|
|
return CLICommand::Ptr();
|
|
|
|
return it->second;
|
|
}
|
|
|
|
void CLICommand::Register(const std::vector<String>& name, const CLICommand::Ptr& function)
|
|
{
|
|
boost::mutex::scoped_lock lock(GetRegistryMutex());
|
|
GetRegistry()[name] = function;
|
|
}
|
|
|
|
void CLICommand::Unregister(const std::vector<String>& name)
|
|
{
|
|
boost::mutex::scoped_lock lock(GetRegistryMutex());
|
|
GetRegistry().erase(name);
|
|
}
|
|
|
|
RegisterCLICommandHelper::RegisterCLICommandHelper(const String& name, const CLICommand::Ptr& command)
|
|
{
|
|
std::vector<String> vname;
|
|
boost::algorithm::split(vname, name, boost::is_any_of("/"));
|
|
CLICommand::Register(vname, command);
|
|
}
|
|
|
|
std::vector<String> CLICommand::GetArgumentSuggestions(const String& argument, const String& word) const
|
|
{
|
|
return std::vector<String>();
|
|
}
|
|
|
|
std::vector<String> CLICommand::GetPositionalSuggestions(const String& word) const
|
|
{
|
|
return std::vector<String>();
|
|
}
|
|
|
|
void CLICommand::InitParameters(boost::program_options::options_description& visibleDesc,
|
|
boost::program_options::options_description& hiddenDesc) const
|
|
{ }
|
|
|
|
bool CLICommand::ParseCommand(int argc, char **argv, po::options_description& visibleDesc,
|
|
po::options_description& hiddenDesc,
|
|
po::positional_options_description& positionalDesc,
|
|
po::variables_map& vm, String& cmdname, CLICommand::Ptr& command, bool autocomplete)
|
|
{
|
|
boost::mutex::scoped_lock lock(GetRegistryMutex());
|
|
|
|
typedef std::map<std::vector<String>, CLICommand::Ptr>::value_type CLIKeyValue;
|
|
|
|
std::vector<String> best_match;
|
|
int arg_end = 1;
|
|
|
|
BOOST_FOREACH(const CLIKeyValue& kv, GetRegistry()) {
|
|
const std::vector<String>& vname = kv.first;
|
|
|
|
for (int i = 0, k = 1; i < vname.size() && k < argc; i++, k++) {
|
|
if (strcmp(argv[k], "--no-stack-rlimit") == 0 || strcmp(argv[k], "--autocomplete") == 0 || strcmp(argv[k], "--scm") == 0) {
|
|
i--;
|
|
continue;
|
|
}
|
|
|
|
if (vname[i] != argv[k])
|
|
break;
|
|
|
|
if (i >= best_match.size())
|
|
best_match.push_back(vname[i]);
|
|
|
|
if (i == vname.size() - 1) {
|
|
cmdname = boost::algorithm::join(vname, " ");
|
|
command = kv.second;
|
|
arg_end = k;
|
|
goto found_command;
|
|
}
|
|
}
|
|
}
|
|
|
|
found_command:
|
|
lock.unlock();
|
|
|
|
po::options_description vdesc("Command options");
|
|
|
|
if (command)
|
|
command->InitParameters(vdesc, hiddenDesc);
|
|
|
|
visibleDesc.add(vdesc);
|
|
|
|
if (autocomplete)
|
|
return true;
|
|
|
|
po::options_description adesc;
|
|
adesc.add(visibleDesc);
|
|
adesc.add(hiddenDesc);
|
|
|
|
po::store(po::command_line_parser(argc - arg_end, argv + arg_end).options(adesc).positional(positionalDesc).run(), vm);
|
|
po::notify(vm);
|
|
|
|
return true;
|
|
}
|
|
|
|
void CLICommand::ShowCommands(int argc, char **argv, po::options_description *visibleDesc,
|
|
po::options_description *hiddenDesc,
|
|
ArgumentCompletionCallback globalArgCompletionCallback,
|
|
bool autocomplete, int autoindex)
|
|
{
|
|
boost::mutex::scoped_lock lock(GetRegistryMutex());
|
|
|
|
typedef std::map<std::vector<String>, CLICommand::Ptr>::value_type CLIKeyValue;
|
|
|
|
std::vector<String> best_match;
|
|
int arg_begin = 0;
|
|
CLICommand::Ptr command;
|
|
|
|
BOOST_FOREACH(const CLIKeyValue& kv, GetRegistry()) {
|
|
const std::vector<String>& vname = kv.first;
|
|
|
|
arg_begin = 0;
|
|
|
|
for (int i = 0, k = 1; i < vname.size() && k < argc; i++, k++) {
|
|
if (strcmp(argv[k], "--no-stack-rlimit") == 0 || strcmp(argv[k], "--autocomplete") == 0 || strcmp(argv[k], "--scm") == 0) {
|
|
i--;
|
|
arg_begin++;
|
|
continue;
|
|
}
|
|
|
|
if (vname[i] != argv[k])
|
|
break;
|
|
|
|
if (i >= best_match.size()) {
|
|
best_match.push_back(vname[i]);
|
|
}
|
|
|
|
if (i == vname.size() - 1) {
|
|
command = kv.second;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
String aword;
|
|
|
|
if (autocomplete) {
|
|
if (autoindex < argc)
|
|
aword = argv[autoindex];
|
|
|
|
if (autoindex - 1 > best_match.size() && !command)
|
|
return;
|
|
} else
|
|
std::cout << "Supported commands: " << std::endl;
|
|
|
|
BOOST_FOREACH(const CLIKeyValue& kv, GetRegistry()) {
|
|
const std::vector<String>& vname = kv.first;
|
|
|
|
if (vname.size() < best_match.size())
|
|
continue;
|
|
|
|
bool match = true;
|
|
|
|
for (int i = 0; i < best_match.size(); i++) {
|
|
if (vname[i] != best_match[i]) {
|
|
match = false;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!match)
|
|
continue;
|
|
|
|
if (autocomplete) {
|
|
String cname;
|
|
|
|
if (autoindex - 1 < vname.size()) {
|
|
cname = vname[autoindex - 1];
|
|
|
|
if (cname.Find(aword) == 0)
|
|
std::cout << cname << "\n";
|
|
}
|
|
} else
|
|
std::cout << " * " << boost::algorithm::join(vname, " ") << " (" << kv.second->GetShortDescription() << ")" << std::endl;
|
|
}
|
|
|
|
if (!autocomplete)
|
|
std::cout << std::endl;
|
|
|
|
if (command && autocomplete) {
|
|
String aname, prefix, pword;
|
|
const po::option_description *odesc;
|
|
|
|
if (autoindex - 2 >= 0 && strcmp(argv[autoindex - 1], "=") == 0 && strstr(argv[autoindex - 2], "--") == argv[autoindex - 2]) {
|
|
aname = argv[autoindex - 2] + 2;
|
|
pword = aword;
|
|
} else if (autoindex - 1 >= 0 && argv[autoindex - 1][0] == '-' && argv[autoindex - 1][1] == '-') {
|
|
aname = argv[autoindex - 1] + 2;
|
|
pword = aword;
|
|
|
|
if (pword == "=")
|
|
pword = "";
|
|
} else if (aword.GetLength() > 1 && aword[0] == '-' && aword[1] != '-') {
|
|
aname = aword.SubStr(1, 1);
|
|
prefix = aword.SubStr(0, 2);
|
|
pword = aword.SubStr(2);
|
|
} else {
|
|
goto complete_option;
|
|
}
|
|
|
|
odesc = visibleDesc->find_nothrow(aname, false);
|
|
|
|
if (!odesc)
|
|
return;
|
|
|
|
if (odesc->semantic()->min_tokens() == 0)
|
|
goto complete_option;
|
|
|
|
BOOST_FOREACH(const String& suggestion, globalArgCompletionCallback(aname, pword)) {
|
|
std::cout << prefix << suggestion << "\n";
|
|
}
|
|
|
|
BOOST_FOREACH(const String& suggestion, command->GetArgumentSuggestions(aname, pword)) {
|
|
std::cout << prefix << suggestion << "\n";
|
|
}
|
|
|
|
return;
|
|
|
|
complete_option:
|
|
BOOST_FOREACH(const shared_ptr<po::option_description>& odesc, visibleDesc->options()) {
|
|
String cname = "--" + odesc->long_name();
|
|
|
|
if (cname.Find(aword) == 0)
|
|
std::cout << cname << "\n";
|
|
}
|
|
|
|
BOOST_FOREACH(const String& suggestion, command->GetPositionalSuggestions(aword)) {
|
|
std::cout << suggestion << "\n";
|
|
}
|
|
}
|
|
|
|
return;
|
|
}
|