diff options
author | Remko Tronçon <git@el-tramo.be> | 2011-03-11 20:22:35 (GMT) |
---|---|---|
committer | Remko Tronçon <git@el-tramo.be> | 2011-03-11 22:22:04 (GMT) |
commit | 59aa5d7e29ca142ae324ad97e6a5f0353d1a6b89 (patch) | |
tree | e450b95ff4c0ba7f770723402a2634773f1a0451 /Swiften/Avatars/AvatarMemoryStorage.h | |
parent | 3ff52013d810f94b6095e93f550f58133e2df239 (diff) | |
download | swift-contrib-59aa5d7e29ca142ae324ad97e6a5f0353d1a6b89.zip swift-contrib-59aa5d7e29ca142ae324ad97e6a5f0353d1a6b89.tar.bz2 |
Store JID->Avatar mappings.
Resolves: #653
Diffstat (limited to 'Swiften/Avatars/AvatarMemoryStorage.h')
-rw-r--r-- | Swiften/Avatars/AvatarMemoryStorage.h | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/Swiften/Avatars/AvatarMemoryStorage.h b/Swiften/Avatars/AvatarMemoryStorage.h index 3fa770a..095b144 100644 --- a/Swiften/Avatars/AvatarMemoryStorage.h +++ b/Swiften/Avatars/AvatarMemoryStorage.h @@ -7,10 +7,11 @@ #pragma once #include <map> - #include <string> -#include "Swiften/Base/ByteArray.h" -#include "Swiften/Avatars/AvatarStorage.h" + +#include <Swiften/Base/ByteArray.h> +#include <Swiften/Avatars/AvatarStorage.h> +#include <Swiften/JID/JID.h> namespace Swift { class AvatarMemoryStorage : public AvatarStorage { @@ -26,7 +27,17 @@ namespace Swift { return boost::filesystem::path("/avatars") / hash; } + virtual void setAvatarForJID(const JID& jid, const std::string& hash) { + jidAvatars[jid] = hash; + } + + virtual std::string getAvatarForJID(const JID& jid) const { + std::map<JID, std::string>::const_iterator i = jidAvatars.find(jid); + return i == jidAvatars.end() ? "" : i->second; + } + private: std::map<std::string, ByteArray> avatars; + std::map<JID, std::string> jidAvatars; }; } |