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 }