1 files changed, 21 insertions, 1 deletions
diff --git a/Swift/Controllers/XMPPEvents/MessageEvent.h b/Swift/Controllers/XMPPEvents/MessageEvent.h
index 7af2be6..12f4c48 100644
--- a/Swift/Controllers/XMPPEvents/MessageEvent.h
+++ b/Swift/Controllers/XMPPEvents/MessageEvent.h
@@ -1,47 +1,67 @@
 /*
- * Copyright (c) 2010-2016 Isode Limited.
+ * Copyright (c) 2010-2017 Isode Limited.
  * All rights reserved.
  * See the COPYING file for more information.
  */
 
 #pragma once
 
 #include <cassert>
 #include <memory>
 
 #include <Swiften/Elements/Message.h>
 
 #include <Swift/Controllers/XMPPEvents/StanzaEvent.h>
 
 namespace Swift {
     class MessageEvent : public StanzaEvent {
         public:
+            class SystemNotification {
+            public:
+                SystemNotification(const std::string& title, const std::string& message) : title(title), message(message) {
+                }
+
+            public:
+                std::string title;
+                std::string message;
+            };
+
+        public:
             typedef std::shared_ptr<MessageEvent> ref;
 
             MessageEvent(std::shared_ptr<Message> stanza) : stanza_(stanza), targetsMe_(true) {}
 
             std::shared_ptr<Message> getStanza() {return stanza_;}
 
             bool isReadable() {
                 return getStanza()->isError() || !getStanza()->getBody().get_value_or("").empty();
             }
 
+            void addNotification(const std::string& title, const std::string& message) {
+                systemNotifications_.push_back(SystemNotification(title, message));
+            }
+
+            const std::vector<SystemNotification>& getNotifications() const {
+                return systemNotifications_;
+            }
+
             void read() {
                 assert (isReadable());
                 conclude();
             }
 
             void setTargetsMe(bool targetsMe) {
                 targetsMe_ = targetsMe;
             }
 
             bool targetsMe() const {
                 return targetsMe_;
             }
 
         private:
             std::shared_ptr<Message> stanza_;
+            std::vector<SystemNotification> systemNotifications_;
             bool targetsMe_;
     };
 }