1 files changed, 4 insertions, 1 deletions
diff --git a/Swift/Controllers/HistoryController.h b/Swift/Controllers/HistoryController.h
index 9949c19..6ead3f1 100644
--- a/Swift/Controllers/HistoryController.h
+++ b/Swift/Controllers/HistoryController.h
@@ -8,9 +8,11 @@
 
 #include <Swiften/JID/JID.h>
 #include <boost/date_time/posix_time/posix_time.hpp>
+#include <vector>
 
 namespace Swift {
 	class HistoryManager;
+	class HistoryMessage;
 	class JID;
 
 	class HistoryController {
@@ -18,7 +20,8 @@ namespace Swift {
 			HistoryController();
 			~HistoryController();
 
-			void addMessage(const JID& baseJID, const JID& fromJID, const JID& toJID, std::string messageBody, boost::posix_time::ptime timeStamp);
+			void addMessage(const JID& baseJID, const JID& fromJID, const std::string& displayNick, const std::string& messageBody, boost::posix_time::ptime timeStamp);
+			std::vector<HistoryMessage> getMessages() const;
 
 		private:
 			HistoryManager* localHistory_;