1 files changed, 22 insertions, 0 deletions
diff --git a/Swift/Controllers/Roster/Roster.cpp b/Swift/Controllers/Roster/Roster.cpp
index 84561e5..77d6b78 100644
--- a/Swift/Controllers/Roster/Roster.cpp
+++ b/Swift/Controllers/Roster/Roster.cpp
@@ -48,12 +48,34 @@ Roster::~Roster() {
 }
 
 GroupRosterItem* Roster::getRoot() const {
 	return root_;
 }
 
+std::set<JID> Roster::getJIDs() const {
+	std::set<JID> jids;
+
+	std::deque<RosterItem*> queue;
+	queue.push_back(root_);
+	while (!queue.empty()) {
+		RosterItem* item = *queue.begin();
+		queue.pop_front();
+		GroupRosterItem* group = dynamic_cast<GroupRosterItem*>(item);
+		ContactRosterItem *contact = dynamic_cast<ContactRosterItem*>(item);
+		if (contact) {
+			jids.insert(contact->getJID());
+			jids.insert(contact->getDisplayJID());
+		}
+		else if (group) {
+			queue.insert(queue.begin(), group->getChildren().begin(), group->getChildren().end());
+		}
+	}
+
+	return jids;
+}
+
 GroupRosterItem* Roster::getGroup(const std::string& groupName) {
 	foreach (RosterItem *item, root_->getChildren()) {
 		GroupRosterItem *group = dynamic_cast<GroupRosterItem*>(item);
 		if (group && group->getDisplayName() == groupName) {
 			return group;
 		}