UPD: Also replace in the footer

[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 b4ebf8b770e8652e00cb37091ec3a392857413cb..a888b0912cb7e4d5eaa17fbc82283306db08394b 100644 (file)
--- a/src/org/cacert/gigi/Gigi.java
+++ b/src/org/cacert/gigi/Gigi.java
@@ -1,6 +1,10 @@
 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.HashMap;
 
@@ -10,14 +14,34 @@ 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());
-               pages.put("/", new MainPage());
+               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
@@ -58,10 +82,15 @@ public class Gigi extends HttpServlet {
                        resp.sendRedirect("/login");
                        return;
                }
-               System.out.println(req.getPathInfo());
                if (pages.containsKey(req.getPathInfo())) {
+                       String b0 = baseTemplate[0];
                        Page p = pages.get(req.getPathInfo());
+                       b0 = b0.replaceAll("\\$title\\$", p.getTitle());
+                       resp.getWriter().print(b0);
                        p.doGet(req, resp);
+                       String b1 = baseTemplate[1];
+                       b1 = b1.replaceAll("\\$title\\$", p.getTitle());
+                       resp.getWriter().print(b1);
                } else {
                        resp.sendError(404, "Page not found.");
                }