1 files changed, 27 insertions, 23 deletions
diff --git a/Swift/Controllers/AdHocManager.h b/Swift/Controllers/AdHocManager.h
index 47b03cd..b2c34c5 100644
--- a/Swift/Controllers/AdHocManager.h
+++ b/Swift/Controllers/AdHocManager.h
@@ -1,40 +1,44 @@
 /*
- * Copyright (c) 2010-2011 Kevin Smith
+ * Copyright (c) 2010-2014 Kevin Smith
  * Licensed under the GNU General Public License v3.
  * See Documentation/Licenses/GPLv3.txt for more information.
  */
 
-#pragma once
-
-#include <boost/shared_ptr.hpp>
 #include <vector>
 
 #include <Swiften/Base/boost_bsignals.h>
 #include <Swiften/JID/JID.h>
 #include <Swiften/Elements/DiscoInfo.h>
 #include <Swiften/Elements/DiscoItems.h>
 #include <Swiften/Elements/ErrorPayload.h>
 #include <Swiften/Disco/GetDiscoItemsRequest.h>
+#include <Swiften/Client/Client.h>
 #include <Swift/Controllers/UIEvents/UIEvent.h>
+#include <Swift/Controllers/AdHocController.h>
 
 namespace Swift {
-	class IQRouter;
-	class MainWindow;
-	class UIEventStream;
-	class AdHocCommandWindowFactory;
-	class AdHocManager {
-		public:
-			AdHocManager(const JID& jid, AdHocCommandWindowFactory* factory, IQRouter* iqRouter, UIEventStream* uiEventStream, MainWindow* mainWindow);
-			~AdHocManager();
-			void setServerDiscoInfo(boost::shared_ptr<DiscoInfo> info);
-		private:
-			void handleUIEvent(boost::shared_ptr<UIEvent> event);
-			void handleServerDiscoItemsResponse(boost::shared_ptr<DiscoItems>, ErrorPayload::ref error);
-			JID jid_;
-			IQRouter* iqRouter_;
-			UIEventStream* uiEventStream_;
-			MainWindow* mainWindow_;
-			AdHocCommandWindowFactory* factory_;
-			GetDiscoItemsRequest::ref discoItemsRequest_;
-	};
+class IQRouter;
+class MainWindow;
+class UIEventStream;
+class AdHocCommandWindowFactory;
+class AdHocManager {
+public:
+	AdHocManager(const JID& jid, AdHocCommandWindowFactory* factory, IQRouter* iqRouter, UIEventStream* uiEventStream, MainWindow* mainWindow);
+	~AdHocManager();
+	void removeController(boost::shared_ptr<AdHocController> contoller);
+	void setServerDiscoInfo(boost::shared_ptr<DiscoInfo> info);
+	void setOnline(bool online);
+private:
+	void handleServerDiscoItemsResponse(boost::shared_ptr<DiscoItems>, ErrorPayload::ref error);
+	void handleUIEvent(boost::shared_ptr<UIEvent> event);
+	boost::signal<void (const AdHocController&)> onControllerComplete;
+	JID jid_;
+	IQRouter* iqRouter_;
+	UIEventStream* uiEventStream_;
+	MainWindow* mainWindow_;
+	AdHocCommandWindowFactory* factory_;
+	GetDiscoItemsRequest::ref discoItemsRequest_;
+	std::vector<boost::shared_ptr<AdHocController> > controllers_;
+};
+
 }