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=5990c4fc341c2072f533a3795dc39e7e2213c1e1;hpb=125e8e399e62384048753c684b168b14922d22d3;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 5990c4f..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; /** @@ -49,4 +52,29 @@ 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); + 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; + } + }