1 files changed, 7 insertions, 1 deletions
diff --git a/Swift/Controllers/HistoryViewController.h b/Swift/Controllers/HistoryViewController.h
index 7dabf46..f1ff286 100644
--- a/Swift/Controllers/HistoryViewController.h
+++ b/Swift/Controllers/HistoryViewController.h
@@ -11,6 +11,8 @@
 #include <boost/shared_ptr.hpp>
 
 #include <Swift/Controllers/UIEvents/UIEventStream.h>
+#include <Swiften/JID/JID.h>
+#include <set>
 
 namespace Swift {
 	class HistoryWindowFactory;
@@ -23,7 +25,7 @@ namespace Swift {
 
 	class HistoryViewController {
 		public:
-			HistoryViewController(UIEventStream* uiEventStream, HistoryController* historyController, NickResolver* nickResolver, AvatarManager* avatarManager, HistoryWindowFactory* historyWindowFactory);
+			HistoryViewController(const JID& selfJID, UIEventStream* uiEventStream, HistoryController* historyController, NickResolver* nickResolver, AvatarManager* avatarManager, HistoryWindowFactory* historyWindowFactory);
 			~HistoryViewController();
 
 		private:
@@ -31,6 +33,7 @@ namespace Swift {
 			void handleSelectedContactChanged(RosterItem* item);
 
 		private:
+			JID selfJID_;
 			UIEventStream* uiEventStream_;
 			HistoryController* historyController_;
 			NickResolver* nickResolver_;
@@ -38,5 +41,8 @@ namespace Swift {
 			HistoryWindowFactory* historyWindowFactory_;
 			HistoryWindow* historyWindow_;
 			Roster* roster_;
+
+			std::set<JID> rooms_;
+			std::set<JID> contacts_;
 	};
 }