package pnews.servlet;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.io.Reader;
+import java.io.UnsupportedEncodingException;
+import java.io.Writer;
+import java.net.URLDecoder;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+import javax.servlet.ServletConfig;
+import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import com.rometools.rome.io.FeedException;
+
+import pnews.Article;
+import pnews.Category;
public class Pnews extends HttpServlet {
+ private static final String CLASS_NAME = Pnews.class.getName();
+ private static final Logger LOG = Logger.getLogger(Pnews.class.getName());
+ private static final long serialVersionUID = 1L;
+ private ArticleProvider provider;
+ private Config config;
+
+ private static String getQueryParameter(HttpServletRequest rq, String key)
+ throws UnsupportedEncodingException {
+ final String METHOD_NAME="getQueryParameter";
+ String[] params;
+ int idx;
+ String q;
+
+
+ LOG.entering(CLASS_NAME, METHOD_NAME, new Object[] { rq, key} );
+
+ q = rq.getQueryString();
+
+ if (q == null)
+ return null;
+
+ params = q.split("&");
+
+ for (String p: params) {
+ idx = p.indexOf('=');
+
+ if (idx > 1 && p.substring(0, idx).equals(key))
+ return URLDecoder.decode(p.substring(idx + 1), "UTF-8");
+ }
+
+ return null;
+ }
+
+ private static void redirect(HttpServletRequest rq, HttpServletResponse rp) {
+ String redirectURL;
+ Article a;
+
+ LOG.entering(Pnews.class.getName(), "redirect", new Object[] { rq, rp });
+
+ 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 {
+ LOG.severe("No redirection URL");
+ rp.setStatus(HttpServletResponse.SC_NOT_FOUND);
+ }
+
+ } catch (UnsupportedEncodingException e) {
+ e.printStackTrace();
+ LOG.log(Level.SEVERE, "redirect failure", e);
+ rp.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
+ }
+
+ LOG.exiting(Pnews.class.getName(), "redirect");
+ }
+
+ private static void doTemporaryRedirect(String newURL, HttpServletResponse rp) {
+ rp.setHeader("Location", newURL);
+ rp.setStatus(HttpServletResponse.SC_TEMPORARY_REDIRECT);
+ }
+
+ private void writeStats(HttpServletResponse rp) throws IOException {
+ rp.setContentType("application/json;charset=utf-8");
+ rp.setCharacterEncoding("utf-8");
+
+ rp.getWriter().write(JSON.getStats(provider, config));
+ }
+
+
+ private void writeArticles(Category cat, HttpServletResponse rp) {
+ String html;
+ List<Article> articles;
+
+ try {
+ articles = provider.getArticles(cat);
+ if (articles != null) {
+ html = HTML.toHTML(articles, cat, config.getCategories());
+ rp.setContentType("text/html;charset=utf-8");
+ rp.getWriter().write(html);
+ rp.setCharacterEncoding("utf-8");
+ } else {
+ LOG.severe("writeArticles cannot retrieve any articles");
+ html = HTML.toHTML(new ArrayList<Article>(), cat, config.getCategories());
+ rp.setContentType("text/html");
+ rp.getWriter().write(html);
+ }
+ } catch (IOException | IllegalArgumentException | FeedException e) {
+ LOG.log(Level.SEVERE, "writeArticles failure", e);
+ 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);
+ }
+
+ @Override
+ protected void doGet(HttpServletRequest req, HttpServletResponse resp) {
+ final String METHOD_NAME = "doGet";
+ String path;
+ InputStream in;
+
+ RequesterLog.singleton.writeRequest(req);
+
+ LOG.info("doGet " + req.getRemoteAddr().toString() + " " + req.getRequestURI() + " " + req.getQueryString());
+
+ LOG.info(METHOD_NAME + " queryString=" + 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(config.getDefaultCategory(), resp);
+ return ;
+ }
+
+ try {
+ if (path.equals("/stats")) {
+ writeStats(resp);
+ return ;
+ }
+
+ for (Category cat: config.getCategories()) {
+ if (path.equals(cat.getURL())) {
+ writeArticles(cat, resp);
+ return ;
+ }
+ }
+
+ for (String l: config.getLanguages()) {
+ if (path.equals("/" + l) || path.equals("/" + l + "/")) {
+ doTemporaryRedirect(config.getDefaultCategory().getURL(), resp);
+ return ;
+ }
+ }
+
+ resp.getWriter().write("Not found " + req.getPathInfo());
+ resp.setStatus(HttpServletResponse.SC_NOT_FOUND);
+ } catch (IOException e) {
+ LOG.log(Level.SEVERE, "doGet failure", e);
+ resp.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
+ }
+ }
+ @Override
+ public void init(ServletConfig cfg) throws ServletException {
+ LOG.info("Pnews servlet init " + cfg.getServletContext().getContextPath());
+
+ config = new Config();
+ try {
+ config.loadConfig();
+ } catch (UnsupportedEncodingException e) {
+ throw new ServletException(e);
+ }
+
+ provider = new ArticleProvider(config);
+ }
}