#include "Swiften/Elements/Message.h"

#include "Swiften/Elements/IQ.h"

#include "Swiften/Elements/Presence.h"

namespace Swift {

	class Error;

	class ServerStanzaChannel : public StanzaChannel {

		public:

			void addSession(boost::shared_ptr<ServerFromClientSession> session);

			void removeSession(boost::shared_ptr<ServerFromClientSession> session);

			void sendIQ(boost::shared_ptr<IQ> iq);

			void sendMessage(boost::shared_ptr<Message> message);

			void sendPresence(boost::shared_ptr<Presence> presence);

			void finishSession(const JID& to, boost::shared_ptr<Element> element, bool last = false);

			bool getStreamManagementEnabled() const {

				return false;

			}

			bool isAvailable() const {

				return true;

			}

		private:

			std::string getNewIQID();

			void send(boost::shared_ptr<Stanza> stanza);

			void handleSessionFinished(const boost::optional<Session::SessionError>&, const boost::shared_ptr<ServerFromClientSession> &session);

			void handleElement(boost::shared_ptr<Element> element, const boost::shared_ptr<ServerFromClientSession> &session);

			void handleDataRead(const SafeByteArray &data, const boost::shared_ptr<ServerFromClientSession> &session);

			void handleSessionInitialized();

		private:

			IDGenerator idGenerator;

			// [JID][resources][ServerFromClientSession]

			std::map<std::string, std::list<boost::shared_ptr<ServerFromClientSession> > > sessions;

	};

}