X-Git-Url: https://git.wpitchoune.net/gitweb/?a=blobdiff_plain;f=war%2Fsrc%2Fmain%2Fjava%2Fpnews%2Fservlet%2FPnews.java;h=2682b9181f78aceb195a50dc608b8c66c3abf6e7;hb=3400b5dc00ffb699f4193dcdd02ac7b6360c89f7;hp=d15315e5491bc2ac4f25a8add53fd8ceade7cd5a;hpb=6225887c0f8a1c22754ac98941908f47281bb2d3;p=pnews.git

diff --git a/war/src/main/java/pnews/servlet/Pnews.java b/war/src/main/java/pnews/servlet/Pnews.java
index d15315e..2682b91 100644
--- a/war/src/main/java/pnews/servlet/Pnews.java
+++ b/war/src/main/java/pnews/servlet/Pnews.java
@@ -9,6 +9,7 @@ import java.io.Writer;
 import java.net.URLDecoder;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Locale;
 import java.util.logging.Level;
 import java.util.logging.Logger;
 
@@ -28,41 +29,48 @@ public class Pnews extends HttpServlet {
         private static final Logger LOG = Logger.getLogger(Pnews.class.getName());
         private static final long serialVersionUID = 1L;
         private static final ArticleProvider provider = ArticleProvider.singleton;
-        
+
         private static String getQueryParameter(HttpServletRequest rq, String key)
                         throws UnsupportedEncodingException {
                 String[] params;
                 int idx;
                 String q;
-                
+
                 q = rq.getQueryString();
-                
+
                 if (q == null)
                         return null;
-                
+
                 params = URLDecoder.decode(q, "UTF-8").split("&");
-                
-                for (String p: params) {                        
+
+                for (String p: params) {
                         idx = p.indexOf('=');
-                                                
+
                         if (idx > 1 && p.substring(0, idx).equals(key))
                                 return p.substring(idx + 1);
                 }
-                
+
                 return null;
         }
-        
+
         private static void redirect(HttpServletRequest rq, HttpServletResponse rp) {
                 String redirectURL;
-                
+                Article a;
+
                 LOG.entering(Pnews.class.getName(), "redirect");
-                
+
                 try {
                         redirectURL = getQueryParameter(rq, "url");
-                                               
+
                         LOG.info("Request redirection to " + redirectURL);
-                        
+
                         if (redirectURL != null) {
+                                a = ArticleStore.singleton.get(redirectURL);
+                                if (a != null)
+                                        a.readCount.incrementAndGet();
+                                else
+                                        LOG.severe("Cannot find the article " + redirectURL);
+                                
                                 rp.setHeader("Location", redirectURL);
                                 rp.setStatus(HttpServletResponse.SC_TEMPORARY_REDIRECT);
                         } else {
@@ -74,24 +82,33 @@ public class Pnews extends HttpServlet {
                         e.printStackTrace();
                         LOG.log(Level.SEVERE, "redirect failure", e);
                         rp.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
-                }                
-                
+                }
+
                 LOG.exiting(Pnews.class.getName(), "redirect");
         }
+
+        private void writeStats(HttpServletResponse rp) throws IOException {
+                rp.setContentType("text/html;charset=utf-8");
+                rp.setCharacterEncoding("utf8-8");
+
+                rp.getWriter().write("" + ArticleStore.singleton);
+        }
+
         
         private void writeArticles(Category cat, HttpServletResponse rp) {
                 String html;
                 List<Article> articles;
-                
-                try {   
+
+                try {
                         articles = provider.getArticles(cat);
                         if (articles != null) {
                                 html = HTML.toHTML(articles, cat);
-                                rp.setContentType("text/html");
+                                rp.setContentType("text/html;charset=utf-8");
                                 rp.getWriter().write(html);
+                                rp.setCharacterEncoding("utf8-8");
                         } else {
                                 LOG.severe("writeArticles cannot retrieve any articles");
-                                html = HTML.toHTML(new ArrayList<>(), cat);
+                                html = HTML.toHTML(new ArrayList<Article>(), cat);
                                 rp.setContentType("text/html");
                                 rp.getWriter().write(html);
                         }
@@ -100,55 +117,68 @@ public class Pnews extends HttpServlet {
                         rp.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
                 }
         }
-        
+
         private void copy(InputStream in, Writer writer) throws IOException {
                 Reader r;
                 char[] buf;
                 int n;
-                
+
                 buf = new char[1024];
                 r = new InputStreamReader(in);
                 while ( (n = r.read(buf, 0, buf.length)) != -1)
-                        writer.write(buf, 0, n);                
+                        writer.write(buf, 0, n);
         }
-        
+
         @Override
         protected void doGet(HttpServletRequest req, HttpServletResponse resp) {
                 String path;
                 InputStream in;
-                
-                LOG.info("doGet " + req.getRequestURI());
 
+                RequesterLog.singleton.writeRequest(req);
+              
+                LOG.info("doGet " + req.getRemoteAddr().toString() + " " + req.getRequestURI() + " " + req.getQueryString());
+                                
                 path = req.getPathInfo();
-                
+
                 if (path.equals("/redirect")) {
                         redirect(req, resp);
                         return ;
-                } 
-                
+                }
+
                 if (path.equals("/style.css")) {
                         try {
                                 in = HTML.class.getClassLoader().getResourceAsStream("style.css");
                                 copy(in, resp.getWriter());
                                 resp.setContentType("text/css");
-                                
+
                                 return ;
                         } catch (IOException e) {
                                 LOG.log(Level.SEVERE, "doGet failure", e);
                                 resp.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
-                                
+
                                 return ;
                         }
                 }
+
+                if (path.equals("/")) {
+                        writeArticles(Category.TOP, resp);
+                        return ;
+                }
+
+                try {
                 
-                for (Category cat: Category.values()) {
-                        if (path.equals('/' + cat.getId())) {
-                                writeArticles(cat, resp);
+                        if (path.equals("/stats")) {
+                                writeStats(resp);
                                 return ;
                         }
-                }
                 
-                try {
+                        for (Category cat: Category.values()) {
+                                if (path.equals('/' + cat.getId())) {
+                                        writeArticles(cat, resp);
+                                        return ;
+                                }
+                        }
+                
                         resp.getWriter().write("Not found " + req.getPathInfo());
                         resp.setStatus(HttpServletResponse.SC_NOT_FOUND);
                 } catch (IOException e) {
@@ -156,10 +186,10 @@ public class Pnews extends HttpServlet {
                         resp.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
                 }
         }
-        
+
         @Override
         public void init(ServletConfig config) throws ServletException {
                 LOG.info("Pnews servlet init " + config.getServletContext().getContextPath());
-                
+
         }
 }