CLI: Add agent command functionality (WIP)

refs #7248

lib/cli/agentaddcommand.cpp

@@ -18,11 +18,13 @@
  ******************************************************************************/
 
 #include "cli/agentaddcommand.hpp"
+#include "cli/agentutility.hpp"
 #include "base/logger.hpp"
 #include "base/application.hpp"
 #include <boost/foreach.hpp>
 #include <boost/algorithm/string/join.hpp>
 #include <boost/algorithm/string/replace.hpp>
+#include <iostream>
 #include <fstream>
 #include <vector>
 
@@ -53,7 +55,10 @@ int AgentAddCommand::Run(const boost::program_options::variables_map& vm, const
 		return 1;
 	}
 
-	//ap[0] must contain name
+	String name = ap[0];
+
+	if (!AgentUtility::AddAgent(name))
+		return 1;
 
 	return 0;
 }

lib/cli/agentblackandwhitelistcommand.cpp

@@ -21,6 +21,7 @@
 #include "base/logger.hpp"
 #include "base/application.hpp"
 #include <boost/algorithm/string/case_conv.hpp>
+#include <iostream>
 #include <fstream>
 
 using namespace icinga;

lib/cli/agentlistcommand.cpp

@@ -18,11 +18,14 @@
  ******************************************************************************/
 
 #include "cli/agentlistcommand.hpp"
+#include "cli/agentutility.hpp"
 #include "base/logger.hpp"
 #include "base/application.hpp"
+#include "base/console.hpp"
 #include <boost/foreach.hpp>
 #include <boost/algorithm/string/join.hpp>
 #include <boost/algorithm/string/replace.hpp>
+#include <iostream>
 #include <fstream>
 #include <vector>
 
@@ -41,6 +44,13 @@ String AgentListCommand::GetShortDescription(void) const
 	return "lists all agents";
 }
 
+void AgentListCommand::InitParameters(boost::program_options::options_description& visibleDesc,
+    boost::program_options::options_description& hiddenDesc) const
+{
+	visibleDesc.add_options()
+		("batch", "list agents in json");
+}
+
 /**
  * The entry point for the "agent list" CLI command.
  *
@@ -53,5 +63,15 @@ int AgentListCommand::Run(const boost::program_options::variables_map& vm, const
 		    << "Ignoring parameters: " << boost::algorithm::join(ap, " ");
 	}
 
+	if (vm.count("batch")) {
+		AgentUtility::PrintAgentsJson(std::cout);
+		std::cout << "\n";
+		return 0;
+	}
+
+	std::cout << "Configured agents: \n";
+	AgentUtility::PrintAgents(std::cout);
+	std::cout << "\n";
+
 	return 0;
 }

lib/cli/agentlistcommand.hpp

@@ -37,6 +37,8 @@ class AgentListCommand : public CLICommand
 
 	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) const;
 	virtual int Run(const boost::program_options::variables_map& vm, const std::vector<std::string>& ap) const;
 };
 

lib/cli/agentremovecommand.cpp

@@ -18,11 +18,13 @@
  ******************************************************************************/
 
 #include "cli/agentremovecommand.hpp"
+#include "cli/agentutility.hpp"
 #include "base/logger.hpp"
 #include "base/application.hpp"
 #include <boost/foreach.hpp>
 #include <boost/algorithm/string/join.hpp>
 #include <boost/algorithm/string/replace.hpp>
+#include <iostream>
 #include <fstream>
 #include <vector>
 
