X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fdemoscenemusic%2Fpage%2FBasePage.java;h=f6f211e3f578c27fa106b1f3e2b4bef0f9eea75f;hb=773eb9791a6a77cfaf7a5f8ee1f103dd21277d36;hp=2e5462ad7786d5f293ed0c871664baf3f976d925;hpb=7086f496546e7569b23146a9bcd5b2c5a0ad1396;p=demoscenemusic.git diff --git a/src/main/java/net/pterodactylus/demoscenemusic/page/BasePage.java b/src/main/java/net/pterodactylus/demoscenemusic/page/BasePage.java index 2e5462a..f6f211e 100644 --- a/src/main/java/net/pterodactylus/demoscenemusic/page/BasePage.java +++ b/src/main/java/net/pterodactylus/demoscenemusic/page/BasePage.java @@ -18,8 +18,11 @@ package net.pterodactylus.demoscenemusic.page; import net.pterodactylus.demoscenemusic.core.Core; +import net.pterodactylus.demoscenemusic.data.User; import net.pterodactylus.util.template.Template; +import net.pterodactylus.util.template.TemplateContext; import net.pterodactylus.util.template.TemplateContextFactory; +import net.pterodactylus.util.web.RedirectException; import net.pterodactylus.util.web.TemplatePage; /** @@ -38,7 +41,7 @@ public class BasePage extends TemplatePage { * @param template */ public BasePage(Core core, TemplateContextFactory templateContextFactory, Template template, String pageName) { - super(pageName, "text/html", templateContextFactory, template); + super(pageName, "text/html; charset=utf-8", templateContextFactory, template); this.core = core; } @@ -49,4 +52,31 @@ public class BasePage extends TemplatePage { return core; } + /** + * {@inheritDoc} + */ + @Override + protected void processTemplate(TemplateContext templateContext, ServletRequest request) throws RedirectException { + super.processTemplate(templateContext, request); + User currentUser = (User) request.getServletRequest().getSession().getAttribute("currentUser"); + templateContext.set("currentUser", currentUser); + templateContext.set("core", getCore()); + templateContext.set("dataManager", getCore().getDataManager()); + int requiredUserLevel = getRequiredUserLevel(); + if (((currentUser == null) && (requiredUserLevel > 0)) || ((currentUser != null) && (requiredUserLevel > currentUser.getLevel()))) { + throw new RedirectException("login"); + } + } + + /** + * Returns the {@link User#getLevel() user level} that is at least required + * to access the given page. If the returned level is {@code 0} (or + * smaller), users don’t have to be logged in. + * + * @return The lowest user level required to view this page + */ + protected int getRequiredUserLevel() { + return 0; + } + }