+#include "Swiften/Avatars/VCardAvatarManager.h"

+

+ vcardAvatarManager = new VCardAvatarManager(vcardManager, avatarStorage, mucRegistry);

+ combinedAvatarProvider.addProvider(vcardAvatarManager);

+

+ combinedAvatarProvider.removeProvider(vcardAvatarManager);

+ delete vcardAvatarManager;

+ std::cout << "getAvatar " << jid << " " << avatarStorage->getAvatarPath(hash) << std::endl;

+ class VCardAvatarManager;

+ VCardUpdateAvatarManager* vcardUpdateAvatarManager;

+ VCardAvatarManager* vcardAvatarManager;

+ "VCardAvatarManager.cpp",

+/*

+ * Copyright (c) 2010 Remko Tronçon

+ * Licensed under the GNU General Public License v3.

+ * See Documentation/Licenses/GPLv3.txt for more information.

+ */

+

+#include <cppunit/extensions/HelperMacros.h>

+#include <cppunit/extensions/TestFactoryRegistry.h>

+#include <boost/bind.hpp>

+

+#include "Swiften/Elements/VCard.h"

+#include "Swiften/Avatars/VCardAvatarManager.h"

+#include "Swiften/Avatars/AvatarMemoryStorage.h"

+#include "Swiften/VCards/VCardMemoryStorage.h"

+#include "Swiften/VCards/VCardManager.h"

+#include "Swiften/MUC/MUCRegistry.h"

+#include "Swiften/Queries/IQRouter.h"

+#include "Swiften/Client/DummyStanzaChannel.h"

+#include "Swiften/StringCodecs/SHA1.h"

+#include "Swiften/StringCodecs/Hexify.h"

+

+using namespace Swift;

+

+class VCardAvatarManagerTest : public CppUnit::TestFixture {

+ CPPUNIT_TEST_SUITE(VCardAvatarManagerTest);

+ CPPUNIT_TEST(testGetAvatarHashKnownAvatar);

+ CPPUNIT_TEST(testGetAvatarHashEmptyAvatar);

+ CPPUNIT_TEST(testGetAvatarHashUnknownAvatarKnownVCardStoresAvatar);

+ CPPUNIT_TEST(testGetAvatarHashUnknownAvatarUnknownVCard);

+ CPPUNIT_TEST(testVCardUpdateTriggersUpdate);

+ CPPUNIT_TEST_SUITE_END();

+

+ public:

+ void setUp() {

+ ownJID = JID("foo@fum.com/bum");

+ stanzaChannel = new DummyStanzaChannel();

+ stanzaChannel->setAvailable(true);

+ iqRouter = new IQRouter(stanzaChannel);

+ mucRegistry = new DummyMUCRegistry();

+ avatarStorage = new AvatarMemoryStorage();

+ vcardStorage = new VCardMemoryStorage();

+ vcardManager = new VCardManager(ownJID, iqRouter, vcardStorage);

+ avatar1 = ByteArray("abcdefg");

+ avatar1Hash = Hexify::hexify(SHA1::getHash(avatar1));

+ user1 = JID("user1@bar.com/bla");

+ user2 = JID("user2@foo.com/baz");

+ }

+

+ void tearDown() {

+ delete vcardManager;

+ delete vcardStorage;

+ delete avatarStorage;

+ delete mucRegistry;

+ delete iqRouter;

+ delete stanzaChannel;

+ }

+

+ void testGetAvatarHashKnownAvatar() {

+ std::auto_ptr<VCardAvatarManager> testling = createManager();

+ storeVCardWithPhoto(user1.toBare(), avatar1);

+ avatarStorage->addAvatar(avatar1Hash, avatar1);

+

+ String result = testling->getAvatarHash(user1);

+

+ CPPUNIT_ASSERT_EQUAL(avatar1Hash, result);

+ }

+

+ void testGetAvatarHashEmptyAvatar() {

+ std::auto_ptr<VCardAvatarManager> testling = createManager();

+ storeEmptyVCard(user1.toBare());

+

+ String result = testling->getAvatarHash(user1);

+

+ CPPUNIT_ASSERT_EQUAL(String(), result);

+ }

+

+ void testGetAvatarHashUnknownAvatarKnownVCardStoresAvatar() {

+ std::auto_ptr<VCardAvatarManager> testling = createManager();

+ storeVCardWithPhoto(user1.toBare(), avatar1);

+

+ String result = testling->getAvatarHash(user1);

+

+ CPPUNIT_ASSERT_EQUAL(avatar1Hash, result);

+ CPPUNIT_ASSERT(avatarStorage->hasAvatar(avatar1Hash));

+ CPPUNIT_ASSERT_EQUAL(avatar1, avatarStorage->getAvatar(avatar1Hash));

+ }

+

+ void testGetAvatarHashUnknownAvatarUnknownVCard() {

+ std::auto_ptr<VCardAvatarManager> testling = createManager();

+

+ String result = testling->getAvatarHash(user1);

+

+ CPPUNIT_ASSERT_EQUAL(String(), result);

+ }

+

+ void testVCardUpdateTriggersUpdate() {

+ std::auto_ptr<VCardAvatarManager> testling = createManager();

+ vcardManager->requestVCard(user1);

+ sendVCardResult();

+

+ CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(changes.size()));

+ }

+

+ private:

