X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FProfileAccessor.java;h=8c7d1e4b8bd25e0502d501d2a4badd527bc42a8a;hp=52f39dda959def54c2c562ccce9ebb0fe562debb;hb=87b2b8dc5a94c2af3cee7a91c122a7240044cddf;hpb=2cee36b47bf421ab43be6ca5a6be7fdb67441418

diff --git a/src/main/java/net/pterodactylus/sone/template/ProfileAccessor.java b/src/main/java/net/pterodactylus/sone/template/ProfileAccessor.java
index 52f39dd..8c7d1e4 100644
--- a/src/main/java/net/pterodactylus/sone/template/ProfileAccessor.java
+++ b/src/main/java/net/pterodactylus/sone/template/ProfileAccessor.java
@@ -17,35 +17,84 @@
 
 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) {
+				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) || (avatarId == null)) {
+				return avatarId;
+			}
+			if ((showCustomAvatars == ShowCustomAvatars.FOLLOWED) && currentSone.hasFriend(remoteSone.getId())) {
+				return avatarId;
+			}
+			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);
 	}
 
 }