@@ -33,12 +35,17 @@ REGISTER_CLICOMMAND("agent/remove", AgentRemoveCommand);
 
 String AgentRemoveCommand::GetDescription(void) const
 {
-	return "Remove Icinga 2 agent.";
+	return "Removes Icinga 2 agent.";
 }
 
 String AgentRemoveCommand::GetShortDescription(void) const
 {
-	return "remove agent";
+	return "removes agent";
+}
+
+std::vector<String> AgentRemoveCommand::GetPositionalSuggestions(const String& word) const
+{
+	return AgentUtility::GetFieldCompletionSuggestions(word);
 }
 
 /**
@@ -54,6 +61,10 @@ int AgentRemoveCommand::Run(const boost::program_options::variables_map& vm, con
 	}
 
 	//ap[0] must contain name
+	String name = ap[0];
+
+	if (!AgentUtility::RemoveAgent(name))
+		return 1;
 
 	return 0;
 }

lib/cli/agentremovecommand.hpp

@@ -37,6 +37,7 @@ class AgentRemoveCommand : public CLICommand
 
 	virtual String GetDescription(void) const;
 	virtual String GetShortDescription(void) const;
+	virtual std::vector<String> GetPositionalSuggestions(const String& word) const;
 	virtual int Run(const boost::program_options::variables_map& vm, const std::vector<std::string>& ap) const;
 };
 

lib/cli/agentsetcommand.cpp

@@ -23,6 +23,7 @@
 #include <boost/foreach.hpp>
 #include <boost/algorithm/string/join.hpp>
 #include <boost/algorithm/string/replace.hpp>
+#include <iostream>
 #include <fstream>
 #include <vector>
 
@@ -44,7 +45,7 @@ String AgentSetCommand::GetShortDescription(void) const
 void AgentSetCommand::InitParameters(boost::program_options::options_description& visibleDesc,
     boost::program_options::options_description& hiddenDesc) const
 {
-	/* Command doesn't support any parameters. */
+
 }
 
 /**

lib/cli/agentsetupcommand.cpp

@@ -23,6 +23,7 @@
 #include <boost/foreach.hpp>
 #include <boost/algorithm/string/join.hpp>
 #include <boost/algorithm/string/replace.hpp>
+#include <iostream>
 #include <fstream>
 #include <vector>
 

lib/cli/agentupdateconfigcommand.cpp

@@ -23,6 +23,7 @@
 #include <boost/foreach.hpp>
 #include <boost/algorithm/string/join.hpp>
 #include <boost/algorithm/string/replace.hpp>
+#include <iostream>
 #include <fstream>
 #include <vector>
 

lib/cli/agentutility.cpp

@@ -18,25 +18,207 @@
  ******************************************************************************/
 
 #include "cli/agentutility.hpp"
+#include "cli/clicommand.hpp"
+#include "base/logger.hpp"
+#include "base/application.hpp"
+#include "base/convert.hpp"
+#include "base/serializer.hpp"
+#include "base/netstring.hpp"
+#include "base/stdiostream.hpp"
+#include "base/debug.hpp"
+#include "base/objectlock.hpp"
+#include "base/console.hpp"
+#include <boost/foreach.hpp>
+#include <boost/algorithm/string/join.hpp>
+#include <fstream>
+#include <iostream>
 
 using namespace icinga;
 
-void AgentUtility::ListAgents(void)
+String AgentUtility::GetRepositoryPath(void)
 {
+	return Application::GetLocalStateDir() + "/lib/icinga2/api/repository/";
+}
+
+std::vector<String> AgentUtility::GetFieldCompletionSuggestions(const String& word)
+{
+	std::vector<String> cache;
+	std::vector<String> suggestions;
+
+	GetAgents(cache);
+
+	std::sort(cache.begin(), cache.end());
 
+	BOOST_FOREACH(const String& suggestion, cache) {
+		if (suggestion.Find(word) == 0)
+			suggestions.push_back(suggestion);
+	}
+
+	return suggestions;
+}
+
+void AgentUtility::PrintAgents(std::ostream& fp)
+{
+	std::vector<String> agents;
+	GetAgents(agents);
+
+	BOOST_FOREACH(const String& agent, agents) {
+		Dictionary::Ptr agent_obj = GetAgentFromRepository(GetRepositoryPath() + agent + ".repo");
+		fp << "=======================\n";
+		fp << "Agent Name: " << agent << "\n";
+
+		if (agent_obj) {
+			fp << "Endpoint: " << agent_obj->Get("endpoint") << "\n";
+			fp << "Zone: " << agent_obj->Get("zone") << "\n";
+			fp << "Repository: ";
+			fp << std::setw(4);
+			PrintAgentRepository(fp, agent_obj->Get("repository"));
+			fp << std::setw(0) << "\n";
+		}
+		fp << "=======================\n";
+	}
+
+	fp << "All agents: " << boost::algorithm::join(agents, " ") << "\n";
+}
+
+void AgentUtility::PrintAgentRepository(std::ostream& fp, const Dictionary::Ptr& repository)
+{
+	//TODO better formatting
+	fp << JsonSerialize(repository);
+}
+
+void AgentUtility::PrintAgentsJson(std::ostream& fp)
+{
+	std::vector<String> agents;
+	GetAgents(agents);
+
+	BOOST_FOREACH(const String& agent, agents) {
+		Dictionary::Ptr agent_obj = GetAgentFromRepository(GetRepositoryPath() + agent + ".repo");
+		if (agent_obj) {
+			fp << JsonSerialize(agent_obj);
+		}
+	}
 }
 
 bool AgentUtility::AddAgent(const String& name)
 {
-	return true;
+	String path = GetRepositoryPath() + name + ".repo";
+
+	Dictionary::Ptr agent = make_shared<Dictionary>();
+
+	agent->Set("seen", Utility::GetTime());
+	agent->Set("endpoint", name);
+	agent->Set("zone", name);
+	agent->Set("repository", Empty);
+
+	return WriteAgentToRepository(path, agent);
+}
+
+bool AgentUtility::AddAgentPeer(const String& name, const String& host, const String& port)
+{
+	String path = GetRepositoryPath() + name + ".peer";
+
+	Dictionary::Ptr peer = make_shared<Dictionary>();
+
+	peer->Set("agent_host", host);
+	peer->Set("agent_port", port);
+
+	return WriteAgentToRepository(path, peer);
 }
 
 bool AgentUtility::RemoveAgent(const String& name)
 {
+	String path = GetRepositoryPath() + name + ".repo";
+
+	if (!RemoveAgentFile(path))
+		return false;
+
+	return true;
+}
+
+bool AgentUtility::RemoveAgentFile(const String& path)
+{
+	if (!Utility::PathExists(path) ) {
+		Log(LogCritical, "cli", "Cannot remove '" + path + "'. Does not exist.");
+		return false;
+	}
+
+	if (unlink(path.CStr()) < 0) {
+		Log(LogCritical, "cli", "Cannot remove file '" + path +
+		    "'. Failed with error code " + Convert::ToString(errno) + ", \"" + Utility::FormatErrorNumber(errno) + "\".");
+		return false;
+	}
+
 	return true;
 }
 
