X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FAlbumAccessor.java;h=740da47165b6da3795a47ecd0c5348c9b5c104d3;hb=61c23dc5c1244abc3efff879ee2d3a99c1acebba;hp=13b4d65c766b6a6828c9ce36636f9e8446877374;hpb=800a866c8d935a386b4fb7182f4647950bb86791;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/template/AlbumAccessor.java b/src/main/java/net/pterodactylus/sone/template/AlbumAccessor.java
index 13b4d65..740da47 100644
--- a/src/main/java/net/pterodactylus/sone/template/AlbumAccessor.java
+++ b/src/main/java/net/pterodactylus/sone/template/AlbumAccessor.java
@@ -34,21 +34,20 @@ 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;
 		if ("backlinks".equals(member)) {
 			List<Link> backlinks = new ArrayList<Link>();
 			Album currentAlbum = album;
-			while (!currentAlbum.equals(album.getSone().getRootAlbum())) {
+			while (!currentAlbum.isRoot()) {
 				backlinks.add(0, new Link("imageBrowser.html?album=" + currentAlbum.getId(), currentAlbum.getTitle()));
 				currentAlbum = currentAlbum.getParent();
 			}
 			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;