From 981513b2a52caccaf5ff1f453d612f1ad81b1a2b Mon Sep 17 00:00:00 2001
From: Jan Kaluza <hanzz.k@gmail.com>
Date: Wed, 18 Jul 2012 13:05:50 +0200
Subject: [PATCH] Added 'register' and 'unregister' command to AdminInterface

---
 include/transport/admininterface.h |  4 ++-
 spectrum/src/main.cpp              |  2 +-
 src/admininterface.cpp             | 48 +++++++++++++++++++++++++++++-
 3 files changed, 51 insertions(+), 3 deletions(-)

diff --git a/include/transport/admininterface.h b/include/transport/admininterface.h
index 393394eb..324d4dd6 100644
--- a/include/transport/admininterface.h
+++ b/include/transport/admininterface.h
@@ -30,10 +30,11 @@ class Component;
 class StorageBackend;
 class UserManager;
 class NetworkPluginServer;
+class UserRegistration;
 
 class AdminInterface {
 	public:
-		AdminInterface(Component *component, UserManager *userManager, NetworkPluginServer *server = NULL, StorageBackend *storageBackend = NULL);
+		AdminInterface(Component *component, UserManager *userManager, NetworkPluginServer *server = NULL, StorageBackend *storageBackend = NULL, UserRegistration *userRegistration = NULL);
 
 		~AdminInterface();
 
@@ -44,6 +45,7 @@ class AdminInterface {
 		StorageBackend *m_storageBackend;
 		UserManager *m_userManager;
 		NetworkPluginServer *m_server;
+		UserRegistration *m_userRegistration;
 };
 
 }
diff --git a/spectrum/src/main.cpp b/spectrum/src/main.cpp
index 45d8b594..0408d16b 100644
--- a/spectrum/src/main.cpp
+++ b/spectrum/src/main.cpp
@@ -309,7 +309,7 @@ int main(int argc, char **argv)
 
 	NetworkPluginServer plugin(&transport, &config, &userManager, &ftManager);
 
-	AdminInterface adminInterface(&transport, &userManager, &plugin, storageBackend);
+	AdminInterface adminInterface(&transport, &userManager, &plugin, storageBackend, userRegistration);
 	StatsResponder statsResponder(&transport, &userManager, &plugin, storageBackend);
 	statsResponder.start();
 
diff --git a/src/admininterface.cpp b/src/admininterface.cpp
index db3c53a6..913484be 100644
--- a/src/admininterface.cpp
+++ b/src/admininterface.cpp
@@ -27,6 +27,7 @@
 #include "transport/usermanager.h"
 #include "transport/networkpluginserver.h"
 #include "transport/logging.h"
+#include "transport/userregistration.h"
 #include "storageresponder.h"
 #include "transport/memoryusage.h"
 #include <boost/foreach.hpp>
@@ -43,11 +44,12 @@ static std::string getArg(const std::string &body) {
 	return body.substr(body.find(" ") + 1);
 }
 
-AdminInterface::AdminInterface(Component *component, UserManager *userManager, NetworkPluginServer *server, StorageBackend *storageBackend) {
+AdminInterface::AdminInterface(Component *component, UserManager *userManager, NetworkPluginServer *server, StorageBackend *storageBackend, UserRegistration *userRegistration) {
 	m_component = component;
 	m_storageBackend = storageBackend;
 	m_userManager = userManager;
 	m_server = server;
+	m_userRegistration = userRegistration;
 
 	m_component->getStanzaChannel()->onMessageReceived.connect(bind(&AdminInterface::handleMessageReceived, this, _1));
 }
@@ -267,6 +269,46 @@ void AdminInterface::handleMessageReceived(Swift::Message::ref message) {
 		int msgCount = m_userManager->getMessagesToXMPP();
 		message->setBody(boost::lexical_cast<std::string>(msgCount));
 	}
+	else if (message->getBody().find("register ") == 0 && m_userRegistration) {
+		std::string body = message->getBody();
+		std::vector<std::string> args;
+		boost::split(args, body, boost::is_any_of(" "));
+		if (args.size() == 4) {
+			UserInfo res;
+			res.jid = args[1];
+			res.uin = args[2];
+			res.password = args[3];
+			res.language = "en";
+			res.encoding = "utf-8";
+			res.vip = 0;
+
+			if (m_userRegistration->registerUser(res)) {
+				message->setBody("User registered.");
+			}
+			else {
+				message->setBody("Registration failed: User is already registered");
+			}
+		}
+		else {
+			message->setBody("Bad argument count. See 'help'.");
+		}
+	}
+	else if (message->getBody().find("unregister ") == 0 && m_userRegistration) {
+		std::string body = message->getBody();
+		std::vector<std::string> args;
+		boost::split(args, body, boost::is_any_of(" "));
+		if (args.size() == 2) {
+			if (m_userRegistration->unregisterUser(args[1])) {
+				message->setBody("User unregistered.");
+			}
+			else {
+				message->setBody("Registration failed: User is not registered");
+			}
+		}
+		else {
+			message->setBody("Bad argument count. See 'help'.");
+		}
+	}
 	else if (message->getBody().find("help") == 0) {
 		std::string help;
 		help += "General:\n";
@@ -277,6 +319,10 @@ void AdminInterface::handleMessageReceived(Swift::Message::ref message) {
 		help += "    online_users_count - number of online users\n";
 		help += "    online_users_per_backend - shows online users per backends\n";
 		help += "    has_online_user <bare_JID> - returns 1 if user is online\n";
+		if (m_userRegistration) {
+			help += "    register <bare_JID> <legacyName> <password> - registers the new user\n";
+			help += "    unregister <bare_JID> - unregisters existing user\n";
+		}
 		help += "Messages:\n";
 		help += "    messages_from_xmpp - get number of messages received from XMPP users\n";
 		help += "    messages_to_xmpp - get number of messages sent to XMPP users\n";