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

diff --git a/src/main/java/net/pterodactylus/sone/web/KnownSonesPage.java b/src/main/java/net/pterodactylus/sone/web/KnownSonesPage.java
index 0e42eb5..82f8ed6 100644
--- a/src/main/java/net/pterodactylus/sone/web/KnownSonesPage.java
+++ b/src/main/java/net/pterodactylus/sone/web/KnownSonesPage.java
@@ -1,5 +1,5 @@
 /*
- * Sone - KnownSonesPage.java - Copyright © 2010 David Roden
+ * Sone - KnownSonesPage.java - Copyright © 2010–2012 David Roden
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -22,6 +22,7 @@ import java.util.Collections;
 import java.util.List;
 
 import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.web.page.FreenetRequest;
 import net.pterodactylus.util.collection.Pagination;
 import net.pterodactylus.util.collection.ReverseComparator;
 import net.pterodactylus.util.filter.Filter;
@@ -57,14 +58,17 @@ public class KnownSonesPage extends SoneTemplatePage {
 	 * {@inheritDoc}
 	 */
 	@Override
-	protected void processTemplate(Request request, TemplateContext templateContext) throws RedirectException {
+	protected void processTemplate(FreenetRequest 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");
+		String filter = request.getHttpRequest().getParam("filter");
+		templateContext.set("sort", (sortField != null) ? sortField : "name");
+		templateContext.set("order", (sortOrder != null) ? sortOrder : "asc");
+		templateContext.set("filter", filter);
 		final Sone currentSone = getCurrentSone(request.getToadletContext(), false);
 		List<Sone> knownSones = Filters.filteredList(new ArrayList<Sone>(webInterface.getCore().getSones()), Sone.EMPTY_SONE_FILTER);
-		if ((currentSone != null) && "show-only".equals(followedSones)) {
+		if ((currentSone != null) && "followed".equals(filter)) {
 			knownSones = Filters.filteredList(knownSones, new Filter<Sone>() {
 
 				@Override
@@ -72,8 +76,7 @@ public class KnownSonesPage extends SoneTemplatePage {
 					return currentSone.hasFriend(sone.getId());
 				}
 			});
-			templateContext.set("followedSones", "show-only");
-		} else if ((currentSone != null) && "hide".equals(followedSones)) {
+		} else if ((currentSone != null) && "not-followed".equals(filter)) {
 			knownSones = Filters.filteredList(knownSones, new Filter<Sone>() {
 
 				@Override
@@ -81,27 +84,55 @@ public class KnownSonesPage extends SoneTemplatePage {
 					return !currentSone.hasFriend(sone.getId());
 				}
 			});
-			templateContext.set("followedSones", "hide");
+		} else if ("new".equals(filter)) {
+			knownSones = Filters.filteredList(knownSones, new Filter<Sone>() {
+				/**
+				 * {@inheritDoc}
+				 */
+				@Override
+				public boolean filterObject(Sone sone) {
+					return !sone.isKnown();
+				}
+			});
+		} else if ("not-new".equals(filter)) {
+			knownSones = Filters.filteredList(knownSones, new Filter<Sone>() {
+				/**
+				 * {@inheritDoc}
+				 */
+				@Override
+				public boolean filterObject(Sone sone) {
+					return sone.isKnown();
+				}
+			});
 		}
-		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 ("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");
+		} else if ("posts".equals(sortField)) {
+			if ("asc".equals(sortOrder)) {
+				Collections.sort(knownSones, new ReverseComparator<Sone>(Sone.POST_COUNT_COMPARATOR));
+			} 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));
+			} else {
+				Collections.sort(knownSones, Sone.NICE_NAME_COMPARATOR);
+			}
 		}
 		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());
 	}
+
 }