X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FSone.kt;h=46ff3362cf540edd75a382a5a4167460b74f1d8f;hb=5473139d890c40c85712f59ba534583961c2772e;hp=c44d6e8b81f9b5009ec42e0f88cacf683504113e;hpb=9b0b5b760adf41d52603a4b65f3e5f3220da28eb;p=Sone.git

diff --git a/src/main/kotlin/net/pterodactylus/sone/data/Sone.kt b/src/main/kotlin/net/pterodactylus/sone/data/Sone.kt
index c44d6e8..46ff336 100644
--- a/src/main/kotlin/net/pterodactylus/sone/data/Sone.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/data/Sone.kt
@@ -48,3 +48,10 @@ val postCountComparator: Comparator<Sone> =
 		comparing<Sone, Int> { it.posts.size }
 				.thenComparing<Int> { it.replies.size }
 				.reversed()
+
+val imageCountComparator: Comparator<Sone> =
+		comparing<Sone, Int> { it.rootAlbum.allImages.size }.reversed()
+
+val Sone.allAlbums: List<Album>
+	get() =
+		rootAlbum.albums.flatMap(Album::allAlbums)