package net.pterodactylus.sone.template;
-import java.util.LinkedHashMap;
-import java.util.Map;
-
+import net.pterodactylus.sone.core.Core;
import net.pterodactylus.sone.data.Profile;
+import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.data.Sone.ShowCustomAvatars;
+import net.pterodactylus.sone.freenet.wot.Trust;
import net.pterodactylus.util.template.Accessor;
-import net.pterodactylus.util.template.DataProvider;
import net.pterodactylus.util.template.ReflectionAccessor;
+import net.pterodactylus.util.template.TemplateContext;
/**
- * {@link Accessor} for {@link Profile}s, especially for the profile fields.
+ * {@link Accessor} for {@link Profile} objects that overwrites the original
+ * “avatar” member to include checks for whether the custom avatar should
+ * actually be shown.
*
* @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
*/
public class ProfileAccessor extends ReflectionAccessor {
+ /** The core. */
+ private final Core core;
+
+ /**
+ * Creates a new profile accessor.
+ *
+ * @param core
+ * The Sone core
+ */
+ public ProfileAccessor(Core core) {
+ this.core = core;
+ }
+
/**
* {@inheritDoc}
*/
@Override
- public Object get(DataProvider dataProvider, Object object, String member) {
+ public Object get(TemplateContext templateContext, Object object, String member) {
Profile profile = (Profile) object;
- if ("fields".equals(member)) {
- Map<String, String> fields = new LinkedHashMap<String, String>();
- for (String field : profile.getFields()) {
- fields.put(field, profile.getField(field));
+ if ("avatar".equals(member)) {
+ Sone currentSone = (Sone) templateContext.get("currentSone");
+ if (currentSone == null) {
+ /* not logged in? don’t show custom avatars, then. */
+ return null;
+ }
+ String avatarId = profile.getAvatar();
+ if (avatarId == null) {
+ return null;
+ }
+ if (core.getImage(avatarId, false) == null) {
+ /* avatar ID but no matching image? show nothing. */
+ return null;
+ }
+ Sone remoteSone = profile.getSone();
+ if (core.isLocalSone(remoteSone)) {
+ /* always show your own avatars. */
+ return avatarId;
+ }
+ ShowCustomAvatars showCustomAvatars = currentSone.getOptions().<ShowCustomAvatars> getEnumOption("ShowCustomAvatars").get();
+ if (showCustomAvatars == ShowCustomAvatars.NEVER) {
+ return null;
+ }
+ if (showCustomAvatars == ShowCustomAvatars.ALWAYS) {
+ return avatarId;
+ }
+ if (showCustomAvatars == ShowCustomAvatars.FOLLOWED) {
+ return currentSone.hasFriend(remoteSone.getId()) ? avatarId : null;
+ }
+ Trust trust = core.getTrust(currentSone, remoteSone);
+ if (trust == null) {
+ return null;
+ }
+ if ((showCustomAvatars == ShowCustomAvatars.MANUALLY_TRUSTED) && (trust.getExplicit() != null) && (trust.getExplicit() > 0)) {
+ return avatarId;
+ }
+ if ((showCustomAvatars == ShowCustomAvatars.TRUSTED) && (((trust.getExplicit() != null) && (trust.getExplicit() > 0)) || ((trust.getImplicit() != null) && (trust.getImplicit() > 0)))) {
+ return avatarId;
}
- return fields;
+ return null;
}
- return super.get(dataProvider, object, member);
+ return super.get(templateContext, object, member);
}
}