src/main/java/net/pterodactylus/demoscenemusic/page/BasePage.java

   1 /*
   2  * DemosceneMusic - BasePage.java - Copyright © 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.demoscenemusic.page;
  19
  20 import net.pterodactylus.demoscenemusic.core.Core;
  21 import net.pterodactylus.demoscenemusic.data.User;
  22 import net.pterodactylus.util.template.Template;
  23 import net.pterodactylus.util.template.TemplateContext;
  24 import net.pterodactylus.util.template.TemplateContextFactory;
  25 import net.pterodactylus.util.web.RedirectException;
  26 import net.pterodactylus.util.web.TemplatePage;
  27
  28 /**
  29  * TODO
  30  *
  31  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  32  */
  33 public class BasePage extends TemplatePage<ServletRequest> {
  34
  35         private final Core core;
  36
  37         /**
  38          * @param path
  39          * @param contentType
  40          * @param templateContextFactory
  41          * @param template
  42          */
  43         public BasePage(Core core, TemplateContextFactory templateContextFactory, Template template, String pageName) {
  44                 super(pageName, "text/html; charset=utf-8", templateContextFactory, template);
  45                 this.core = core;
  46         }
  47
  48         /**
  49          * @return the core
  50          */
  51         public Core getCore() {
  52                 return core;
  53         }
  54
  55         /**
  56          * {@inheritDoc}
  57          */
  58         @Override
  59         protected void processTemplate(TemplateContext templateContext, ServletRequest request) throws RedirectException {
  60                 super.processTemplate(templateContext, request);
  61                 User currentUser = (User) request.getServletRequest().getSession().getAttribute("currentUser");
  62                 templateContext.set("currentUser", currentUser);
  63                 templateContext.set("core", getCore());
  64                 templateContext.set("dataManager", getCore().getDataManager());
  65                 int requiredUserLevel = getRequiredUserLevel();
  66                 if (((currentUser == null) && (requiredUserLevel > 0)) || ((currentUser != null) && (requiredUserLevel > currentUser.getLevel()))) {
  67                         throw new RedirectException("login");
  68                 }
  69         }
  70
  71         /**
  72          * Returns the {@link User#getLevel() user level} that is at least required
  73          * to access the given page. If the returned level is {@code 0} (or
  74          * smaller), users don’t have to be logged in.
  75          *
  76          * @return The lowest user level required to view this page
  77          */
  78         protected int getRequiredUserLevel() {
  79                 return 0;
  80         }
  81
  82 }