Expose Link class.

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

diff --git a/src/main/java/net/pterodactylus/sone/template/AlbumAccessor.java b/src/main/java/net/pterodactylus/sone/template/AlbumAccessor.java
index 9ef168a..740da47 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/template/AlbumAccessor.java
+++ b/src/main/java/net/pterodactylus/sone/template/AlbumAccessor.java
@@ -34,9 +34,6 @@ import net.pterodactylus.util.template.TemplateContext;
  */
 public class AlbumAccessor extends ReflectionAccessor {
 
-       /**
-        * {@inheritDoc}
-        */
        @Override
        public Object get(TemplateContext templateContext, Object object, String member) {
                Album album = (Album) object;
@@ -49,6 +46,8 @@ public class AlbumAccessor extends ReflectionAccessor {
                        }
                        backlinks.add(0, new Link("imageBrowser.html?sone=" + album.getSone().getId(), SoneAccessor.getNiceName(album.getSone())));
                        return backlinks;
+               } else if ("albumImage".equals(member)) {
+                       return album.getAlbumImage().orNull();
                }
                return super.get(templateContext, object, member);
        }
@@ -58,7 +57,7 @@ public class AlbumAccessor extends ReflectionAccessor {
         *
         * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
         */
-       private static class Link {
+       static class Link {
 
                /** The target of the link. */
                private final String target;