X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FKnownSonesPage.java;h=0e42eb50918ffaa89660fba387f19ea4073a9f9c;hb=9a254c2af6f176c8a95ef6accfdb288ad159d125;hp=c0e6364769db9abddc5c0ca67cac73a3f21ebfda;hpb=ee05a67b3f1e796b6c4bdcd709ef0983103c455f;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/web/KnownSonesPage.java b/src/main/java/net/pterodactylus/sone/web/KnownSonesPage.java
index c0e6364..0e42eb5 100644
--- a/src/main/java/net/pterodactylus/sone/web/KnownSonesPage.java
+++ b/src/main/java/net/pterodactylus/sone/web/KnownSonesPage.java
@@ -23,6 +23,8 @@ import java.util.List;
 
 import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.util.collection.Pagination;
+import net.pterodactylus.util.collection.ReverseComparator;
+import net.pterodactylus.util.filter.Filter;
 import net.pterodactylus.util.filter.Filters;
 import net.pterodactylus.util.number.Numbers;
 import net.pterodactylus.util.template.Template;
@@ -57,24 +59,49 @@ public class KnownSonesPage extends SoneTemplatePage {
 	@Override
 	protected void processTemplate(Request request, TemplateContext templateContext) throws RedirectException {
 		super.processTemplate(request, templateContext);
+		String sortField = request.getHttpRequest().getParam("sort");
+		String sortOrder = request.getHttpRequest().getParam("order");
+		String followedSones = request.getHttpRequest().getParam("followedSones");
+		final Sone currentSone = getCurrentSone(request.getToadletContext(), false);
 		List<Sone> knownSones = Filters.filteredList(new ArrayList<Sone>(webInterface.getCore().getSones()), Sone.EMPTY_SONE_FILTER);
-		Collections.sort(knownSones, Sone.NICE_NAME_COMPARATOR);
+		if ((currentSone != null) && "show-only".equals(followedSones)) {
+			knownSones = Filters.filteredList(knownSones, new Filter<Sone>() {
+
+				@Override
+				public boolean filterObject(Sone sone) {
+					return currentSone.hasFriend(sone.getId());
+				}
+			});
+			templateContext.set("followedSones", "show-only");
+		} else if ((currentSone != null) && "hide".equals(followedSones)) {
+			knownSones = Filters.filteredList(knownSones, new Filter<Sone>() {
+
+				@Override
+				public boolean filterObject(Sone sone) {
+					return !currentSone.hasFriend(sone.getId());
+				}
+			});
+			templateContext.set("followedSones", "hide");
+		}
+		if ("name".equals(sortField)) {
+			if ("desc".equals(sortOrder)) {
+				Collections.sort(knownSones, new ReverseComparator<Sone>(Sone.NICE_NAME_COMPARATOR));
+			} else {
+				Collections.sort(knownSones, Sone.NICE_NAME_COMPARATOR);
+			}
+			templateContext.set("sort", "name");
+			templateContext.set("order", "desc".equals(sortOrder) ? "desc" : "asc");
+		} else if ("activity".equals(sortField)) {
+			if ("asc".equals(sortOrder)) {
+				Collections.sort(knownSones, new ReverseComparator<Sone>(Sone.LAST_ACTIVITY_COMPARATOR));
+			} else {
+				Collections.sort(knownSones, Sone.LAST_ACTIVITY_COMPARATOR);
+			}
+			templateContext.set("sort", "activity");
+			templateContext.set("order", "asc".equals(sortOrder) ? "asc" : "desc");
+		}
 		Pagination<Sone> sonePagination = new Pagination<Sone>(knownSones, 25).setPage(Numbers.safeParseInteger(request.getHttpRequest().getParam("page"), 0));
 		templateContext.set("pagination", sonePagination);
 		templateContext.set("knownSones", sonePagination.getItems());
 	}
-
-	/**
-	 * {@inheritDoc}
-	 */
-	@Override
-	protected void postProcess(Request request, TemplateContext templateContext) {
-		super.postProcess(request, templateContext);
-		@SuppressWarnings("unchecked")
-		List<Sone> sones = (List<Sone>) templateContext.get("knownSones");
-		for (Sone sone : sones) {
-			webInterface.getCore().markSoneKnown(sone);
-		}
-	}
-
 }