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

#pragma once

#include <map>

#include <Swiften/Base/API.h>
#include <Swiften/Queries/GetResponder.h>
#include <Swiften/Elements/DiscoInfo.h>
#include <Swiften/JID/JID.h>

namespace Swift {
	class IQRouter;

	class SWIFTEN_API JIDDiscoInfoResponder : public GetResponder<DiscoInfo> {
		public:
			JIDDiscoInfoResponder(IQRouter* router);

			void clearDiscoInfo(const JID& jid);
			void setDiscoInfo(const JID& jid, const DiscoInfo& info);
			void setDiscoInfo(const JID& jid, const std::string& node, const DiscoInfo& info);

		private:
			virtual bool handleGetRequest(const JID& from, const JID& to, const std::string& id, boost::shared_ptr<DiscoInfo> payload);

		private:
			struct JIDDiscoInfo {
				DiscoInfo discoInfo;
				std::map<std::string, DiscoInfo> nodeDiscoInfo;
			};
			typedef std::map<JID, JIDDiscoInfo> JIDDiscoInfoMap;
			JIDDiscoInfoMap info;
	};
}