Render templates in a thread-safe way.

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

diff --git a/src/main/java/net/pterodactylus/sone/web/LoginPage.java b/src/main/java/net/pterodactylus/sone/web/LoginPage.java
index 8cebf49..ff81562 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/web/LoginPage.java
+++ b/src/main/java/net/pterodactylus/sone/web/LoginPage.java
@@ -17,11 +17,18 @@
 
 package net.pterodactylus.sone.web;
 
-import java.util.Set;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.logging.Logger;
 
 import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.freenet.wot.OwnIdentity;
 import net.pterodactylus.sone.web.page.Page.Request.Method;
+import net.pterodactylus.util.logging.Logging;
+import net.pterodactylus.util.template.DataProvider;
 import net.pterodactylus.util.template.Template;
+import freenet.clients.http.ToadletContext;
 
 /**
  * The login page manages logging the user in.
@@ -30,6 +37,10 @@ import net.pterodactylus.util.template.Template;
  */
 public class LoginPage extends SoneTemplatePage {
 
+       /** The logger. */
+       @SuppressWarnings("unused")
+       private static final Logger logger = Logging.getLogger(LoginPage.class);
+
        /**
         * Creates a new login page.
         *
@@ -39,7 +50,7 @@ public class LoginPage extends SoneTemplatePage {
         *            The Sone web interface
         */
        public LoginPage(Template template, WebInterface webInterface) {
-               super("login.html", template, "Page.Login.Title", webInterface);
+               super("login.html", template, "Page.Login.Title", webInterface, false);
        }
 
        //
@@ -50,9 +61,22 @@ public class LoginPage extends SoneTemplatePage {
         * {@inheritDoc}
         */
        @Override
-       protected void processTemplate(Request request, Template template) {
-               Set<Sone> localSones = webInterface.core().localSones();
-               template.set("sones", localSones);
+       protected void processTemplate(Request request, DataProvider dataProvider) throws RedirectException {
+               super.processTemplate(request, dataProvider);
+               /* get all own identities. */
+               List<Sone> localSones = new ArrayList<Sone>(webInterface.getCore().getLocalSones());
+               Collections.sort(localSones, Sone.NICE_NAME_COMPARATOR);
+               dataProvider.set("sones", localSones);
+               if (request.getMethod() == Method.POST) {
+                       String soneId = request.getHttpRequest().getPartAsStringFailsafe("sone-id", 100);
+                       Sone selectedSone = webInterface.getCore().getLocalSone(soneId, false);
+                       if (selectedSone != null) {
+                               setCurrentSone(request.getToadletContext(), selectedSone);
+                               throw new RedirectException("index.html");
+                       }
+               }
+               List<OwnIdentity> ownIdentitiesWithoutSone = CreateSonePage.getOwnIdentitiesWithoutSone(webInterface.getCore());
+               dataProvider.set("identitiesWithoutSone", ownIdentitiesWithoutSone);
        }
 
        /**
@@ -60,21 +84,22 @@ public class LoginPage extends SoneTemplatePage {
         */
        @Override
        protected String getRedirectTarget(Request request) {
-               if (request.getMethod() == Method.POST) {
-                       String soneId = request.getHttpRequest().getParam("sone-id");
-                       Sone selectedSone = null;
-                       for (Sone sone : webInterface.core().localSones()) {
-                               if (sone.getId().equals(soneId)) {
-                                       selectedSone = sone;
-                                       break;
-                               }
-                       }
-                       if (selectedSone != null) {
-                               setCurrentSone(request, selectedSone);
-                               return "index.html";
-                       }
+               if (getCurrentSone(request.getToadletContext()) != null) {
+                       return "index.html";
                }
                return null;
        }
 
+       //
+       // SONETEMPLATEPAGE METHODS
+       //
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       public boolean isEnabled(ToadletContext toadletContext) {
+               return getCurrentSone(toadletContext) == null;
+       }
+
 }