Add method to set the current Sone.

[Sone.git] / src / main / java / net / pterodactylus / sone / web / SoneTemplatePage.java

diff --git a/src/main/java/net/pterodactylus/sone/web/SoneTemplatePage.java b/src/main/java/net/pterodactylus/sone/web/SoneTemplatePage.java
index 67c5293..e0b42b1 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/web/SoneTemplatePage.java
+++ b/src/main/java/net/pterodactylus/sone/web/SoneTemplatePage.java
@@ -19,6 +19,7 @@ package net.pterodactylus.sone.web;
 
 import java.util.Arrays;
 import java.util.Collection;
+import java.util.UUID;
 
 import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.web.page.Page;
@@ -60,7 +61,8 @@ public class SoneTemplatePage extends TemplatePage {
        //
 
        /**
-        * Returns the current session.
+        * Returns the current session, creating a new session if there is no
+        * current session.
         *
         * @param request
         *            The request to extract the session information from
@@ -68,8 +70,28 @@ public class SoneTemplatePage extends TemplatePage {
         *         session
         */
        protected Session getCurrentSession(Request request) {
+               return getCurrentSession(request, true);
+       }
+
+       /**
+        * Returns the current session, creating a new session if there is no
+        * current session and {@code create} is {@code true}.
+        *
+        * @param request
+        *            The request to extract the session information from
+        * @param create
+        *            {@code true} to create a new session if there is no current
+        *            session, {@code false} otherwise
+        * @return The current session, or {@code null} if there is no current
+        *         session
+        */
+       protected Session getCurrentSession(Request request, boolean create) {
                try {
-                       return webInterface.sessionManager().useSession(request.getToadletContext());
+                       Session session = webInterface.sessionManager().useSession(request.getToadletContext());
+                       if (create && (session == null)) {
+                               session = webInterface.sessionManager().createSession(UUID.randomUUID().toString(), request.getToadletContext());
+                       }
+                       return session;
                } catch (RedirectException re1) {
                        return null;
                }
@@ -91,6 +113,19 @@ public class SoneTemplatePage extends TemplatePage {
                return (Sone) session.getAttribute("Sone.CurrentSone");
        }
 
+       /**
+        * Sets the currently logged in Sone.
+        *
+        * @param request
+        *            The request
+        * @param sone
+        *            The Sone to set as currently logged in
+        */
+       protected void setCurrentSone(Request request, Sone sone) {
+               Session session = getCurrentSession(request);
+               session.setAttribute("Sone.CurrentSone", sone);
+       }
+
        //
        // TEMPLATEPAGE METHODS
        //