Redirect to index page if the user is already logged in.

[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 1e281b5..c34d8ca 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/web/LoginPage.java
+++ b/src/main/java/net/pterodactylus/sone/web/LoginPage.java
@@ -68,13 +68,7 @@ public class LoginPage extends SoneTemplatePage {
                template.set("sones", localSones);
                if (request.getMethod() == Method.POST) {
                        String soneId = request.getHttpRequest().getPartAsStringFailsafe("sone-id", 100);
-                       Sone selectedSone = null;
-                       for (Sone sone : webInterface.getCore().getSones()) {
-                               if (sone.getId().equals(soneId)) {
-                                       selectedSone = sone;
-                                       break;
-                               }
-                       }
+                       Sone selectedSone = webInterface.getCore().getLocalSone(soneId);
                        if (selectedSone != null) {
                                setCurrentSone(request.getToadletContext(), selectedSone);
                                throw new RedirectException("index.html");
@@ -84,6 +78,17 @@ public class LoginPage extends SoneTemplatePage {
                template.set("identitiesWithoutSone", ownIdentitiesWithoutSone);
        }
 
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       protected String getRedirectTarget(Request request) {
+               if (getCurrentSone(request.getToadletContext()) != null) {
+                       return "index.html";
+               }
+               return null;
+       }
+
        //
        // SONETEMPLATEPAGE METHODS
        //