1 files changed, 46 insertions, 44 deletions
diff --git a/Swift/Controllers/XMPPEvents/EventController.cpp b/Swift/Controllers/XMPPEvents/EventController.cpp
index 8cb259b..0e9429d 100644
--- a/Swift/Controllers/XMPPEvents/EventController.cpp
+++ b/Swift/Controllers/XMPPEvents/EventController.cpp
@@ -1,20 +1,21 @@
 /*
- * Copyright (c) 2010-2012 Kevin Smith
- * Licensed under the GNU General Public License v3.
- * See Documentation/Licenses/GPLv3.txt for more information.
+ * Copyright (c) 2010-2018 Isode Limited.
+ * All rights reserved.
+ * See the COPYING file for more information.
  */
 
 #include <Swift/Controllers/XMPPEvents/EventController.h>
 
+#include <algorithm>
+
 #include <boost/bind.hpp>
 #include <boost/numeric/conversion/cast.hpp>
-#include <algorithm>
 
-#include <Swiften/Base/foreach.h>
-#include <Swift/Controllers/XMPPEvents/MessageEvent.h>
 #include <Swift/Controllers/XMPPEvents/ErrorEvent.h>
-#include <Swift/Controllers/XMPPEvents/SubscriptionRequestEvent.h>
+#include <Swift/Controllers/XMPPEvents/IncomingFileTransferEvent.h>
 #include <Swift/Controllers/XMPPEvents/MUCInviteEvent.h>
+#include <Swift/Controllers/XMPPEvents/MessageEvent.h>
+#include <Swift/Controllers/XMPPEvents/SubscriptionRequestEvent.h>
 
 namespace Swift {
 
@@ -22,55 +23,56 @@ EventController::EventController() {
 }
 
 EventController::~EventController() {
-	foreach(boost::shared_ptr<StanzaEvent> event, events_) {
-		event->onConclusion.disconnect(boost::bind(&EventController::handleEventConcluded, this, event));
-	}
+    for (auto&& event : events_) {
+        event->onConclusion.disconnect(boost::bind(&EventController::handleEventConcluded, this, event));
+    }
 }
 
-void EventController::handleIncomingEvent(boost::shared_ptr<StanzaEvent> sourceEvent) {
-	boost::shared_ptr<MessageEvent> messageEvent = boost::dynamic_pointer_cast<MessageEvent>(sourceEvent);
-	boost::shared_ptr<SubscriptionRequestEvent> subscriptionEvent = boost::dynamic_pointer_cast<SubscriptionRequestEvent>(sourceEvent);
-	boost::shared_ptr<ErrorEvent> errorEvent = boost::dynamic_pointer_cast<ErrorEvent>(sourceEvent);
-	boost::shared_ptr<MUCInviteEvent> mucInviteEvent = boost::dynamic_pointer_cast<MUCInviteEvent>(sourceEvent);
+void EventController::handleIncomingEvent(std::shared_ptr<StanzaEvent> sourceEvent) {
+    std::shared_ptr<MessageEvent> messageEvent = std::dynamic_pointer_cast<MessageEvent>(sourceEvent);
+    std::shared_ptr<SubscriptionRequestEvent> subscriptionEvent = std::dynamic_pointer_cast<SubscriptionRequestEvent>(sourceEvent);
+    std::shared_ptr<ErrorEvent> errorEvent = std::dynamic_pointer_cast<ErrorEvent>(sourceEvent);
+    std::shared_ptr<MUCInviteEvent> mucInviteEvent = std::dynamic_pointer_cast<MUCInviteEvent>(sourceEvent);
+    std::shared_ptr<IncomingFileTransferEvent> incomingFileTransferEvent = std::dynamic_pointer_cast<IncomingFileTransferEvent>(sourceEvent);
 
-	/* If it's a duplicate subscription request, remove the previous request first */
-	if (subscriptionEvent) {
-		EventList existingEvents(events_);
-		foreach(boost::shared_ptr<StanzaEvent> existingEvent, existingEvents) {
-			boost::shared_ptr<SubscriptionRequestEvent> existingSubscriptionEvent = boost::dynamic_pointer_cast<SubscriptionRequestEvent>(existingEvent);
-			if (existingSubscriptionEvent) {
-				if (existingSubscriptionEvent->getJID() == subscriptionEvent->getJID()) {
-					existingEvent->conclude();
-				}
-			}
-		}
-	}
+    /* If it's a duplicate subscription request, remove the previous request first */
+    if (subscriptionEvent) {
+        EventList existingEvents(events_);
+        for (auto&& existingEvent : existingEvents) {
+            std::shared_ptr<SubscriptionRequestEvent> existingSubscriptionEvent = std::dynamic_pointer_cast<SubscriptionRequestEvent>(existingEvent);
+            if (existingSubscriptionEvent) {
+                if (existingSubscriptionEvent->getJID() == subscriptionEvent->getJID()) {
+                    existingEvent->conclude();
+                }
+            }
+        }
+    }
 
-	if ((messageEvent && messageEvent->isReadable()) || subscriptionEvent || errorEvent || mucInviteEvent) {
-		events_.push_back(sourceEvent);
-		sourceEvent->onConclusion.connect(boost::bind(&EventController::handleEventConcluded, this, sourceEvent));
-		onEventQueueLengthChange(boost::numeric_cast<int>(events_.size()));
-		onEventQueueEventAdded(sourceEvent);
-		if (sourceEvent->getConcluded()) {
-			handleEventConcluded(sourceEvent);
-		}
-	}
+    if ((messageEvent && messageEvent->isReadable()) || subscriptionEvent || errorEvent || mucInviteEvent || incomingFileTransferEvent) {
+        events_.push_back(sourceEvent);
+        sourceEvent->onConclusion.connect(boost::bind(&EventController::handleEventConcluded, this, sourceEvent));
+        onEventQueueLengthChange(events_.size());
+        onEventQueueEventAdded(sourceEvent);
+        if (sourceEvent->getConcluded()) {
+            handleEventConcluded(sourceEvent);
+        }
+    }
 }
 
-void EventController::handleEventConcluded(boost::shared_ptr<StanzaEvent> event) {
-	event->onConclusion.disconnect(boost::bind(&EventController::handleEventConcluded, this, event));
-	events_.erase(std::remove(events_.begin(), events_.end(), event), events_.end());
-	onEventQueueLengthChange(boost::numeric_cast<int>(events_.size()));
+void EventController::handleEventConcluded(std::shared_ptr<StanzaEvent> event) {
+    event->onConclusion.disconnect(boost::bind(&EventController::handleEventConcluded, this, event));
+    events_.erase(std::remove(events_.begin(), events_.end(), event), events_.end());
+    onEventQueueLengthChange(events_.size());
 }
 
 void EventController::disconnectAll() {
-	onEventQueueLengthChange.disconnect_all_slots();
-	onEventQueueEventAdded.disconnect_all_slots();
+    onEventQueueLengthChange.disconnect_all_slots();
+    onEventQueueEventAdded.disconnect_all_slots();
 }
 
 void EventController::clear() {
-	events_.clear();
-	onEventQueueLengthChange(0);
+    events_.clear();
+    onEventQueueLengthChange(0);
 }
 
 }