+ std::auto_ptr<VCardAvatarManager> createManager() {

+ std::auto_ptr<VCardAvatarManager> result(new VCardAvatarManager(vcardManager, avatarStorage, mucRegistry));

+ result->onAvatarChanged.connect(boost::bind(&VCardAvatarManagerTest::handleAvatarChanged, this, _1));

+ return result;

+ }

+

+ void storeVCardWithPhoto(const JID& jid, const ByteArray& avatar) {

+ VCard::ref vcard(new VCard());

+ vcard->setPhoto(avatar);

+ vcardStorage->setVCard(jid, vcard);

+ }

+

+ void storeEmptyVCard(const JID& jid) {

+ VCard::ref vcard(new VCard());

+ vcardStorage->setVCard(jid, vcard);

+ }

+

+ void handleAvatarChanged(const JID& jid) {

+ changes.push_back(jid);

+ }

+

+ void sendVCardResult() {

+ VCard::ref vcard(new VCard());

+ vcard->setFullName("Foo Bar");

+ stanzaChannel->onIQReceived(IQ::createResult(JID("baz@fum.com/dum"), stanzaChannel->sentStanzas[0]->getID(), vcard));

+ }

+

+ private:

+ struct DummyMUCRegistry : public MUCRegistry {

+ bool isMUC(const JID& jid) const { return std::find(mucs_.begin(), mucs_.end(), jid) != mucs_.end(); }

+ std::vector<JID> mucs_;

+ };

+

+ JID ownJID;

+ DummyStanzaChannel* stanzaChannel;

+ IQRouter* iqRouter;

+ DummyMUCRegistry* mucRegistry;

+ AvatarMemoryStorage* avatarStorage;

+ VCardManager* vcardManager;

+ VCardMemoryStorage* vcardStorage;

+ ByteArray avatar1;

+ String avatar1Hash;

+ std::vector<JID> changes;

+ JID user1;

+ JID user2;

+};

+

+CPPUNIT_TEST_SUITE_REGISTRATION(VCardAvatarManagerTest);

+/*

+ * Copyright (c) 2010 Remko Tronçon

+ * Licensed under the GNU General Public License v3.

+ * See Documentation/Licenses/GPLv3.txt for more information.

+ */

+

+#include "Swiften/Avatars/VCardAvatarManager.h"

+

+#include <boost/bind.hpp>

+

+#include "Swiften/Elements/VCard.h"

+#include "Swiften/StringCodecs/SHA1.h"

+#include "Swiften/StringCodecs/Hexify.h"

+#include "Swiften/Avatars/AvatarStorage.h"

+#include "Swiften/MUC/MUCRegistry.h"

+#include "Swiften/VCards/VCardManager.h"

+

+namespace Swift {

+

+VCardAvatarManager::VCardAvatarManager(VCardManager* vcardManager, AvatarStorage* avatarStorage, MUCRegistry* mucRegistry) : vcardManager_(vcardManager), avatarStorage_(avatarStorage), mucRegistry_(mucRegistry) {

+ vcardManager_->onVCardChanged.connect(boost::bind(&VCardAvatarManager::handleVCardChanged, this, _1));

+}

+

+void VCardAvatarManager::handleVCardChanged(const JID& from) {

+ // We don't check whether the avatar actually changed. Direct use of this

+ // manager could cause unnecessary updates, but in practice, this will be

+ // caught by the wrapping CombinedAvatarManager anyway.

+ onAvatarChanged(from);

+}

+

+String VCardAvatarManager::getAvatarHash(const JID& jid) const {

+ VCard::ref vCard = vcardManager_->getVCard(getAvatarJID(jid));

+ if (vCard && !vCard->getPhoto().isEmpty()) {

+ String hash = Hexify::hexify(SHA1::getHash(vCard->getPhoto()));

+ if (!avatarStorage_->hasAvatar(hash)) {

+ avatarStorage_->addAvatar(hash, vCard->getPhoto());

+ }

+ return hash;

+ }

+ else {

+ return "";

+ }

+}

+

+JID VCardAvatarManager::getAvatarJID(const JID& jid) const {

+ JID bareFrom = jid.toBare();

+ return (mucRegistry_ && mucRegistry_->isMUC(bareFrom)) ? jid : bareFrom;

+}

+

+}

+/*

+ * Copyright (c) 2010 Remko Tronçon

+ * Licensed under the GNU General Public License v3.

+ * See Documentation/Licenses/GPLv3.txt for more information.

+ */

+

+#pragma once

+

+#include <map>

+

+#include "Swiften/Avatars/AvatarProvider.h"

+#include "Swiften/JID/JID.h"

+#include "Swiften/Elements/VCard.h"

+

+namespace Swift {

+ class MUCRegistry;

+ class AvatarStorage;

+ class VCardManager;

+

+ class VCardAvatarManager : public AvatarProvider {

+ public:

+ VCardAvatarManager(VCardManager*, AvatarStorage*, MUCRegistry* = NULL);

+

+ String getAvatarHash(const JID&) const;

+

+ private:

+ void handleVCardChanged(const JID& from);

+ JID getAvatarJID(const JID& o) const;

+

+ private:

+ VCardManager* vcardManager_;

+ AvatarStorage* avatarStorage_;

+ MUCRegistry* mucRegistry_;

+ };

+}

+ if (!avatarStorage_->hasAvatar(hash)) {

+ avatarStorage_->addAvatar(hash, vCard->getPhoto());

+ }