X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fdemoscenemusic%2Fpage%2FBasePage.java;h=65505fa354d9673cfd17c74c0a70a56918769ea4;hb=8502cb24d28777f25f91af84c77c6d250c40716a;hp=c24e6aff34177bb1da47f834829f88c912edfe4d;hpb=02b429ae28a89151cd75227485e6956433e50c65;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 c24e6af..65505fa 100644
--- a/src/main/java/net/pterodactylus/demoscenemusic/page/BasePage.java
+++ b/src/main/java/net/pterodactylus/demoscenemusic/page/BasePage.java
@@ -18,6 +18,7 @@
 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;
@@ -57,7 +58,23 @@ public class BasePage extends TemplatePage<ServletRequest> {
 	@Override
 	protected void processTemplate(TemplateContext templateContext, ServletRequest request) throws RedirectException {
 		super.processTemplate(templateContext, request);
-		templateContext.set("currentUser", request.getServletRequest().getSession().getAttribute("currentUser"));
+		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;
 	}
 
 }