X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FReplyAccessor.java;h=f439abe8ac984994f350e914233ee34cce7c13de;hb=13e00a0611cd80e43f813171d8b6d84870afbcbc;hp=d5fd7e4d2500e6a54ec0aaa4b4c6277404ddec82;hpb=63e85ddb7cebbb91f7be41640edb6104bd74650a;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/template/ReplyAccessor.java b/src/main/java/net/pterodactylus/sone/template/ReplyAccessor.java
index d5fd7e4..f439abe 100644
--- a/src/main/java/net/pterodactylus/sone/template/ReplyAccessor.java
+++ b/src/main/java/net/pterodactylus/sone/template/ReplyAccessor.java
@@ -20,18 +20,27 @@ package net.pterodactylus.sone.template;
 import net.pterodactylus.sone.core.Core;
 import net.pterodactylus.sone.data.Reply;
 import net.pterodactylus.sone.data.Sone;
-import net.pterodactylus.util.template.DataProvider;
+import net.pterodactylus.util.template.Accessor;
 import net.pterodactylus.util.template.ReflectionAccessor;
+import net.pterodactylus.util.template.TemplateContext;
 
 /**
- * TODO
+ * {@link Accessor} implementation that adds a couple of properties to
+ * {@link Reply}s.
  *
  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
 public class ReplyAccessor extends ReflectionAccessor {
 
+	/** The core. */
 	private final Core core;
 
+	/**
+	 * Creates a new reply accessor.
+	 *
+	 * @param core
+	 *            The core
+	 */
 	public ReplyAccessor(Core core) {
 		this.core = core;
 	}
@@ -40,15 +49,17 @@ public class ReplyAccessor extends ReflectionAccessor {
 	 * {@inheritDoc}
 	 */
 	@Override
-	public Object get(DataProvider dataProvider, Object object, String member) {
+	public Object get(TemplateContext templateContext, Object object, String member) {
 		Reply reply = (Reply) object;
 		if ("likes".equals(member)) {
 			return core.getLikes(reply);
 		} else if (member.equals("liked")) {
-			Sone currentSone = (Sone) dataProvider.getData("currentSone");
+			Sone currentSone = (Sone) templateContext.get("currentSone");
 			return (currentSone != null) && (currentSone.isLikedReplyId(reply.getId()));
+		} else if (member.equals("new")) {
+			return core.isNewReply(reply.getId());
 		}
-		return super.get(dataProvider, object, member);
+		return super.get(templateContext, object, member);
 	}
 
 }