src/org/cacert/gigi/Gigi.java

   1 package org.cacert.gigi;
   2
   3 import java.io.BufferedReader;
   4 import java.io.File;
   5 import java.io.FileInputStream;
   6 import java.io.IOException;
   7 import java.io.InputStreamReader;
   8 import java.util.Calendar;
   9 import java.util.HashMap;
  10
  11 import javax.servlet.ServletException;
  12 import javax.servlet.http.HttpServlet;
  13 import javax.servlet.http.HttpServletRequest;
  14 import javax.servlet.http.HttpServletResponse;
  15 import javax.servlet.http.HttpSession;
  16
  17 import org.cacert.gigi.pages.LoginPage;
  18 import org.cacert.gigi.pages.MainPage;
  19 import org.cacert.gigi.pages.Page;
  20 import org.cacert.gigi.pages.TestSecure;
  21 import org.cacert.gigi.pages.main.RegisterPage;
  22 import org.eclipse.jetty.util.log.Log;
  23
  24 public class Gigi extends HttpServlet {
  25         public static final String LOGGEDIN = "loggedin";
  26         public static final String USER = "user";
  27         private static final long serialVersionUID = -6386785421902852904L;
  28         private String[] baseTemplate;
  29         private HashMap<String, Page> pages = new HashMap<String, Page>();
  30
  31         @Override
  32         public void init() throws ServletException {
  33                 pages.put("/login", new LoginPage("CACert - Login"));
  34                 pages.put("/", new MainPage("CACert - Home"));
  35                 pages.put("/secure", new TestSecure());
  36                 pages.put(RegisterPage.PATH, new RegisterPage());
  37                 String templ = "";
  38                 try {
  39                         BufferedReader reader = new BufferedReader(new InputStreamReader(
  40                                         new FileInputStream(new File("templates/base.html"))));
  41                         String tmp;
  42                         while ((tmp = reader.readLine()) != null) {
  43                                 templ += tmp;
  44                         }
  45                         baseTemplate = templ.split("\\$content\\$");
  46                 } catch (Exception e) {
  47                         Log.getLogger(Gigi.class).warn("Error loading template!", e);
  48                 }
  49                 super.init();
  50
  51         }
  52         @Override
  53         protected void service(HttpServletRequest req, HttpServletResponse resp)
  54                         throws ServletException, IOException {
  55                 HttpSession hs = req.getSession();
  56                 if (req.getPathInfo() != null && req.getPathInfo().equals("/logout")) {
  57                         if (hs != null) {
  58                                 hs.setAttribute(LOGGEDIN, null);
  59                                 hs.invalidate();
  60                         }
  61                         resp.sendRedirect("/");
  62                         return;
  63                 }
  64
  65                 if (pages.containsKey(req.getPathInfo())) {
  66                         Page p = pages.get(req.getPathInfo());
  67                         if (p.needsLogin() && hs.getAttribute("loggedin") == null) {
  68                                 resp.sendRedirect("/login");
  69                                 return;
  70                         }
  71
  72                         String b0 = baseTemplate[0];
  73                         b0 = makeDynTempl(b0, p);
  74                         resp.setContentType("text/html; charset=utf-8");
  75                         resp.getWriter().print(b0);
  76                         if (req.getMethod().equals("POST")) {
  77                                 p.doPost(req, resp);
  78                         } else {
  79                                 p.doGet(req, resp);
  80                         }
  81                         String b1 = baseTemplate[1];
  82                         b1 = makeDynTempl(b1, p);
  83                         resp.getWriter().print(b1);
  84                 } else {
  85                         resp.sendError(404, "Page not found.");
  86                 }
  87
  88         }
  89         private String makeDynTempl(String in, Page p) {
  90                 int year = Calendar.getInstance().get(Calendar.YEAR);
  91                 in = in.replaceAll("\\$title\\$", p.getTitle());
  92                 in = in.replaceAll("\\$year\\$", year + "");
  93                 return in;
  94         }
  95
  96 }