summaryrefslogtreecommitdiffstats
blob: ef1d29310e6690a6290df6791754ce7554e661c0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
/*
 * 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>
#include <Swiften/Base/Log.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);
}

std::string VCardAvatarManager::getAvatarHash(const JID& jid) const {
	JID avatarJID = getAvatarJID(jid);
	std::string hash = vcardManager_->getPhotoHash(avatarJID);
	if (!hash.empty()) {
		if (!avatarStorage_->hasAvatar(hash)) {
			VCard::ref vCard = vcardManager_->getVCard(avatarJID);
			if (vCard) {
				std::string newHash = Hexify::hexify(SHA1::getHash(vCard->getPhoto()));
				if (newHash != hash) {
					// Shouldn't happen, but sometimes seem to. Might be fixed if we
					// move to a safer backend.
					SWIFT_LOG(warning) << "Inconsistent vCard photo hash cache";
					hash = newHash;
				}
				avatarStorage_->addAvatar(hash, vCard->getPhoto());
			}
			else {
				// Can happen if the cache is inconsistent.
				hash = "";
			}
		}
	}
	return hash;
}

JID VCardAvatarManager::getAvatarJID(const JID& jid) const {
	JID bareFrom = jid.toBare();
	return (mucRegistry_ && mucRegistry_->isMUC(bareFrom)) ? jid : bareFrom;
}

}