Add accessor for track derivatives.

diff --git a/src/main/java/net/pterodactylus/demoscenemusic/core/TemplateServlet.java b/src/main/java/net/pterodactylus/demoscenemusic/core/TemplateServlet.java
index c1090f2..df753f1 100644 (file)
--- a/src/main/java/net/pterodactylus/demoscenemusic/core/TemplateServlet.java
+++ b/src/main/java/net/pterodactylus/demoscenemusic/core/TemplateServlet.java
@@ -41,9 +41,11 @@ import net.pterodactylus.demoscenemusic.data.Artist;
 import net.pterodactylus.demoscenemusic.data.Properties;
 import net.pterodactylus.demoscenemusic.data.Style;
 import net.pterodactylus.demoscenemusic.data.Track;
+import net.pterodactylus.demoscenemusic.data.TrackDerivative;
 import net.pterodactylus.demoscenemusic.data.User;
 import net.pterodactylus.demoscenemusic.page.ServletRequest;
 import net.pterodactylus.demoscenemusic.template.PropertiesAccessor;
+import net.pterodactylus.demoscenemusic.template.TrackDerivativeAccessor;
 import net.pterodactylus.demoscenemusic.template.UserAccessor;
 import net.pterodactylus.util.io.Closer;
 import net.pterodactylus.util.io.StreamCopier;
@@ -85,6 +87,7 @@ public class TemplateServlet extends HttpServlet {
                templateContextFactory.addAccessor(Object.class, new ReflectionAccessor());
                templateContextFactory.addAccessor(User.class, new UserAccessor());
                templateContextFactory.addAccessor(Properties.class, new PropertiesAccessor());
+               templateContextFactory.addAccessor(TrackDerivative.class, new TrackDerivativeAccessor());
 
                templateContextFactory.addFilter("html", new HtmlFilter());
                CollectionSortFilter sortFilter = new CollectionSortFilter();

diff --git a/src/main/java/net/pterodactylus/demoscenemusic/template/TrackDerivativeAccessor.java b/src/main/java/net/pterodactylus/demoscenemusic/template/TrackDerivativeAccessor.java
new file mode 100644 (file)
index 0000000..66424f5
--- /dev/null
+++ b/src/main/java/net/pterodactylus/demoscenemusic/template/TrackDerivativeAccessor.java
@@ -0,0 +1,47 @@
+/*
+ * DemosceneMusic - TrackDerivativeAccessor.java - Copyright © 2012 David Roden
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package net.pterodactylus.demoscenemusic.template;
+
+import net.pterodactylus.demoscenemusic.core.Core;
+import net.pterodactylus.demoscenemusic.data.TrackDerivative;
+import net.pterodactylus.util.template.ReflectionAccessor;
+import net.pterodactylus.util.template.TemplateContext;
+
+/**
+ * {@link ReflectionAccessor} implementation that adds “size” and “path”
+ * properties to a {@link TrackDerivative}.
+ *
+ * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+ */
+public class TrackDerivativeAccessor extends ReflectionAccessor {
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       public Object get(TemplateContext templateContext, Object object, String member) {
+               TrackDerivative trackDerivative = (TrackDerivative) object;
+               if ("path".equals(member)) {
+                       return ((Core) templateContext.get("core")).getDataDirectory().getPath(trackDerivative.getId());
+               } else if ("size".equals(member)) {
+                       return ((Core) templateContext.get("core")).getDataDirectory().getFile(trackDerivative.getId()).length();
+               }
+               return super.get(templateContext, object, member);
+       }
+
+}