1 files changed, 80 insertions, 0 deletions
diff --git a/Slimber/MainController.cpp b/Slimber/MainController.cpp
new file mode 100644
index 0000000..2d196a5
--- /dev/null
+++ b/Slimber/MainController.cpp
@@ -0,0 +1,80 @@
+#include "Slimber/MainController.h"
+
+#include <boost/bind.hpp>
+
+#include "Swiften/Base/foreach.h"
+#include "Swiften/Application/Platform/PlatformApplication.h"
+#include "Swiften/LinkLocal/LinkLocalService.h"
+#include "Swiften/LinkLocal/LinkLocalServiceBrowser.h"
+#include "Swiften/LinkLocal/DNSSD/Bonjour/BonjourQuerier.h"
+#include "Slimber/Server.h"
+#include "Slimber/FileVCardCollection.h"
+#include "Slimber/MenuletController.h"
+#include "Slimber/Menulet.h"
+
+using namespace Swift;
+
+MainController::MainController(Menulet* menulet) : menulet(menulet) {
+	dnsSDQuerier = boost::shared_ptr<BonjourQuerier>(new BonjourQuerier());
+	dnsSDQuerier->start();
+
+	linkLocalServiceBrowser = new LinkLocalServiceBrowser(dnsSDQuerier);
+	linkLocalServiceBrowser->onServiceAdded.connect(
+			boost::bind(&MainController::handleServicesChanged, this));
+	linkLocalServiceBrowser->onServiceRemoved.connect(
+			boost::bind(&MainController::handleServicesChanged, this));
+	linkLocalServiceBrowser->onServiceChanged.connect(
+			boost::bind(&MainController::handleServicesChanged, this));
+	linkLocalServiceBrowser->start();
+
+	vCardCollection = new FileVCardCollection(
+			PlatformApplication("Slimber").getSettingsDir());
+
+	server = new Server(5222, 5562, linkLocalServiceBrowser, vCardCollection);
+	server->onStopped.connect(
+			boost::bind(&MainController::handleServerStopped, this, _1));
+	server->onSelfConnected.connect(
+			boost::bind(&MainController::handleSelfConnected, this, _1));
+
+	menuletController = new MenuletController(menulet);
+
+	handleSelfConnected(false);
+	handleServicesChanged();
+
+	server->start();
+}
+
+MainController::~MainController() {
+	delete menuletController;
+	delete server;
+	delete vCardCollection;
+	linkLocalServiceBrowser->stop();
+	delete linkLocalServiceBrowser;
+	dnsSDQuerier->stop();
+}
+
+void MainController::handleSelfConnected(bool b) {
+	if (b) {
+		menuletController->setXMPPStatus("You are logged in", MenuletController::Online);
+	}
+	else {
+		menuletController->setXMPPStatus("You are not logged in", MenuletController::Offline);
+	}
+}
+
+void MainController::handleServicesChanged() {
+	std::vector<String> names;
+	foreach(const LinkLocalService& service, linkLocalServiceBrowser->getServices()) {
+		names.push_back(service.getDescription());
+	}
+	menuletController->setUserNames(names);
+}
+
+void MainController::handleServerStopped(boost::optional<ServerError> error) {
+	if (error) {
+		menuletController->setXMPPStatus("XMPP Server Error", MenuletController::Offline);
+	}
+	else {
+		menuletController->setXMPPStatus("XMPP Server Not Running", MenuletController::Offline);
+	}
+}