1 files changed, 9 insertions, 4 deletions
diff --git a/Swift/QtUI/Whiteboard/QtWhiteboardWindow.h b/Swift/QtUI/Whiteboard/QtWhiteboardWindow.h
index f1a9ed5..4c0a065 100644
--- a/Swift/QtUI/Whiteboard/QtWhiteboardWindow.h
+++ b/Swift/QtUI/Whiteboard/QtWhiteboardWindow.h
@@ -8,6 +8,7 @@
 
 #include <Swift/Controllers/UIInterfaces/WhiteboardWindow.h>
 #include <Swiften/Elements/Message.h>
+#include <Swiften/Whiteboard/WhiteboardSession.h>
 
 #include <QWidget>
 #include <QGraphicsView>
@@ -20,19 +21,19 @@
 #include <QSpinBox>
 #include <QColorDialog>
 #include <QToolButton>
+#include <QCloseEvent>
 
 #include "GView.h"
 
 namespace Swift {
-	class WhiteboardSession;
-
 	class QtWhiteboardWindow : public QWidget, public WhiteboardWindow
 	{
 		Q_OBJECT;
 	public:
-		QtWhiteboardWindow(WhiteboardSession* whiteboardSession);
+		QtWhiteboardWindow(WhiteboardSession::ref whiteboardSession);
 		void addItem(const std::string& item);
 		void show();
+		void setSession(WhiteboardSession::ref session);
 
 	private slots:
 		void changeLineWidth(int i);
@@ -50,6 +51,10 @@ namespace Swift {
 		void handleLastItemChanged(QGraphicsItem* item);
 
 	private:
+		void handleSessionTerminate();
+		void closeEvent(QCloseEvent* event);
+
+	private:
 		QGraphicsScene* scene;
 		GView* graphicsView;
 		QVBoxLayout* layout;
@@ -72,6 +77,6 @@ namespace Swift {
 		QToolButton* polygonButton;
 		QToolButton* selectButton;
 
-		WhiteboardSession* whiteboardSession_;
+		WhiteboardSession::ref whiteboardSession_;
 	};
 }