package net.pterodactylus.sone.template;
+import net.pterodactylus.sone.core.Core;
+import net.pterodactylus.sone.core.Core.SoneStatus;
import net.pterodactylus.sone.data.Profile;
import net.pterodactylus.sone.data.Sone;
import net.pterodactylus.util.template.Accessor;
* <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>isFriend</dt>
+ * <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>
+ * <dt>current</dt>
* <dd>Will return {@code true} if the sone in question is the currently logged
* in Sone.</dd>
* </dl>
*/
public class SoneAccessor extends ReflectionAccessor {
+ /** The core. */
+ private final Core core;
+
+ /**
+ * Creates a new Sone accessor.
+ *
+ * @param core
+ * The Sone core
+ */
+ public SoneAccessor(Core core) {
+ this.core = core;
+ }
+
/**
* {@inheritDoc}
*/
public Object get(DataProvider dataProvider, 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 firstName + ((middleName != null) ? " " + middleName : "") + ((lastName != null) ? " " + lastName : "");
- } else if (member.equals("isFriend")) {
+ return getNiceName(sone);
+ } else if (member.equals("local")) {
+ return sone.getInsertUri() != null;
+ } else if (member.equals("friend")) {
Sone currentSone = (Sone) dataProvider.getData("currentSone");
- return currentSone.hasFriendSone(sone) ? true : null;
- } else if (member.equals("isCurrent")) {
+ return (currentSone != null) && currentSone.hasFriend(sone.getId());
+ } else if (member.equals("current")) {
Sone currentSone = (Sone) dataProvider.getData("currentSone");
- return currentSone.equals(sone);
+ return (currentSone != null) && currentSone.equals(sone);
+ } else if (member.equals("modified")) {
+ return core.isModifiedSone(sone);
+ } else if (member.equals("status")) {
+ return core.getSoneStatus(sone);
+ } else if (member.equals("unknown")) {
+ return core.getSoneStatus(sone) == SoneStatus.unknown;
+ } else if (member.equals("idle")) {
+ return core.getSoneStatus(sone) == SoneStatus.idle;
+ } else if (member.equals("inserting")) {
+ return core.getSoneStatus(sone) == SoneStatus.inserting;
+ } else if (member.equals("downloading")) {
+ return core.getSoneStatus(sone) == SoneStatus.downloading;
+ } else if (member.equals("new")) {
+ return core.isNewSone(sone);
+ } else if (member.equals("locked")) {
+ return core.isLocked(sone);
}
return super.get(dataProvider, 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 lastName;
+ }
+ return firstName + ((middleName != null) ? " " + middleName : "") + ((lastName != null) ? " " + lastName : "");
+ }
+
}