1 files changed, 45 insertions, 35 deletions
diff --git a/Swift/QtUI/QtAdHocCommandWindow.h b/Swift/QtUI/QtAdHocCommandWindow.h
index d42a77d..1135ef9 100644
--- a/Swift/QtUI/QtAdHocCommandWindow.h
+++ b/Swift/QtUI/QtAdHocCommandWindow.h
@@ -1,48 +1,58 @@
 /*
- * Copyright (c) 2010-2012 Kevin Smith
- * Licensed under the GNU General Public License v3.
- * See Documentation/Licenses/GPLv3.txt for more information.
+ * Copyright (c) 2010-2016 Isode Limited.
+ * All rights reserved.
+ * See the COPYING file for more information.
  */
 
 #pragma once
 
-#include <QWidget>
-#include <QPushButton>
+#include <QDialogButtonBox>
 #include <QLabel>
+#include <QPushButton>
+#include <QWidget>
 
-#include <Swift/Controllers/UIInterfaces/AdHocCommandWindow.h>
 #include <Swiften/AdHoc/OutgoingAdHocCommandSession.h>
 
+#include <Swift/Controllers/UIInterfaces/AdHocCommandWindow.h>
+
 class QBoxLayout;
 
 namespace Swift {
-	class QtFormWidget;
-	class QtAdHocCommandWindow : public QWidget, public AdHocCommandWindow {
-		Q_OBJECT
-		public:
-			QtAdHocCommandWindow(boost::shared_ptr<OutgoingAdHocCommandSession> command);
-			virtual ~QtAdHocCommandWindow();
-		private:
-			void handleNextStageReceived(Command::ref command);
-			void handleError(ErrorPayload::ref error);
-			void setForm(Form::ref);
-			void setNoForm(bool andHide);
-			void setAvailableActions(Command::ref commandResult);
-		private slots:
-			void handleCancelClicked();
-			void handlePrevClicked();
-			void handleNextClicked();
-			void handleCompleteClicked();
-		private:
-			boost::shared_ptr<OutgoingAdHocCommandSession> command_;
-			QtFormWidget* formWidget_;
-			Form::ref form_;
-			QLabel* label_;
-			QPushButton* backButton_;
-			QPushButton* nextButton_;
-			QPushButton* completeButton_;
-			QPushButton* cancelButton_;
-			std::map<Command::Action, QPushButton*> actions_;
-			QBoxLayout* layout_;
-	};
+    class QtFormWidget;
+    class QtAdHocCommandWindow : public QWidget, public AdHocCommandWindow {
+        Q_OBJECT
+        public:
+            QtAdHocCommandWindow(std::shared_ptr<OutgoingAdHocCommandSession> command);
+            virtual ~QtAdHocCommandWindow();
+            virtual void setOnline(bool online);
+
+        private:
+            void closeEvent(QCloseEvent* event);
+            void handleNextStageReceived(Command::ref command);
+            void handleError(ErrorPayload::ref error);
+            void setForm(Form::ref);
+            void setNoForm(bool andHide);
+            void setAvailableActions(Command::ref commandResult);
+
+        private slots:
+            void handleCancelClicked();
+            void handlePrevClicked();
+            void handleNextClicked();
+            void handleCompleteClicked();
+
+        private:
+            std::shared_ptr<OutgoingAdHocCommandSession> command_;
+            QtFormWidget* formWidget_;
+            Form::ref form_;
+            QLabel* label_;
+            QLabel* errorLabel_;
+            QPushButton* backButton_;
+            QPushButton* nextButton_;
+            QPushButton* completeButton_;
+            QPushButton* cancelButton_;
+            QPushButton* okButton_;
+            std::map<Command::Action, QPushButton*> actions_;
+            QDialogButtonBox* dialogButtons_;
+            QBoxLayout* layout_;
+    };
 }