#pragma once

#include <boost/filesystem.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/optional.hpp>
#include <boost/signal.hpp>
#include <map>

#include "Swiften/JID/JID.h"
#include "Swiften/Elements/Presence.h"
#include "Swiften/Elements/VCard.h"
#include "Swiften/Elements/ErrorPayload.h"

namespace Swift {
	class MUCRegistry;
	class AvatarStorage;
	class StanzaChannel;
	class IQRouter;

	class AvatarManager {
		public:
			AvatarManager(StanzaChannel*, IQRouter*, AvatarStorage*, MUCRegistry* = NULL);
			virtual ~AvatarManager();

			virtual void setMUCRegistry(MUCRegistry*);

			virtual String getAvatarHash(const JID&) const;
			virtual boost::filesystem::path getAvatarPath(const JID&) const;
			virtual void setAvatar(const JID&, const ByteArray& avatar);

		public:
			boost::signal<void (const JID&, const String&)> onAvatarChanged;

		protected:
			/** Used only for testing. Leads to a non-functional object. */
			AvatarManager();

		private:
			void handlePresenceReceived(boost::shared_ptr<Presence>);
			void handleVCardReceived(const JID& from, const String& hash, boost::shared_ptr<VCard>, const boost::optional<ErrorPayload>&);
			void setAvatarHash(const JID& from, const String& hash);
			JID getAvatarJID(const JID& o) const;

		private:
			StanzaChannel* stanzaChannel_;
			IQRouter* iqRouter_;
			AvatarStorage* avatarStorage_;
			MUCRegistry* mucRegistry_;
			std::map<JID, String> avatarHashes_;
	};
}