Add plugins to template factory.
[Sone.git] / src / main / java / net / pterodactylus / sone / web / SoneTemplatePage.java
1 /*
2  * Freetalk - FreetalkTemplatePage.java - Copyright © 2010 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.util.Arrays;
21 import java.util.Collection;
22 import java.util.UUID;
23
24 import net.pterodactylus.sone.data.Sone;
25 import net.pterodactylus.sone.web.page.Page;
26 import net.pterodactylus.sone.web.page.TemplatePage;
27 import net.pterodactylus.util.template.Template;
28 import freenet.clients.http.SessionManager.Session;
29 import freenet.clients.http.ToadletContext;
30
31 /**
32  * Base page for the Freetalk web interface.
33  *
34  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
35  */
36 public class SoneTemplatePage extends TemplatePage {
37
38         /** The Sone core. */
39         protected final WebInterface webInterface;
40
41         /**
42          * Creates a new template page for Freetalk.
43          *
44          * @param path
45          *            The path of the page
46          * @param template
47          *            The template to render
48          * @param pageTitleKey
49          *            The l10n key of the page title
50          * @param webInterface
51          *            The Sone web interface
52          */
53         public SoneTemplatePage(String path, Template template, String pageTitleKey, WebInterface webInterface) {
54                 super(path, template, webInterface.l10n(), pageTitleKey, "noPermission.html");
55                 this.webInterface = webInterface;
56                 template.set("webInterface", webInterface);
57         }
58
59         //
60         // PROTECTED METHODS
61         //
62
63         /**
64          * Returns the current session, creating a new session if there is no
65          * current session.
66          *
67          * @param toadletContenxt
68          *            The toadlet context
69          * @return The current session, or {@code null} if there is no current
70          *         session
71          */
72         protected Session getCurrentSession(ToadletContext toadletContenxt) {
73                 return getCurrentSession(toadletContenxt, true);
74         }
75
76         /**
77          * Returns the current session, creating a new session if there is no
78          * current session and {@code create} is {@code true}.
79          *
80          * @param toadletContenxt
81          *            The toadlet context
82          * @param create
83          *            {@code true} to create a new session if there is no current
84          *            session, {@code false} otherwise
85          * @return The current session, or {@code null} if there is no current
86          *         session
87          */
88         protected Session getCurrentSession(ToadletContext toadletContenxt, boolean create) {
89                 try {
90                         Session session = webInterface.sessionManager().useSession(toadletContenxt);
91                         if (create && (session == null)) {
92                                 session = webInterface.sessionManager().createSession(UUID.randomUUID().toString(), toadletContenxt);
93                         }
94                         return session;
95                 } catch (freenet.clients.http.RedirectException re1) {
96                         return null;
97                 }
98         }
99
100         /**
101          * Returns the currently logged in Sone.
102          *
103          * @param toadletContext
104          *            The toadlet context
105          * @return The currently logged in Sone, or {@code null} if no Sone is
106          *         currently logged in
107          */
108         protected Sone getCurrentSone(ToadletContext toadletContext) {
109                 Session session = getCurrentSession(toadletContext);
110                 if (session == null) {
111                         return null;
112                 }
113                 String soneId = (String) session.getAttribute("Sone.CurrentSone");
114                 if (soneId == null) {
115                         return null;
116                 }
117                 for (Sone sone : webInterface.core().getSones()) {
118                         if (sone.getId().equals(soneId)) {
119                                 return sone;
120                         }
121                 }
122                 return null;
123         }
124
125         /**
126          * Sets the currently logged in Sone.
127          *
128          * @param toadletContext
129          *            The toadlet context
130          * @param sone
131          *            The Sone to set as currently logged in
132          */
133         protected void setCurrentSone(ToadletContext toadletContext, Sone sone) {
134                 Session session = getCurrentSession(toadletContext);
135                 if (sone == null) {
136                         session.removeAttribute("Sone.CurrentSone");
137                 } else {
138                         session.setAttribute("Sone.CurrentSone", sone.getId());
139                 }
140         }
141
142         //
143         // TEMPLATEPAGE METHODS
144         //
145
146         /**
147          * {@inheritDoc}
148          */
149         @Override
150         protected Collection<String> getStyleSheets() {
151                 return Arrays.asList("css/sone.css");
152         }
153
154         /**
155          * {@inheritDoc}
156          */
157         @Override
158         protected String getShortcutIcon() {
159                 return "images/icon.png";
160         }
161
162         /**
163          * Returns whether this page requires the user to log in.
164          *
165          * @return {@code true} if the user is required to be logged in to use this
166          *         page, {@code false} otherwise
167          */
168         protected boolean requiresLogin() {
169                 return false;
170         }
171
172         /**
173          * {@inheritDoc}
174          */
175         @Override
176         protected void processTemplate(Request request, Template template) throws RedirectException {
177                 super.processTemplate(request, template);
178                 template.set("currentSone", getCurrentSone(request.getToadletContext()));
179                 template.set("request", request);
180         }
181
182         /**
183          * {@inheritDoc}
184          */
185         @Override
186         protected String getRedirectTarget(Page.Request request) {
187                 if (requiresLogin() && (getCurrentSone(request.getToadletContext()) == null)) {
188                         return "login.html";
189                 }
190                 return null;
191         }
192
193         /**
194          * {@inheritDoc}
195          */
196         @Override
197         public boolean isEnabled(ToadletContext toadletContext) {
198                 if (requiresLogin()) {
199                         return getCurrentSone(toadletContext) != null;
200                 }
201                 return true;
202         }
203
204 }