tyler.durden
/
icinga2
mirror of https://github.com/Icinga/icinga2.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add "repository <type> <add/remove/list>" commands 

refs #7255
This commit is contained in:
Gunnar Beutner 2014-10-17 14:13:31 +02:00
parent 8e7fd1f40d
commit 4632b93b89
5 changed files with 210 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lib/base/clicommand.cpp
View File

@ -69,15 +69,10 @@ static std::vector<String> BashArgumentCompletionHelper(const String& type, cons
return result;
}
void icinga::AddTypeFields(const String& type, boost::program_options::options_description& desc)
void icinga::AddTypeFields(const Type *type, boost::program_options::options_description& desc)
{
const Type *ptype = Type::GetByName(type);
if (!ptype)
return;
for (int i = 0; i < ptype->GetFieldCount(); i++) {
Field field = ptype->GetFieldInfo(i);
for (int i = 0; i < type->GetFieldCount(); i++) {
Field field = type->GetFieldInfo(i);
if (strcmp(field.Name, "__name") == 0)
continue;

2
lib/base/clicommand.hpp
View File

@ -33,7 +33,7 @@ typedef boost::function<std::vector<String> (const String&)> ArgumentCompletionC
typedef std::map<String, ArgumentCompletionCallback> ArgumentCompletionDescription;
I2_BASE_API ArgumentCompletionCallback BashArgumentCompletion(const String& type);
I2_BASE_API void AddTypeFields(const String& type, boost::program_options::options_description& desc);
I2_BASE_API void AddTypeFields(const Type *type, boost::program_options::options_description& desc);
/**
* A CLI command.

1
lib/cli/CMakeLists.txt
View File

@ -19,6 +19,7 @@ set(cli_SOURCES
featureenablecommand.cpp featuredisablecommand.cpp featurelistcommand.cpp
objectlistcommand.cpp
pkinewcacommand.cpp pkinewcertcommand.cpp pkisigncsrcommand.cpp pkirequestcommand.cpp pkiticketcommand.cpp
repositoryobjectcommand.cpp
daemoncommand.cpp
)

127
lib/cli/repositoryobjectcommand.cpp Normal file
View File

@ -0,0 +1,127 @@
/******************************************************************************
* 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/repositoryobjectcommand.hpp"
#include "base/logger_fwd.hpp"
#include "base/clicommand.hpp"
#include "base/application.hpp"
#include "base/tlsutility.hpp"
#include <boost/algorithm/string/case_conv.hpp>
#include <fstream>
using namespace icinga;
REGISTER_REPOSITORY_CLICOMMAND("Host");
REGISTER_REPOSITORY_CLICOMMAND("Service");
REGISTER_REPOSITORY_CLICOMMAND("Zone");
REGISTER_REPOSITORY_CLICOMMAND("Endpoint");
RegisterRepositoryCLICommandHelper::RegisterRepositoryCLICommandHelper(const String& type)
{
String ltype = type;
boost::algorithm::to_lower(ltype);
std::vector<String> name;
name.push_back("repository");
name.push_back(ltype);
name.push_back("add");
CLICommand::Register(name, make_shared<RepositoryObjectCommand>(type, RepositoryCommandAdd));
name[2] = "remove";
CLICommand::Register(name, make_shared<RepositoryObjectCommand>(type, RepositoryCommandRemove));
name[2] = "list";
CLICommand::Register(name, make_shared<RepositoryObjectCommand>(type, RepositoryCommandList));
}
RepositoryObjectCommand::RepositoryObjectCommand(const String& type, RepositoryCommandType command)
: m_Type(type), m_Command(command)
{ }
String RepositoryObjectCommand::GetDescription(void) const
{
String description;
switch (m_Command) {
case RepositoryCommandAdd:
description = "Adds a new";
break;
case RepositoryCommandRemove:
description = "Removes a";
break;
case RepositoryCommandList:
description = "Lists all";
break;
}
description += " " + m_Type + " object";
if (m_Command == RepositoryCommandList)
description += "s";
return description;
}
String RepositoryObjectCommand::GetShortDescription(void) const
{
String description;
switch (m_Command) {
case RepositoryCommandAdd:
description = "adds a new";
break;
case RepositoryCommandRemove:
description = "removes a";
break;
case RepositoryCommandList:
description = "lists all";
break;
}
description += " " + m_Type + " object";
if (m_Command == RepositoryCommandList)
description += "s";
return description;
}
void RepositoryObjectCommand::InitParameters(boost::program_options::options_description& visibleDesc,
boost::program_options::options_description& hiddenDesc,
ArgumentCompletionDescription& argCompletionDesc) const
{
visibleDesc.add_options()
("name", po::value<std::string>(), "The name of the object");
if (m_Command == RepositoryCommandAdd) {
const Type *ptype = Type::GetByName(m_Type);
ASSERT(ptype);
AddTypeFields(ptype, visibleDesc);
}
}
/**
* The entry point for the "repository <type> <add/remove/list>" CLI command.
*
* @returns An exit status.
*/
int RepositoryObjectCommand::Run(const boost::program_options::variables_map& vm, const std::vector<std::string>& ap) const
{
return 0;
}

78
lib/cli/repositoryobjectcommand.hpp Normal file
View File

@ -0,0 +1,78 @@
/******************************************************************************
* 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. *
******************************************************************************/
#ifndef REPOSITORYOBJECTCOMMAND_H
#define REPOSITORYOBJECTCOMMAND_H
#include "base/qstring.hpp"
#include "base/clicommand.hpp"
namespace icinga
{
enum RepositoryCommandType
{
RepositoryCommandAdd,
RepositoryCommandRemove,
RepositoryCommandList
};
/**
* The "repository <type> <add/remove/list>" command.
*
* @ingroup cli
*/
class RepositoryObjectCommand : public CLICommand
{
public:
DECLARE_PTR_TYPEDEFS(RepositoryObjectCommand);
RepositoryObjectCommand(const String& type, RepositoryCommandType command);
virtual String GetDescription(void) const;
virtual String GetShortDescription(void) const;
virtual void InitParameters(boost::program_options::options_description& visibleDesc,
boost::program_options::options_description& hiddenDesc,
ArgumentCompletionDescription& argCompletionDesc) const;
virtual int Run(const boost::program_options::variables_map& vm, const std::vector<std::string>& ap) const;
private:
String m_Type;
RepositoryCommandType m_Command;
};
/**
* Helper class for registering repository CLICommand implementation classes.
*
* @ingroup cli
*/
class I2_BASE_API RegisterRepositoryCLICommandHelper
{
public:
RegisterRepositoryCLICommandHelper(const String& type);
};
#define REGISTER_REPOSITORY_CLICOMMAND(type) \
namespace { namespace UNIQUE_NAME(cli) { \
I2_EXPORT icinga::RegisterRepositoryCLICommandHelper l_RegisterRepositoryCLICommand(type); \
} }
}
#endif /* REPOSITORYOBJECTCOMMAND_H */