X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FSoneAccessor.java;h=c93f4e76f7db2170ae08658505442fd8165d6903;hb=7fb49938b9198110c34bcc600c545bfa91acf6f2;hp=c54fc0d80d106d0ae641dc0f18731d8c9f10594c;hpb=6e9a43ccd93ae125720547c0fe421dc81a54ba90;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 c54fc0d..c93f4e7 100644
--- a/src/main/java/net/pterodactylus/sone/template/SoneAccessor.java
+++ b/src/main/java/net/pterodactylus/sone/template/SoneAccessor.java
@@ -17,6 +17,11 @@
 
 package net.pterodactylus.sone.template;
 
+import static com.google.common.collect.FluentIterable.from;
+import static java.util.Arrays.asList;
+import static net.pterodactylus.sone.data.Album.FLATTENER;
+import static net.pterodactylus.sone.data.Album.IMAGES;
+
 import java.util.logging.Level;
 import java.util.logging.Logger;
 
@@ -68,9 +73,6 @@ public class SoneAccessor extends ReflectionAccessor {
 		this.core = core;
 	}
 
-	/**
-	 * {@inheritDoc}
-	 */
 	@Override
 	public Object get(TemplateContext templateContext, Object object, String member) {
 		Sone sone = (Sone) object;
@@ -111,6 +113,10 @@ public class SoneAccessor extends ReflectionAccessor {
 				return new Trust(null, null, null);
 			}
 			return trust;
+		} else if (member.equals("allImages")) {
+			return from(asList(sone.getRootAlbum())).transformAndConcat(FLATTENER).transformAndConcat(IMAGES);
+		} else if (member.equals("albums")) {
+			return sone.getRootAlbum().getAlbums();
 		}
 		return super.get(templateContext, object, member);
 	}