summaryrefslogtreecommitdiffstats
blob: b2efcf22c694f8f0cf0d23ead45d7599d7df3fc9 (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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
/*
 * Copyright (c) 2010-2013 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */
/*
 * Copyright (c) 2015 Tarun Gupta.
 * Licensed under the simplified BSD license.
 * See Documentation/Licenses/BSD-simplified.txt for more information.
 */

package com.isode.stroke.avatars;

import com.isode.stroke.avatars.AvatarProvider;
import com.isode.stroke.jid.JID;
import com.isode.stroke.elements.VCard;
import com.isode.stroke.crypto.CryptoProvider;
import com.isode.stroke.stringcodecs.Hexify;
import com.isode.stroke.avatars.AvatarStorage;
import com.isode.stroke.muc.MUCRegistry;
import com.isode.stroke.vcards.VCardManager;
import com.isode.stroke.signals.Slot2;
import java.util.logging.Logger;
import com.isode.stroke.signals.SignalConnection;

public class VCardAvatarManager implements AvatarProvider {

	private VCardManager vcardManager_;
	private AvatarStorage avatarStorage_;
	private CryptoProvider crypto_;
	private MUCRegistry mucRegistry_;
	private SignalConnection onVCardChangedConnection_;
	private Logger logger_ = Logger.getLogger(this.getClass().getName());

	public VCardAvatarManager(VCardManager vcardManager, AvatarStorage avatarStorage, CryptoProvider crypto) {
		this(vcardManager, avatarStorage, crypto, null);
	}

	public VCardAvatarManager(VCardManager vcardManager, AvatarStorage avatarStorage, CryptoProvider crypto, MUCRegistry mucRegistry) {
		this.vcardManager_ = vcardManager;
		this.avatarStorage_ = avatarStorage;
		this.crypto_ = crypto;
		this.mucRegistry_ = mucRegistry;
		onVCardChangedConnection_ = vcardManager.onVCardChanged.connect(new Slot2<JID, VCard>() {

			public void call(JID p1, VCard vcard) {
				handleVCardChanged(p1);
			}	
		});
	}

	public String getAvatarHash(JID jid) {
		JID avatarJID = getAvatarJID(jid);
		String hash = vcardManager_.getPhotoHash(avatarJID);
		if(hash.length() != 0) {
			if (!avatarStorage_.hasAvatar(hash)) {
				VCard vCard = vcardManager_.getVCard(avatarJID);
				if (vCard != null) {
					String newHash = Hexify.hexify(crypto_.getSHA1Hash(vCard.getPhoto()));
					if (!newHash.equals(hash)) {
						// Shouldn't happen, but sometimes seem to. Might be fixed if we
						// move to a safer backend.
						logger_.warning("Inconsistent vCard photo hash cache");
						hash = newHash;
					}
					avatarStorage_.addAvatar(hash, vCard.getPhoto());
				}
				else {
					// Can happen if the cache is inconsistent.
					hash = "";
				}
			}
		}
		return hash;
	}

	private void handleVCardChanged(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.emit(from);
	}

	private JID getAvatarJID(JID jid) {
		JID bareFrom = jid.toBare();
		return (mucRegistry_ != null && mucRegistry_.isMUC(bareFrom)) ? jid : bareFrom;
	}
}