Fix pages redirection

AMT-Project · Sep 25, 2020 · 8170784 · 8170784
1 parent a2fc71f
commit 8170784
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/main/java/ch/heigvd/amt/stack/presentation/LoginCommandServlet.java b/src/main/java/ch/heigvd/amt/stack/presentation/LoginCommandServlet.java
@@ -29,12 +29,17 @@ protected void doPost(HttpServletRequest request, HttpServletResponse response)
       out.println("USER = " + user + " : " + pass);
     });
 
-    if(RegisterCommandServlet.users.containsKey(userModel.getUsername())) {
-      if(RegisterCommandServlet.users.get(userModel.getUsername()).equals(userModel.getPassword())) {
-        request.getRequestDispatcher("/WEB-INF/views/questions.jsp").forward(request, response);
+    boolean exist = false;
+    if (RegisterCommandServlet.users.containsKey(userModel.getUsername())) {
+      if (RegisterCommandServlet.users.get(userModel.getUsername()).equals(userModel.getPassword())) {
+        exist = true;
       }
     }
 
-    request.getRequestDispatcher("/WEB-INF/views/login.jsp").forward(request, response);
+    if (exist) {
+      response.sendRedirect(request.getContextPath() + "/");
+    } else {
+      response.sendRedirect(request.getContextPath() + "/login");
+    }
   }
 }