1 files changed, 9 insertions, 1 deletions
diff --git a/Swift/QtUI/Whiteboard/QtWhiteboardWindow.h b/Swift/QtUI/Whiteboard/QtWhiteboardWindow.h
index f472629..9ced322 100644
--- a/Swift/QtUI/Whiteboard/QtWhiteboardWindow.h
+++ b/Swift/QtUI/Whiteboard/QtWhiteboardWindow.h
@@ -7,6 +7,8 @@
 #pragma once
 
 #include <Swift/Controllers/UIInterfaces/WhiteboardWindow.h>
+#include <Swiften/Elements/Message.h>
+#include "Swiften/JID/JID.h"
 
 #include <QWidget>
 #include <QGraphicsView>
@@ -23,11 +25,14 @@
 #include "GView.h"
 
 namespace Swift {
+	class StanzaChannel;
+
 	class QtWhiteboardWindow : public QWidget, public WhiteboardWindow
 	{
 		Q_OBJECT;
 	public:
-		QtWhiteboardWindow();
+		QtWhiteboardWindow(StanzaChannel *stanzaChannel, const JID& jid);
+		void handleIncommingMessage(boost::shared_ptr<Message> message);
 		void addItem(const std::string& item);
 		void show();
 
@@ -67,5 +72,8 @@ namespace Swift {
 		QToolButton* textButton;
 		QToolButton* polygonButton;
 		QToolButton* selectButton;
+
+		StanzaChannel* stanzaChannel_;
+		JID jid_;
 	};
 }