X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fajax%2FGetStatusAjaxPage.java;h=426c08faf6bc04090cd096eaf07d749b212dffc3;hb=a23c4f218c3adf236d89d5927cae37d6e6e4feda;hp=8b7f5132613243af67de5328df87413e36051a12;hpb=c8bdd972b5ac8aa160a85afdcf0db2cbb120d33e;p=Sone.git diff --git a/src/main/java/net/pterodactylus/sone/web/ajax/GetStatusAjaxPage.java b/src/main/java/net/pterodactylus/sone/web/ajax/GetStatusAjaxPage.java index 8b7f513..426c08f 100644 --- a/src/main/java/net/pterodactylus/sone/web/ajax/GetStatusAjaxPage.java +++ b/src/main/java/net/pterodactylus/sone/web/ajax/GetStatusAjaxPage.java @@ -17,26 +17,30 @@ package net.pterodactylus.sone.web.ajax; +import java.io.IOException; import java.io.StringWriter; import java.text.DateFormat; import java.text.SimpleDateFormat; import java.util.ArrayList; import java.util.Collections; import java.util.Date; +import java.util.HashSet; import java.util.List; import java.util.Set; import net.pterodactylus.sone.data.Post; import net.pterodactylus.sone.data.Reply; import net.pterodactylus.sone.data.Sone; +import net.pterodactylus.sone.notify.ListNotificationFilters; import net.pterodactylus.sone.template.SoneAccessor; import net.pterodactylus.sone.web.WebInterface; -import net.pterodactylus.util.io.Closer; +import net.pterodactylus.util.filter.Filter; +import net.pterodactylus.util.filter.Filters; import net.pterodactylus.util.json.JsonArray; import net.pterodactylus.util.json.JsonObject; import net.pterodactylus.util.notify.Notification; -import net.pterodactylus.util.template.Template; -import net.pterodactylus.util.template.TemplateException; +import net.pterodactylus.util.notify.TemplateNotification; +import net.pterodactylus.util.template.TemplateContext; /** * The “get status” AJAX handler returns all information that is necessary to @@ -49,26 +53,14 @@ public class GetStatusAjaxPage extends JsonPage { /** Date formatter. */ private static final DateFormat dateFormat = new SimpleDateFormat("MMM d, yyyy, HH:mm:ss"); - /** The template to render posts. */ - private final Template postTemplate; - - /** The template to render replies. */ - private final Template replyTemplate; - /** * Creates a new “get status” AJAX handler. * * @param webInterface * The Sone web interface - * @param postTemplate - * The template to render for posts - * @param replyTemplate - * The template to render for replies */ - public GetStatusAjaxPage(WebInterface webInterface, Template postTemplate, Template replyTemplate) { - super("ajax/getStatus.ajax", webInterface); - this.postTemplate = postTemplate; - this.replyTemplate = replyTemplate; + public GetStatusAjaxPage(WebInterface webInterface) { + super("getStatus.ajax", webInterface); } /** @@ -76,41 +68,71 @@ public class GetStatusAjaxPage extends JsonPage { */ @Override protected JsonObject createJsonObject(Request request) { + final Sone currentSone = getCurrentSone(request.getToadletContext(), false); /* load Sones. */ boolean loadAllSones = Boolean.parseBoolean(request.getHttpRequest().getParam("loadAllSones", "true")); - Set sones = loadAllSones ? webInterface.getCore().getSones() : Collections.singleton(getCurrentSone(request.getToadletContext())); + Set sones = new HashSet(Collections.singleton(getCurrentSone(request.getToadletContext(), false))); + if (loadAllSones) { + sones.addAll(webInterface.getCore().getSones()); + } JsonArray jsonSones = new JsonArray(); for (Sone sone : sones) { + if (sone == null) { + continue; + } JsonObject jsonSone = createJsonSone(sone); jsonSones.add(jsonSone); } /* load notifications. */ - List notifications = new ArrayList(webInterface.getNotifications().getChangedNotifications()); - Set removedNotifications = webInterface.getNotifications().getRemovedNotifications(); + List notifications = ListNotificationFilters.filterNotifications(new ArrayList(webInterface.getNotifications().getNotifications()), currentSone); Collections.sort(notifications, Notification.LAST_UPDATED_TIME_SORTER); JsonArray jsonNotifications = new JsonArray(); for (Notification notification : notifications) { jsonNotifications.add(createJsonNotification(notification)); } - JsonArray jsonRemovedNotifications = new JsonArray(); - for (Notification notification : removedNotifications) { - jsonRemovedNotifications.add(createJsonNotification(notification)); - } /* load new posts. */ - postTemplate.set("currentSone", getCurrentSone(request.getToadletContext())); Set newPosts = webInterface.getNewPosts(); + if (currentSone != null) { + newPosts = Filters.filteredSet(newPosts, new Filter() { + + @Override + public boolean filterObject(Post post) { + return currentSone.hasFriend(post.getSone().getId()) || currentSone.equals(post.getSone()) || currentSone.equals(post.getRecipient()); + } + + }); + } JsonArray jsonPosts = new JsonArray(); for (Post post : newPosts) { - jsonPosts.add(createJsonPost(post)); + JsonObject jsonPost = new JsonObject(); + jsonPost.put("id", post.getId()); + jsonPost.put("sone", post.getSone().getId()); + jsonPost.put("recipient", (post.getRecipient() != null) ? post.getRecipient().getId() : null); + jsonPost.put("time", post.getTime()); + jsonPosts.add(jsonPost); } /* load new replies. */ - replyTemplate.set("currentSone", getCurrentSone(request.getToadletContext())); Set newReplies = webInterface.getNewReplies(); + if (currentSone != null) { + newReplies = Filters.filteredSet(newReplies, new Filter() { + + @Override + public boolean filterObject(Reply reply) { + return currentSone.hasFriend(reply.getPost().getSone().getId()) || currentSone.equals(reply.getPost().getSone()) || currentSone.equals(reply.getPost().getRecipient()); + } + + }); + } JsonArray jsonReplies = new JsonArray(); for (Reply reply : newReplies) { - jsonReplies.add(createJsonReply(reply)); + JsonObject jsonReply = new JsonObject(); + jsonReply.put("id", reply.getId()); + jsonReply.put("sone", reply.getSone().getId()); + jsonReply.put("post", reply.getPost().getId()); + jsonReply.put("postSone", reply.getPost().getSone().getId()); + jsonReplies.add(jsonReply); } - return createSuccessJsonObject().put("sones", jsonSones).put("notifications", jsonNotifications).put("removedNotifications", jsonRemovedNotifications).put("newPosts", jsonPosts).put("newReplies", jsonReplies); + return createSuccessJsonObject().put("sones", jsonSones).put("notifications", jsonNotifications).put("newPosts", jsonPosts).put("newReplies", jsonReplies); } /** @@ -121,6 +143,14 @@ public class GetStatusAjaxPage extends JsonPage { return false; } + /** + * {@inheritDoc} + */ + @Override + protected boolean requiresLogin() { + return false; + } + // // PRIVATE METHODS // @@ -140,6 +170,7 @@ public class GetStatusAjaxPage extends JsonPage { jsonSone.put("status", webInterface.getCore().getSoneStatus(sone).name()); jsonSone.put("modified", webInterface.getCore().isModifiedSone(sone)); jsonSone.put("locked", webInterface.getCore().isLocked(sone)); + jsonSone.put("lastUpdatedUnknown", sone.getTime() == 0); synchronized (dateFormat) { jsonSone.put("lastUpdated", dateFormat.format(new Date(sone.getTime()))); } @@ -148,66 +179,28 @@ public class GetStatusAjaxPage extends JsonPage { } /** - * Creates a JSON object from the given post. The JSON object will only - * contain the ID of the post, its time, and its rendered HTML code. - * - * @param post - * The post to create a JSON object from - * @return The JSON representation of the post - */ - private JsonObject createJsonPost(Post post) { - JsonObject jsonPost = new JsonObject(); - jsonPost.put("id", post.getId()); - jsonPost.put("time", post.getTime()); - StringWriter stringWriter = new StringWriter(); - postTemplate.set("post", post); - try { - postTemplate.render(stringWriter); - } catch (TemplateException te1) { - /* TODO - shouldn’t happen. */ - } finally { - Closer.close(stringWriter); - } - return jsonPost.put("html", stringWriter.toString()); - } - - /** - * Creates a JSON object from the given reply. The JSON object will only - * contain the ID of the reply, the ID of its post, its time, and its - * rendered HTML code. - * - * @param reply - * The reply to create a JSON object from - * @return The JSON representation of the reply - */ - private JsonObject createJsonReply(Reply reply) { - JsonObject jsonPost = new JsonObject(); - jsonPost.put("postId", reply.getPost().getId()); - jsonPost.put("id", reply.getId()); - jsonPost.put("time", reply.getTime()); - StringWriter stringWriter = new StringWriter(); - replyTemplate.set("reply", reply); - try { - replyTemplate.render(stringWriter); - } catch (TemplateException te1) { - /* TODO - shouldn’t happen. */ - } finally { - Closer.close(stringWriter); - } - return jsonPost.put("html", stringWriter.toString()); - } - - /** * Creates a JSON object from the given notification. * * @param notification * The notification to create a JSON object * @return The JSON object */ - private static JsonObject createJsonNotification(Notification notification) { + private JsonObject createJsonNotification(Notification notification) { JsonObject jsonNotification = new JsonObject(); jsonNotification.put("id", notification.getId()); - jsonNotification.put("text", notification.toString()); + StringWriter notificationWriter = new StringWriter(); + try { + if (notification instanceof TemplateNotification) { + TemplateContext templateContext = webInterface.getTemplateContextFactory().createTemplateContext().mergeContext(((TemplateNotification) notification).getTemplateContext()); + templateContext.set("notification", notification); + ((TemplateNotification) notification).render(templateContext, notificationWriter); + } else { + notification.render(notificationWriter); + } + } catch (IOException ioe1) { + /* StringWriter never throws, ignore. */ + } + jsonNotification.put("text", notificationWriter.toString()); jsonNotification.put("createdTime", notification.getCreatedTime()); jsonNotification.put("lastUpdatedTime", notification.getLastUpdatedTime()); jsonNotification.put("dismissable", notification.isDismissable());