Fix ALL the logging!

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

diff --git a/src/main/java/net/pterodactylus/sone/template/SoneAccessor.java b/src/main/java/net/pterodactylus/sone/template/SoneAccessor.java
index bfb289e..f6d3d2d 100644 (file)
--- 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.
@@ -29,32 +38,107 @@ import net.pterodactylus.util.template.ReflectionAccessor;
  * <dt>niceName</dt>
  * <dd>Will show a combination of first name, middle name, and last name, if
  * available, otherwise the username of the Sone is returned.</dd>
+ * <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 TemplateContext}).</dd>
+ * <dt>current</dt>
+ * <dd>Will return {@code true} if the sone in question is the currently logged
+ * in Sone.</dd>
  * </dl>
  *
  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
 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")) {
-                       Profile profile = sone.getProfile();
-                       String firstName = profile.getFirstName();
-                       String middleName = profile.getMiddleName();
-                       String lastName = profile.getLastName();
-                       if (firstName == null) {
-                               if (lastName == null) {
-                                       return sone.getName();
-                               }
-                               return lastName;
+                       return getNiceName(sone);
+               } else if (member.equals("local")) {
+                       return core.isLocalSone(sone);
+               } else if (member.equals("friend")) {
+                       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(templateContext, object, member);
+       }
+
+       //
+       // STATIC METHODS
+       //
+
+       /**
+        * Returns the nice name of the given Sone.
+        *
+        * @param sone
+        *            The Sone to get the nice name for
+        * @return The nice name of the Sone
+        */
+       public static String getNiceName(Sone sone) {
+               Profile profile = sone.getProfile();
+               String firstName = profile.getFirstName();
+               String middleName = profile.getMiddleName();
+               String lastName = profile.getLastName();
+               if (firstName == null) {
+                       if (lastName == null) {
+                               return String.valueOf(sone.getName());
                        }
-                       return firstName + ((middleName != null) ? " " + middleName : "") + ((lastName != null) ? " " + lastName : "");
+                       return lastName;
                }
-               return super.get(dataProvider, object, member);
+               return firstName + ((middleName != null) ? " " + middleName : "") + ((lastName != null) ? " " + lastName : "");
        }
 
 }