-bool AgentUtility::SetAgentAttribute(const String& attr, const Value& val)
+bool AgentUtility::SetAgentAttribute(const String& name, const String& attr, const Value& val)
 {
+	GetAgentFromRepository(GetRepositoryPath() + name + ".repo");
+
+
 	return true;
 }
+
+bool AgentUtility::WriteAgentToRepository(const String& filename, const Dictionary::Ptr& item)
+{
+	Log(LogInformation, "cli", "Dumping config items to file '" + filename + "'");
+
+	String tempFilename = filename + ".tmp";
+
+        std::ofstream fp(tempFilename.CStr(), std::ofstream::out | std::ostream::trunc);
+        fp << JsonSerialize(item);
+        fp.close();
+
+#ifdef _WIN32
+	_unlink(filename.CStr());
+#endif /* _WIN32 */
+
+	if (rename(tempFilename.CStr(), filename.CStr()) < 0) {
+		BOOST_THROW_EXCEPTION(posix_error()
+		    << boost::errinfo_api_function("rename")
+		    << boost::errinfo_errno(errno)
+		    << boost::errinfo_file_name(tempFilename));
+	}
+
+	return true;
+}
+
+Dictionary::Ptr AgentUtility::GetAgentFromRepository(const String& filename)
+{
+	std::fstream fp;
+	fp.open(filename.CStr(), std::ifstream::in);
+
+	if (!fp)
+		return Dictionary::Ptr();
+
+	String content((std::istreambuf_iterator<char>(fp)), std::istreambuf_iterator<char>());
+
+	std::cout << "Content: " << content << "\n";
+
+	fp.close();
+
+	return JsonDeserialize(content);
+}
+
+bool AgentUtility::GetAgents(std::vector<String>& agents)
+{
+	String path = GetRepositoryPath();
+
+	if (!Utility::Glob(path + "/*.repo",
+	    boost::bind(&AgentUtility::CollectAgents, _1, boost::ref(agents)), GlobFile)) {
+		Log(LogCritical, "cli", "Cannot access path '" + path + "'.");
+		return false;
+	}
+
+	return true;
+}
+
+void AgentUtility::CollectAgents(const String& agent_file, std::vector<String>& agents)
+{
+	String agent = Utility::BaseName(agent_file);
+	boost::algorithm::replace_all(agent, ".repo", "");
+
+	Log(LogDebug, "cli", "Adding agent: " + agent);
+	agents.push_back(agent);
+}