Pages in a own package

[gigi.git] / src / org / cacert / gigi / Gigi.java

diff --git a/src/org/cacert/gigi/Gigi.java b/src/org/cacert/gigi/Gigi.java
index 7aa26d87ef70a0752907f818339c07f60d260318..1b730f29568cb3e1cd67532816df35d5a72b8355 100644 (file)
--- a/src/org/cacert/gigi/Gigi.java
+++ b/src/org/cacert/gigi/Gigi.java
@@ -2,6 +2,7 @@ package org.cacert.gigi;
 
 import java.io.IOException;
 import java.security.cert.X509Certificate;
+import java.util.HashMap;
 
 import javax.servlet.ServletException;
 import javax.servlet.http.HttpServlet;
@@ -9,7 +10,19 @@ import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import javax.servlet.http.HttpSession;
 
+import org.cacert.gigi.pages.LoginPage;
+import org.cacert.gigi.pages.MainPage;
+
 public class Gigi extends HttpServlet {
+       private HashMap<String, Page> pages = new HashMap<String, Page>();
+
+       @Override
+       public void init() throws ServletException {
+               pages.put("/login", new LoginPage());
+               pages.put("/", new MainPage());
+               super.init();
+       }
+
        @Override
        protected void service(HttpServletRequest req, HttpServletResponse resp)
                        throws ServletException, IOException {
@@ -22,6 +35,11 @@ public class Gigi extends HttpServlet {
                                hs = req.getSession(false);
                        }
                }
+               if (hs != null && ((Boolean) hs.getAttribute("loggedin"))
+                               && req.getPathInfo().equals("/login")) {
+                       resp.sendRedirect("/");
+                       return;
+               }
                if (req.getMethod().equals("POST") && req.getPathInfo() != null
                                && req.getPathInfo().equals("/login")) {
                        authWithUnpw(req);
@@ -33,24 +51,25 @@ public class Gigi extends HttpServlet {
                                hs.setAttribute("loggedin", false);
                                hs.invalidate();
                        }
-                       authWithUnpw(req);
                        resp.sendRedirect("/");
                        return;
                }
 
-               if (hs == null || !((Boolean) hs.getAttribute("loggedin"))) {
-                       resp.setContentType("text/html");
-                       resp.getWriter().println("Access denied. Sending login form.");
-                       resp.getWriter()
-                                       .println(
-                                                       "<form method='POST' action='/login'>"
-                                                                       + "<input type='text' name='username'>"
-                                                                       + "<input type='password' name='password'> <input type='submit' value='login'></form>");
+               if ((hs == null || !((Boolean) hs.getAttribute("loggedin")))
+                               && !"/login".equals(req.getPathInfo())) {
+                       System.out.println(req.getPathInfo());
+                       resp.sendRedirect("/login");
                        return;
                }
-               resp.getWriter().println("Access granted.");
+               if (pages.containsKey(req.getPathInfo())) {
+                       Page p = pages.get(req.getPathInfo());
+                       p.doGet(req, resp);
+               } else {
+                       resp.sendError(404, "Page not found.");
+               }
 
        }
+
        private void authWithUnpw(HttpServletRequest req) {
                String un = req.getParameter("username");
                String pw = req.getParameter("password");
@@ -58,6 +77,7 @@ public class Gigi extends HttpServlet {
                HttpSession hs = req.getSession();
                hs.setAttribute("loggedin", true);
        }
+
        private void tryAuthWithCertificate(HttpServletRequest req,
                        X509Certificate x509Certificate) {
                // TODO ckeck if certificate is valid