X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fdemoscenemusic%2Fpage%2FBasePage.java;h=65505fa354d9673cfd17c74c0a70a56918769ea4;hb=577d2d5723539b8dc25d09a9a51963985f1c449f;hp=988fee78151a6e28d1d6cd6186e9033e4288ba47;hpb=80f8e65ec0bf6a02ef20a1e89e708cecefe76ad3;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 988fee7..65505fa 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; /** @@ -29,6 +32,8 @@ import net.pterodactylus.util.web.TemplatePage; */ public class BasePage extends TemplatePage { + private final Core core; + /** * @param path * @param contentType @@ -36,8 +41,40 @@ 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; } -} + /** + * @return the core + */ + public Core getCore() { + 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); + 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; + } + +}