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;
//
/**
- * Returns the currently logged in Sone.
+ * Returns the current session, creating a new session if there is no
+ * current session.
*
* @param request
* The request to extract the session information from
- * @return The currently logged in Sone, or {@code null} if no Sone is
- * currently logged in
+ * @return The current session, or {@code null} if there is no current
+ * session
*/
- protected Sone getCurrentSone(Request request) {
+ 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 {
Session session = webInterface.sessionManager().useSession(request.getToadletContext());
- if (session == null) {
- return null;
+ if (create && (session == null)) {
+ session = webInterface.sessionManager().createSession(UUID.randomUUID().toString(), request.getToadletContext());
}
- return (Sone) session.getAttribute("Sone.CurrentSone");
+ return session;
} catch (RedirectException re1) {
- /* okay, no current session, return null. */
return null;
}
}
+ /**
+ * Returns the currently logged in Sone.
+ *
+ * @param request
+ * The request to extract the session information from
+ * @return The currently logged in Sone, or {@code null} if no Sone is
+ * currently logged in
+ */
+ protected Sone getCurrentSone(Request request) {
+ Session session = getCurrentSession(request);
+ if (session == null) {
+ return null;
+ }
+ 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
//