X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FSoneAccessor.java;h=f6d3d2d15bc7c398b66d57d9a8bd714ef1ff1ea3;hp=54a5a863520f35b8c7710c892561a2e992e39e09;hb=c9e306ac8e3ada846e87a0cc256a20fc148f381c;hpb=1c57e8be327fe76dbc82ba438bcaeb4023259e71

diff --git a/src/main/java/net/pterodactylus/sone/template/SoneAccessor.java b/src/main/java/net/pterodactylus/sone/template/SoneAccessor.java
index 54a5a86..f6d3d2d 100644
--- a/src/main/java/net/pterodactylus/sone/template/SoneAccessor.java
+++ b/src/main/java/net/pterodactylus/sone/template/SoneAccessor.java
@@ -17,11 +17,20 @@
 
 package net.pterodactylus.sone.template;
 
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+import net.pterodactylus.sone.core.Core;
 import net.pterodactylus.sone.data.Profile;
 import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.data.Sone.SoneStatus;
+import net.pterodactylus.sone.freenet.wot.Trust;
+import net.pterodactylus.sone.web.WebInterface;
+import net.pterodactylus.sone.web.ajax.GetTimesAjaxPage;
+import net.pterodactylus.util.logging.Logging;
 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 Sone}s that adds a couple of properties to Sones.
@@ -32,8 +41,8 @@ import net.pterodactylus.util.template.ReflectionAccessor;
  * <dt>friend</dt>
  * <dd>Will return {@code true} if the sone in question is a friend of the
  * currently logged in Sone (as determined by accessing the “currentSone”
- * variable of the given {@link DataProvider}).</dd>
- * <dt>isCurrent</dt>
+ * variable of the given {@link TemplateContext}).</dd>
+ * <dt>current</dt>
  * <dd>Will return {@code true} if the sone in question is the currently logged
  * in Sone.</dd>
  * </dl>
@@ -42,25 +51,69 @@ import net.pterodactylus.util.template.ReflectionAccessor;
  */
 public class SoneAccessor extends ReflectionAccessor {
 
+	/** The logger. */
+	private static final Logger logger = Logging.getLogger(SoneAccessor.class);
+
+	/** The core. */
+	private final Core core;
+
+	/**
+	 * Creates a new Sone accessor.
+	 *
+	 * @param core
+	 *            The Sone core
+	 */
+	public SoneAccessor(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) {
 		Sone sone = (Sone) object;
 		if (member.equals("niceName")) {
 			return getNiceName(sone);
+		} else if (member.equals("local")) {
+			return core.isLocalSone(sone);
 		} else if (member.equals("friend")) {
-			Sone currentSone = (Sone) dataProvider.getData("currentSone");
-			return currentSone.hasFriend(sone) ? true : null;
-		} else if (member.equals("isCurrent")) {
-			Sone currentSone = (Sone) dataProvider.getData("currentSone");
-			return currentSone.equals(sone);
-		} else if (member.equals("blocked")) {
-			Sone currentSone = (Sone) dataProvider.getData("currentSone");
-			return currentSone.isSoneBlocked(sone.getId());
+			Sone currentSone = (Sone) templateContext.get("currentSone");
+			return (currentSone != null) && currentSone.hasFriend(sone.getId());
+		} else if (member.equals("current")) {
+			Sone currentSone = (Sone) templateContext.get("currentSone");
+			return (currentSone != null) && currentSone.equals(sone);
+		} else if (member.equals("modified")) {
+			return core.isModifiedSone(sone);
+		} else if (member.equals("status")) {
+			return sone.getStatus();
+		} else if (member.equals("unknown")) {
+			return sone.getStatus() == SoneStatus.unknown;
+		} else if (member.equals("idle")) {
+			return sone.getStatus() == SoneStatus.idle;
+		} else if (member.equals("inserting")) {
+			return sone.getStatus() == SoneStatus.inserting;
+		} else if (member.equals("downloading")) {
+			return sone.getStatus() == SoneStatus.downloading;
+		} else if (member.equals("new")) {
+			return !sone.isKnown();
+		} else if (member.equals("locked")) {
+			return core.isLocked(sone);
+		} else if (member.equals("lastUpdatedText")) {
+			return GetTimesAjaxPage.getTime((WebInterface) templateContext.get("webInterface"), sone.getTime());
+		} else if (member.equals("trust")) {
+			Sone currentSone = (Sone) templateContext.get("currentSone");
+			if (currentSone == null) {
+				return null;
+			}
+			Trust trust = core.getTrust(currentSone, sone);
+			logger.log(Level.FINEST, String.format("Trust for %s by %s: %s", sone, currentSone, trust));
+			if (trust == null) {
+				return new Trust(null, null, null);
+			}
+			return trust;
 		}
-		return super.get(dataProvider, object, member);
+		return super.get(templateContext, object, member);
 	}
 
 	//
@@ -81,7 +134,7 @@ public class SoneAccessor extends ReflectionAccessor {
 		String lastName = profile.getLastName();
 		if (firstName == null) {
 			if (lastName == null) {
-				return sone.getName();
+				return String.valueOf(sone.getName());
 			}
 			return lastName;
 		}