Enhance Parser with a ParserContext.

[Sone.git] / src / main / java / net / pterodactylus / sone / template / ReplyAccessor.java

diff --git a/src/main/java/net/pterodactylus/sone/template/ReplyAccessor.java b/src/main/java/net/pterodactylus/sone/template/ReplyAccessor.java
index d5fd7e4..db6c528 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/template/ReplyAccessor.java
+++ b/src/main/java/net/pterodactylus/sone/template/ReplyAccessor.java
@@ -17,23 +17,44 @@
 
 package net.pterodactylus.sone.template;
 
+import java.io.IOException;
+import java.io.StringReader;
+
 import net.pterodactylus.sone.core.Core;
 import net.pterodactylus.sone.data.Reply;
 import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.text.FreenetLinkParser;
+import net.pterodactylus.sone.text.FreenetLinkParserContext;
+import net.pterodactylus.util.template.Accessor;
 import net.pterodactylus.util.template.DataProvider;
 import net.pterodactylus.util.template.ReflectionAccessor;
+import net.pterodactylus.util.template.TemplateFactory;
 
 /**
- * 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 {
 
+       /** Parser for Freenet links. */
+       private final FreenetLinkParser linkParser;
+
+       /** The core. */
        private final Core core;
 
-       public ReplyAccessor(Core core) {
+       /**
+        * Creates a new reply accessor.
+        *
+        * @param core
+        *            The core
+        * @param templateFactory
+        *            The template factory for the text parser
+        */
+       public ReplyAccessor(Core core, TemplateFactory templateFactory) {
                this.core = core;
+               linkParser = new FreenetLinkParser(templateFactory);
        }
 
        /**
@@ -47,6 +68,15 @@ public class ReplyAccessor extends ReflectionAccessor {
                } else if (member.equals("liked")) {
                        Sone currentSone = (Sone) dataProvider.getData("currentSone");
                        return (currentSone != null) && (currentSone.isLikedReplyId(reply.getId()));
+               } else if (member.equals("new")) {
+                       return core.isNewReply(reply.getId(), false);
+               } else if (member.equals("text")) {
+                       String text = reply.getText();
+                       try {
+                               return linkParser.parse(new FreenetLinkParserContext(reply.getSone()), new StringReader(text));
+                       } catch (IOException ioe1) {
+                               /* ignore. */
+                       }
                }
                return super.get(dataProvider, object, member);
        }