From a6603fc8a432b92a6e18db86156efc7b31724093 Mon Sep 17 00:00:00 2001 From: Vitaly Takmazov Date: Tue, 13 Dec 2016 19:29:36 +0300 Subject: Working RSS spring-webmvc servlet, ugnich-rss generator moved to tests --- src/test/java/com/juick/rss/LegacyRSS.java | 219 +++++++++++++++++++++++++++++ 1 file changed, 219 insertions(+) create mode 100644 src/test/java/com/juick/rss/LegacyRSS.java (limited to 'src/test/java/com') diff --git a/src/test/java/com/juick/rss/LegacyRSS.java b/src/test/java/com/juick/rss/LegacyRSS.java new file mode 100644 index 00000000..ff84b784 --- /dev/null +++ b/src/test/java/com/juick/rss/LegacyRSS.java @@ -0,0 +1,219 @@ +/* + * Juick + * Copyright (C) 2008-2013, ugnich + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as + * published by the Free Software Foundation, either version 3 of the + * License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see . + */ +package com.juick.rss; + +import com.juick.Message; +import com.juick.Tag; +import com.juick.server.MessagesQueries; +import com.juick.server.UserQueries; +import com.juick.util.DateFormattersHolder; +import com.juick.util.MessageUtils; +import org.apache.commons.dbcp2.BasicDataSource; +import org.apache.commons.lang3.StringEscapeUtils; +import org.apache.commons.lang3.math.NumberUtils; +import org.springframework.jdbc.core.JdbcTemplate; + +import javax.servlet.ServletException; +import javax.servlet.annotation.WebServlet; +import javax.servlet.http.HttpServlet; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.io.IOException; +import java.io.PrintWriter; +import java.util.Date; +import java.util.List; +import java.util.Properties; + +/** + * @author ugnich + */ +public class LegacyRSS { + + JdbcTemplate sql; + + public void LegacyRSS(JdbcTemplate sql) { + this.sql = sql; + } + + protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { + if (request.getCharacterEncoding() == null) { + request.setCharacterEncoding("UTF-8"); + } + + String uri = request.getRequestURI(); + if (uri.equals("/")) { + int hours = NumberUtils.toInt(request.getParameter("hours"), 0); + if (hours > 0 && hours < 13) { + List mids = getLastMessages(hours); + List msgs = MessagesQueries.getMessages(sql, mids); + responseMessages(response, 0, null, msgs); + } else { + response.sendError(HttpServletResponse.SC_NOT_FOUND); + } + } else if (uri.equals("/comments")) { + int hours = NumberUtils.toInt(request.getParameter("hours"), 0); + if (hours > 0 && hours < 13) { + responseReplies(response, hours); + } else { + response.sendError(HttpServletResponse.SC_NOT_FOUND); + } + } else if (uri.matches("^/[a-zA-Z0-9\\-]{2,16}/blog$")) { + String uname = uri.substring(1, uri.length() - 5); + int uid = UserQueries.getUIDbyName(sql, uname); + if (uid > 0) { + List mids = MessagesQueries.getUserBlog(sql, uid, 0, 0); + if (!mids.isEmpty()) { + List msgs = MessagesQueries.getMessages(sql, mids); + responseMessages(response, uid, uname, msgs); + } else { + response.sendError(HttpServletResponse.SC_NOT_FOUND); + } + } else { + response.sendError(HttpServletResponse.SC_NOT_FOUND); + } + } else { + response.sendError(HttpServletResponse.SC_NOT_FOUND); + } + } + + private void responseMessages(HttpServletResponse response, int uid, String uname, List msgs) throws IOException { + response.setContentType("application/rss+xml; charset=UTF-8"); + + try (PrintWriter out = response.getWriter()) { + out.println(""); + out.println(""); + out.println(""); + if (uid > 0) { + out.println(""); + out.println("" + uname + " - Juick"); + out.println("http://juick.com/" + uname + "/"); + out.println("The latest messages by @" + uname + " at Juick"); + out.println("http://i.juick.com/a/" + uid + ".png" + uname + " - Juickhttp://juick.com/" + uname + "/"); + } else { + out.println("Juick"); + out.println("http://juick.com/"); + out.println("The latest messages at Juick"); + } + + for (Message msg : msgs) { + out.println(""); + out.println("http://juick.com/" + msg.getUser().getName() + "/" + msg.getMid() + ""); + out.println("http://juick.com/" + msg.getUser().getName() + "/" + msg.getMid() + ""); + + out.print("<![CDATA[@" + msg.getUser().getName() + ":"); + if (!msg.getTags().isEmpty()) { + for (Tag tag : msg.getTags()) { + out.print(" *" + tag); + } + } + out.println("]]>"); + out.println(""); + + out.println("" + DateFormattersHolder.getRssFormatterInstance().format(msg.getDate()) + ""); + + out.println("http://juick.com/" + msg.getUser().getName() + "/" + msg.getMid() + ""); + if (!msg.getTags().isEmpty()) { + for (Tag tag : msg.getTags()) { + out.println("" + StringEscapeUtils.escapeHtml4(tag.getName()) + ""); + } + } + if (msg.getAttachmentType() != null) { + if (msg.getAttachmentType().equals("jpg")) { + out.println(""); + out.println(""); + } else if (msg.getAttachmentType().equals("png")) { + out.println(""); + out.println(""); + } + } + out.println(""); + out.println(""); + } + + out.println(""); + } + } + + private class ResponseReply { + String muname; + int mid; + int rid; + String uname; + String description; + Date pubDate; + String attachmentType; + } + + private void responseReplies(HttpServletResponse response, int hours) throws IOException { + response.setContentType("application/rss+xml; charset=UTF-8"); + + try (PrintWriter out = response.getWriter()) { + + out.println(""); + out.println(""); + out.println(""); + out.println("Juick"); + out.println("http://juick.com/"); + out.println("The latest comments at Juick"); + + sql.query("SELECT users2.nick,replies.message_id,replies.reply_id," + + "users.nick,replies.txt," + + "replies.ts,replies.attach,replies.ts+0 " + + "FROM ((replies INNER JOIN users ON replies.user_id=users.id) " + + "INNER JOIN messages ON replies.message_id=messages.message_id) " + + "INNER JOIN users AS users2 ON messages.user_id=users2.id " + + "WHERE replies.ts>TIMESTAMPADD(HOUR,?,NOW()) AND messages.privacy>0", (rs, rowNum) -> { + ResponseReply reply = new ResponseReply(); + reply.muname = rs.getString(1); + reply.mid = rs.getInt(2); + reply.rid = rs.getInt(3); + reply.uname = rs.getString(4); + reply.description = rs.getString(5); + reply.pubDate = rs.getTimestamp(6); + reply.attachmentType = rs.getString(7); + return reply; + }, -hours).forEach(r -> { + out.println(""); + out.println("http://juick.com/" + r.muname + "/" + r.mid + ""); + out.println("http://juick.com/" + r.muname + "/" + r.mid + "#" + r.rid + ""); + out.println("http://juick.com/" + r.muname + "/" + r.mid + "#" + r.rid + ""); + out.println("http://juick.com/" + r.uname + "/"); + out.println("@" + r.uname + ":"); + out.println(""); + out.println("" + DateFormattersHolder.getRssFormatterInstance().format(r.pubDate) + ""); + String attachment = r.attachmentType; + if (attachment != null && !attachment.isEmpty()) { + if (attachment.equals("jpg")) { + out.println(""); + out.println(""); + } else if (attachment.equals("png")) { + out.println(""); + out.println(""); + } + } + out.println(""); + }); + out.println(""); + } + } + + private List getLastMessages(int hours) { + return sql.queryForList("SELECT message_id FROM messages WHERE messages.ts>TIMESTAMPADD(HOUR,?,NOW())", + Integer.class, -hours); + } +} -- cgit v1.2.3