Rename “localSones” method to a more traditional “getSones”.

[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 df06c87..39c3cdc 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/web/LoginPage.java
+++ b/src/main/java/net/pterodactylus/sone/web/LoginPage.java
@@ -17,7 +17,12 @@
 
 package net.pterodactylus.sone.web;
 
+import java.util.Set;
+
+import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.web.page.Page.Request.Method;
 import net.pterodactylus.util.template.Template;
+import freenet.clients.http.ToadletContext;
 
 /**
  * The login page manages logging the user in.
@@ -38,4 +43,40 @@ public class LoginPage extends SoneTemplatePage {
                super("login.html", template, "Page.Login.Title", webInterface);
        }
 
+       //
+       // TEMPLATEPAGE METHODS
+       //
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       protected void processTemplate(Request request, Template template) throws RedirectException {
+               super.processTemplate(request, template);
+               Set<Sone> localSones = webInterface.core().getSones();
+               template.set("sones", localSones);
+               if (request.getMethod() == Method.POST) {
+                       String soneId = request.getHttpRequest().getPartAsStringFailsafe("sone-id", 100);
+                       Sone selectedSone = null;
+                       for (Sone sone : webInterface.core().getSones()) {
+                               if (sone.getId().equals(soneId)) {
+                                       selectedSone = sone;
+                                       break;
+                               }
+                       }
+                       if (selectedSone != null) {
+                               setCurrentSone(request.getToadletContext(), selectedSone);
+                               throw new RedirectException("index.html");
+                       }
+               }
+       }
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       public boolean isEnabled(ToadletContext toadletContext) {
+               return getCurrentSone(toadletContext) == null;
+       }
+
 }