X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FSoneAccessor.java;h=e22738aa1f05b9d48ed44e27b125742e3ea7ca30;hb=b1660609f6574ed43afd61d332be9f3764900405;hp=d0a287f34d8b249e662816b4c57ce32fa655df5a;hpb=4d0705ef1773ee51717911bf5a1bc8f6ca71d339;p=Sone.git diff --git a/src/main/java/net/pterodactylus/sone/template/SoneAccessor.java b/src/main/java/net/pterodactylus/sone/template/SoneAccessor.java index d0a287f..e22738a 100644 --- a/src/main/java/net/pterodactylus/sone/template/SoneAccessor.java +++ b/src/main/java/net/pterodactylus/sone/template/SoneAccessor.java @@ -17,6 +17,8 @@ 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; @@ -42,6 +44,19 @@ import net.pterodactylus.util.template.ReflectionAccessor; */ 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} */ @@ -61,6 +76,14 @@ public class SoneAccessor extends ReflectionAccessor { return currentSone.isSoneBlocked(sone.getId()); } else if (member.equals("modified")) { return sone.getModificationCounter() > 0; + } else if (member.equals("status")) { + return core.getSoneStatus(sone).toString(); + } 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; } return super.get(dataProvider, object, member); }