1 files changed, 99 insertions, 0 deletions
diff --git a/Swift/Controllers/DiscoServiceWalker.cpp b/Swift/Controllers/DiscoServiceWalker.cpp
new file mode 100644
index 0000000..95ce23b
--- /dev/null
+++ b/Swift/Controllers/DiscoServiceWalker.cpp
@@ -0,0 +1,99 @@
+/*
+ * Copyright (c) 2010 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include <Swift/Controllers/DiscoServiceWalker.h>
+
+#include <boost/bind.hpp>
+
+#include "Swiften/Disco/GetDiscoInfoRequest.h"
+#include "Swiften/Disco/GetDiscoItemsRequest.h"
+
+namespace Swift {
+
+DiscoServiceWalker::DiscoServiceWalker(const JID& service, IQRouter* iqRouter, size_t maxSteps) : service_(service), iqRouter_(iqRouter), maxSteps_(maxSteps) {
+
+}
+
+void DiscoServiceWalker::beginWalk() {
+	assert(servicesBeingSearched_.size() == 0);
+	walkNode(service_);
+}
+
+void DiscoServiceWalker::walkNode(const JID& jid) {
+	servicesBeingSearched_.push_back(jid);
+	searchedServices_.push_back(jid);
+	GetDiscoInfoRequest::ref discoInfoRequest = GetDiscoInfoRequest::create(jid, iqRouter_);
+	discoInfoRequest->onResponse.connect(boost::bind(&DiscoServiceWalker::handleDiscoInfoResponse, this, _1, _2, jid));
+	discoInfoRequest->send();
+}
+
+void DiscoServiceWalker::handleReceivedDiscoItem(const JID& item) {
+	if (std::find(searchedServices_.begin(), searchedServices_.end(), item) != searchedServices_.end()) {
+		/* Don't recurse infinitely */
+		return;
+	}
+	walkNode(item);
+}
+
+void DiscoServiceWalker::handleDiscoInfoResponse(boost::shared_ptr<DiscoInfo> info, ErrorPayload::ref error, const JID& jid) {
+	if (error) {
+		handleDiscoError(jid, error);
+		return;
+	}
+
+	bool couldContainServices = false;
+	foreach (DiscoInfo::Identity identity, info->getIdentities()) {
+		if (identity.getCategory() == "server") {
+			couldContainServices = true;
+		}
+	}
+	bool completed = false;
+	if (couldContainServices) {
+		GetDiscoItemsRequest::ref discoItemsRequest = GetDiscoItemsRequest::create(jid, iqRouter_);
+		discoItemsRequest->onResponse.connect(boost::bind(&DiscoServiceWalker::handleDiscoItemsResponse, this, _1, _2, jid));
+		discoItemsRequest->send();
+	} else {
+		completed = true;
+	}
+	onServiceFound(jid, info);
+	if (completed) {
+		markNodeCompleted(jid);
+	}
+}
+
+void DiscoServiceWalker::handleDiscoItemsResponse(boost::shared_ptr<DiscoItems> items, ErrorPayload::ref error, const JID& jid) {
+	if (error) {
+		handleDiscoError(jid, error);
+		return;
+	}
+	foreach (DiscoItems::Item item, items->getItems()) {
+		if (item.getNode().isEmpty()) {
+			/* Don't look at noded items. It's possible that this will exclude some services,
+			 * but I've never seen one in the wild, and it's an easy fix for not looping.
+			 */
+			handleReceivedDiscoItem(item.getJID());
+		}
+	}
+	markNodeCompleted(jid);
+}
+
+void DiscoServiceWalker::handleDiscoError(const JID& jid, ErrorPayload::ref /*error*/) {
+	markNodeCompleted(jid);
+}
+
+void DiscoServiceWalker::markNodeCompleted(const JID& jid) {
+	servicesBeingSearched_.erase(std::remove(servicesBeingSearched_.begin(), servicesBeingSearched_.end(), jid), servicesBeingSearched_.end());
+	/* All results are in */
+	if (servicesBeingSearched_.size() == 0) {
+		onWalkComplete();
+	}
+	/* Check if we're on a rampage */
+	if (searchedServices_.size() >= maxSteps_) {
+		onWalkComplete();
+	}
+}
+
+}