1 files changed, 16 insertions, 3 deletions
diff --git a/Swift/Controllers/UIEvents/AddContactUIEvent.h b/Swift/Controllers/UIEvents/AddContactUIEvent.h
index b2bf5ce..d92d3af 100644
--- a/Swift/Controllers/UIEvents/AddContactUIEvent.h
+++ b/Swift/Controllers/UIEvents/AddContactUIEvent.h
@@ -7,17 +7,30 @@
 #pragma once
 
 #include <string>
+#include <set>
 
 #include "Swift/Controllers/UIEvents/UIEvent.h"
 
 namespace Swift {
 	class AddContactUIEvent : public UIEvent {
 		public:
-			AddContactUIEvent(const JID& jid, const std::string& name) : jid_(jid), name_(name) {};
-			std::string getName() {return name_;};
-			JID getJID() {return jid_;};
+			AddContactUIEvent(const JID& jid, const std::string& name, const std::set<std::string>& groups) : jid_(jid), name_(name), groups_(groups) {};
+
+			const std::string& getName() const {
+				return name_;
+			};
+
+			const JID& getJID() const {
+				return jid_;
+			};
+
+			const std::set<std::string>& getGroups() const {
+				return groups_;
+			}
+
 		private:
 			JID jid_;
 			std::string name_;
+			std::set<std::string> groups_;
 	};
 }