Get current user from session correctly.

[demoscenemusic.git] / src / main / java / net / pterodactylus / demoscenemusic / page / BasePage.java

diff --git a/src/main/java/net/pterodactylus/demoscenemusic/page/BasePage.java b/src/main/java/net/pterodactylus/demoscenemusic/page/BasePage.java
index 988fee7..c24e6af 100644 (file)
--- a/src/main/java/net/pterodactylus/demoscenemusic/page/BasePage.java
+++ b/src/main/java/net/pterodactylus/demoscenemusic/page/BasePage.java
@@ -19,7 +19,9 @@ package net.pterodactylus.demoscenemusic.page;
 
 import net.pterodactylus.demoscenemusic.core.Core;
 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 +31,8 @@ import net.pterodactylus.util.web.TemplatePage;
  */
 public class BasePage extends TemplatePage<ServletRequest> {
 
+       private final Core core;
+
        /**
         * @param path
         * @param contentType
@@ -36,8 +40,24 @@ public class BasePage extends TemplatePage<ServletRequest> {
         * @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);
+               templateContext.set("currentUser", request.getServletRequest().getSession().getAttribute("currentUser"));
+       }
+
+}