Add: Added support for $year$

[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 88c85ce17144d935b52519d9c7f650d0a9f783b5..7144ea270870821dcf5c566ea496898ecbfe035c 100644 (file)
--- a/src/org/cacert/gigi/Gigi.java
+++ b/src/org/cacert/gigi/Gigi.java
@@ -1,7 +1,13 @@
 package org.cacert.gigi;
 
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileInputStream;
 import java.io.IOException;
+import java.io.InputStreamReader;
 import java.security.cert.X509Certificate;
+import java.util.Calendar;
+import java.util.HashMap;
 
 import javax.servlet.ServletException;
 import javax.servlet.http.HttpServlet;
@@ -9,7 +15,36 @@ 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;
+import org.cacert.gigi.pages.Page;
+import org.eclipse.jetty.util.log.Log;
+
 public class Gigi extends HttpServlet {
+       private static final long serialVersionUID = -6386785421902852904L;
+       private String[] baseTemplate;
+       private HashMap<String, Page> pages = new HashMap<String, Page>();
+
+       @Override
+       public void init() throws ServletException {
+               pages.put("/login", new LoginPage("CACert - Login"));
+               pages.put("/", new MainPage("CACert - Home"));
+               String templ = "";
+               try {
+                       BufferedReader reader = new BufferedReader(new InputStreamReader(
+                                       new FileInputStream(new File("templates/base.html"))));
+                       String tmp;
+                       while ((tmp = reader.readLine()) != null) {
+                               templ += tmp;
+                       }
+                       baseTemplate = templ.split("\\$content\\$");
+               } catch (Exception e) {
+                       Log.getLogger(Gigi.class).warn("Error loading template!", e);
+               }
+               super.init();
+
+       }
+
        @Override
        protected void service(HttpServletRequest req, HttpServletResponse resp)
                        throws ServletException, IOException {
@@ -22,6 +57,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 +77,30 @@ 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())) {
+                       String b0 = baseTemplate[0];
+                       int year = Calendar.getInstance().get(Calendar.YEAR);
+                       Page p = pages.get(req.getPathInfo());
+                       b0 = b0.replaceAll("\\$title\\$", p.getTitle());
+                       b0 = b0.replaceAll("\\$year\\$", year + "");
+                       resp.getWriter().print(b0);
+                       p.doGet(req, resp);
+                       String b1 = baseTemplate[1];
+                       b1 = b1.replaceAll("\\$title\\$", p.getTitle());
+                       b1 = b1.replaceAll("\\$year\\$", year + "");
+                       resp.getWriter().print(b1);
+               } else {
+                       resp.sendError(404, "Page not found.");
+               }
 
        }
+
        private void authWithUnpw(HttpServletRequest req) {
                String un = req.getParameter("username");
                String pw = req.getParameter("password");
@@ -57,6 +108,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