/*
 * Copyright (c) 2010-2011 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#pragma once

#include <vector>
#include <map>

#include <boost/shared_ptr.hpp>
#include "Swiften/Base/boost_bsignals.h"

#include <string>
#include "Swiften/JID/JID.h"

#include "Swift/Controllers/UIEvents/UIEvent.h"
#include "Swift/Controllers/ProfileSettingsProvider.h"
#include "Swiften/Elements/DiscoInfo.h"
#include "Swiften/Elements/DiscoItems.h"
#include "Swiften/Elements/ErrorPayload.h"

namespace Swift {
	class UIEventStream;
	class MUCSearchWindow;
	class MUCSearchWindowFactory;
	class IQRouter;
	class DiscoServiceWalker;
	class NickResolver;

	class MUCService {
		public:
			class MUCRoom {
				public:
					MUCRoom(const std::string& node, const std::string& name, int occupants) : node_(node), name_(name), occupants_(occupants) {}
					std::string getNode() {return node_;}
					std::string getName() {return name_;}
					int getOccupantCount() {return occupants_;}
				private:
					std::string node_;
					std::string name_;
					int occupants_;
			};

			MUCService() {error_ = false; complete_ = false;}

			void setComplete(bool complete) {
				complete_ = complete;
			}

			void setName(const std::string& name) {
				name_ = name;
			}

			void setJID(const JID& jid) {
				jid_ = jid;
			}

			bool getComplete() const {
				return complete_;
			}

			JID getJID() const {
				return jid_;
			}

			std::string getName() const {
				return name_;
			}

			void setError(const std::string& errorText) {error_ = true; errorText_ = errorText;}

			void clearRooms() {rooms_.clear();}

			void addRoom(const MUCRoom& room) {rooms_.push_back(room);}

			std::vector<MUCRoom> getRooms() const {return rooms_;}
		private:
			std::string name_;
			JID jid_;
			std::vector<MUCRoom> rooms_;
			bool complete_;
			bool error_;
			std::string errorText_;
	};

	class MUCSearchController {
		public:
			MUCSearchController(const JID& jid, MUCSearchWindowFactory* mucSearchWindowFactory, IQRouter* iqRouter, ProfileSettingsProvider* settings);
			~MUCSearchController();

			void openSearchWindow();

		public:
			boost::signal<void (const JID&)> onMUCSelected;

		private:
			void handleSearchService(const JID& jid);
			void handleRoomsItemsResponse(boost::shared_ptr<DiscoItems> items, ErrorPayload::ref error, const JID& jid);
			void handleDiscoError(const JID& jid, ErrorPayload::ref error);
			void handleDiscoServiceFound(const JID&, boost::shared_ptr<DiscoInfo>);
			void handleDiscoWalkFinished();
			void handleMUCSearchFinished(const boost::optional<JID>& result);
			void removeService(const JID& jid);
			void refreshView();
			void loadSavedServices();
			void addToSavedServices(const JID& jid);
			void updateInProgressness();

		private:
			JID jid_;
			MUCSearchWindowFactory* factory_;
			IQRouter* iqRouter_;
			ProfileSettingsProvider* settings_;
			MUCSearchWindow* window_;
			DiscoServiceWalker* walker_;
			std::list<JID> services_;
			std::list<JID> savedServices_;
			std::map<JID, MUCService> serviceDetails_;
			std::vector<DiscoServiceWalker*> walksInProgress_;
			int itemsInProgress_;
	};
}