Fix pagination on album pages.
[Sone.git] / src / main / java / net / pterodactylus / sone / web / ImageBrowserPage.java
1 /*
2  * Sone - ImageBrowserPage.java - Copyright © 2011–2012 David Roden
3  *
4  * This program is free software: you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 3 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
16  */
17
18 package net.pterodactylus.sone.web;
19
20 import java.net.URI;
21 import java.util.ArrayList;
22 import java.util.Collections;
23 import java.util.List;
24
25 import net.pterodactylus.sone.data.Album;
26 import net.pterodactylus.sone.data.Image;
27 import net.pterodactylus.sone.data.Sone;
28 import net.pterodactylus.sone.web.page.FreenetRequest;
29 import net.pterodactylus.util.collection.Pagination;
30 import net.pterodactylus.util.number.Numbers;
31 import net.pterodactylus.util.template.Template;
32 import net.pterodactylus.util.template.TemplateContext;
33
34 /**
35  * The image browser page is the entry page for the image management.
36  *
37  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
38  */
39 public class ImageBrowserPage extends SoneTemplatePage {
40
41         /**
42          * Creates a new image browser page.
43          *
44          * @param template
45          *            The template to render
46          * @param webInterface
47          *            The Sone web interface
48          */
49         public ImageBrowserPage(Template template, WebInterface webInterface) {
50                 super("imageBrowser.html", template, "Page.ImageBrowser.Title", webInterface, true);
51         }
52
53         //
54         // SONETEMPLATEPAGE METHODS
55         //
56
57         /**
58          * {@inheritDoc}
59          */
60         @Override
61         protected void processTemplate(FreenetRequest request, TemplateContext templateContext) throws RedirectException {
62                 super.processTemplate(request, templateContext);
63                 String albumId = request.getHttpRequest().getParam("album", null);
64                 if (albumId != null) {
65                         Album album = webInterface.getCore().getAlbum(albumId, false);
66                         templateContext.set("albumRequested", true);
67                         templateContext.set("album", album);
68                         templateContext.set("page", request.getHttpRequest().getParam("page"));
69                         return;
70                 }
71                 String imageId = request.getHttpRequest().getParam("image", null);
72                 if (imageId != null) {
73                         Image image = webInterface.getCore().getImage(imageId, false);
74                         templateContext.set("imageRequested", true);
75                         templateContext.set("image", image);
76                         return;
77                 }
78                 String soneId = request.getHttpRequest().getParam("sone", null);
79                 if (soneId != null) {
80                         Sone sone = webInterface.getCore().getSone(soneId, false);
81                         templateContext.set("soneRequested", true);
82                         templateContext.set("sone", sone);
83                         return;
84                 }
85                 String mode = request.getHttpRequest().getParam("mode", null);
86                 if ("gallery".equals(mode)) {
87                         templateContext.set("galleryRequested", true);
88                         List<Album> albums = new ArrayList<Album>();
89                         for (Sone sone : webInterface.getCore().getSones()) {
90                                 albums.addAll(sone.getAllAlbums());
91                         }
92                         Collections.sort(albums, Album.TITLE_COMPARATOR);
93                         Pagination<Album> albumPagination = new Pagination<Album>(albums, 12).setPage(Numbers.safeParseInteger(request.getHttpRequest().getParam("page"), 0));
94                         templateContext.set("albumPagination", albumPagination);
95                         templateContext.set("albums", albumPagination.getItems());
96                         return;
97                 }
98                 Sone sone = getCurrentSone(request.getToadletContext(), false);
99                 templateContext.set("soneRequested", true);
100                 templateContext.set("sone", sone);
101         }
102
103         /**
104          * {@inheritDoc}
105          */
106         @Override
107         public boolean isLinkExcepted(URI link) {
108                 return true;
109         }
110
111 }