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
89
90
91
92
93
94
95
96
97
98
|
/*
* Copyright (c) 2010-2015 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.base.ByteArray;
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 extends AvatarProvider {
private VCardManager vcardManager_;
private AvatarStorage avatarStorage_;
private CryptoProvider crypto_;
private MUCRegistry mucRegistry_;
private final SignalConnection onVCardChangedConnection_;
private final 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>() {
@Override
public void call(JID p1, VCard vcard) {
handleVCardChanged(p1);
}
});
}
@Override
public void delete() {
onVCardChangedConnection_.disconnect();
}
@Override
public String getAvatarHash(JID jid) {
JID avatarJID = getAvatarJID(jid);
String hash = vcardManager_.getPhotoHash(avatarJID);
if(hash.length() != 0) {
if (!avatarStorage_.hasAvatar(hash)) {
final VCard vCard = vcardManager_.getVCard(avatarJID);
final ByteArray photo = vCard != null ? vCard.getPhoto() : null;
if (photo != null) {
String newHash = Hexify.hexify(crypto_.getSHA1Hash(photo));
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, photo);
}
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;
}
}
|