1 files changed, 51 insertions, 0 deletions
diff --git a/Swiften/Avatars/AvatarManager.h b/Swiften/Avatars/AvatarManager.h
new file mode 100644
index 0000000..fd308d9
--- /dev/null
+++ b/Swiften/Avatars/AvatarManager.h
@@ -0,0 +1,51 @@
+#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_;
+	};
+}