Add “sort by image count” for Known Sones.

[Sone.git] / src / main / java / net / pterodactylus / sone / web / KnownSonesPage.java

diff --git a/src/main/java/net/pterodactylus/sone/web/KnownSonesPage.java b/src/main/java/net/pterodactylus/sone/web/KnownSonesPage.java
index e5acad5..0e27265 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/web/KnownSonesPage.java
+++ b/src/main/java/net/pterodactylus/sone/web/KnownSonesPage.java
@@ -97,6 +97,12 @@ public class KnownSonesPage extends SoneTemplatePage {
                        } else {
                                Collections.sort(knownSones, Sone.POST_COUNT_COMPARATOR);
                        }
+               } else if ("images".equals(sortField)) {
+                       if ("asc".equals(sortOrder)) {
+                               Collections.sort(knownSones, new ReverseComparator<Sone>(Sone.IMAGE_COUNT_COMPARATOR));
+                       } else {
+                               Collections.sort(knownSones, Sone.IMAGE_COUNT_COMPARATOR);
+                       }
                } else {
                        if ("desc".equals(sortOrder)) {
                                Collections.sort(knownSones, new ReverseComparator<Sone>(Sone.NICE_NAME_COMPARATOR));