X-Git-Url: https://code.wpia.club/?p=gigi.git;a=blobdiff_plain;f=src%2Forg%2Fcacert%2Fgigi%2FGigi.java;h=1b730f29568cb3e1cd67532816df35d5a72b8355;hp=88c85ce17144d935b52519d9c7f650d0a9f783b5;hb=f00a29a230689e952d413818073fff0644e04786;hpb=46b696eff985b3500a267c021a2ff0550bd34b1f

diff --git a/src/org/cacert/gigi/Gigi.java b/src/org/cacert/gigi/Gigi.java
index 88c85ce1..1b730f29 100644
--- 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);
@@ -37,19 +55,21 @@ public class Gigi extends HttpServlet {
 			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");
@@ -57,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