Initial commit

2 files changed, 407 insertions, 0 deletions

diff --git a/src/java/com/juick/rss/Main.java b/src/java/com/juick/rss/Main.java
new file mode 100644
index 00000000..65169fbb
--- /dev/null
+++ b/src/java/com/juick/rss/Main.java
@@ -0,0 +1,333 @@
+/*
+ * 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 <http://www.gnu.org/licenses/>.
+ */
+package com.juick.rss;
+
+import com.juick.Message;
+import com.juick.server.MessagesQueries;
+import com.juick.server.UserQueries;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.PrintWriter;
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.text.ParseException;
+import java.text.SimpleDateFormat;
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.Iterator;
+import java.util.Properties;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+import javax.servlet.ServletException;
+import javax.servlet.annotation.WebServlet;
+import javax.servlet.http.HttpServlet;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+/**
+ *
+ * @author ugnich
+ */
+@WebServlet(name = "Main", urlPatterns = {"/"})
+public class Main extends HttpServlet {
+
+    private static SimpleDateFormat sdfSQL = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+    private static SimpleDateFormat sdfRSS = new SimpleDateFormat("EEE, d MMM yyyy HH:mm:ss Z");
+    Connection sql;
+
+    @Override
+    public void init() throws ServletException {
+        super.init();
+        try {
+            Properties conf = new Properties();
+            conf.load(new FileInputStream("/etc/juick/rss.conf"));
+
+            Class.forName("com.mysql.jdbc.Driver");
+            sql = DriverManager.getConnection("jdbc:mysql://localhost/juick?autoReconnect=true&user=" + conf.getProperty("mysql_username", "") + "&password=" + conf.getProperty("mysql_password", ""));
+
+        } catch (Exception e) {
+            log(null, e);
+        }
+    }
+
+    @Override
+    public void destroy() {
+        super.destroy();
+        if (sql != null) {
+            try {
+                sql.close();
+                sql = null;
+            } catch (SQLException e) {
+                log(null, e);
+            }
+        }
+    }
+
+    @Override
+    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 = Utils.parseInt(request.getParameter("hours"), 0);
+            if (hours > 0 && hours < 13) {
+                ArrayList<Integer> mids = getLastMessages(hours);
+                ArrayList<Message> msgs = MessagesQueries.getMessages(sql, mids);
+                responseMessages(response, 0, null, msgs);
+            } else {
+                response.sendError(404);
+            }
+        } else if (uri.equals("/comments")) {
+            int hours = Utils.parseInt(request.getParameter("hours"), 0);
+            if (hours > 0 && hours < 13) {
+                responseReplies(response, hours);
+            } else {
+                response.sendError(404);
+            }
+        } 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) {
+                ArrayList<Integer> mids = MessagesQueries.getUserBlog(sql, uid, 0, 0);
+                ArrayList<Message> msgs = MessagesQueries.getMessages(sql, mids);
+                responseMessages(response, uid, uname, msgs);
+            } else {
+                response.sendError(404);
+            }
+        } else {
+            response.sendError(404);
+        }
+    }
+
+    private void responseMessages(HttpServletResponse response, int uid, String uname, ArrayList<Message> msgs) throws IOException {
+        response.setContentType("application/rss+xml; charset=UTF-8");
+
+        PrintWriter out = response.getWriter();
+        try {
+            out.println("<?xml version='1.0' encoding='utf-8'?>");
+            out.println("<rss version='2.0' xmlns:atom='http://www.w3.org/2005/Atom' xmlns:slash='http://purl.org/rss/1.0/modules/slash/' xmlns:rdf='http://www.w3.org/1999/02/22-rdf-syntax-ns#' xmlns:media='http://search.yahoo.com/mrss/'  xmlns:juick='http://juick.com/'>");
+            out.println("<channel>");
+            if (uid > 0) {
+                out.println("<atom:link href='http://rss.juick.com/" + uname + "/blog' rel='self' type='application/rss+xml'/>");
+                out.println("<title>" + uname + " - Juick</title>");
+                out.println("<link>http://juick.com/" + uname + "/</link>");
+                out.println("<description>The latest messages by @" + uname + " at Juick</description>");
+                out.println("<image><url>http://i.juick.com/a/" + uid + ".png</url><title>" + uname + " - Juick</title><link>http://juick.com/" + uname + "/</link></image>");
+            } else {
+                out.println("<title>Juick</title>");
+                out.println("<link>http://juick.com/</link>");
+                out.println("<description>The latest messages at Juick</description>");
+            }
+
+            Iterator<Message> i = msgs.iterator();
+            while (i.hasNext()) {
+                Message msg = i.next();
+
+                out.println("<item>");
+                out.println("<link>http://juick.com/" + msg.User.UName + "/" + msg.MID + "</link>");
+                out.println("<guid>http://juick.com/" + msg.User.UName + "/" + msg.MID + "</guid>");
+
+                out.print("<title><![CDATA[@" + msg.User.UName + ":");
+                if (!msg.Tags.isEmpty()) {
+                    for (int n = 0; n < msg.Tags.size(); n++) {
+                        out.print(" *" + msg.Tags.get(n));
+                    }
+                }
+                out.println("]]></title>");
+                out.println("<description><![CDATA[" + formatMessage(msg.Text) + "]]></description>");
+
+                try {
+                    Date date = sdfSQL.parse(msg.TimestampString);
+                    out.println("<pubDate>" + sdfRSS.format(date) + "</pubDate>");
+                } catch (ParseException e) {
+                }
+
+                out.println("<comments>http://juick.com/" + msg.User.UName + "/" + msg.MID + "</comments>");
+                if (!msg.Tags.isEmpty()) {
+                    for (int n = 0; n < msg.Tags.size(); n++) {
+                        out.println("<category>" + msg.Tags.get(n) + "</category>");
+                    }
+                }
+                if (msg.AttachmentType != null) {
+                    if (msg.AttachmentType.equals("jpg")) {
+                        out.println("<media:content url='http://i.juick.com/photos-1024/" + msg.MID + ".jpg' type='image/jpeg'/>");
+                        out.println("<media:thumbnail url='http://i.juick.com/ps/" + msg.MID + ".jpg'/>");
+                    } else if (msg.AttachmentType.equals("png")) {
+                        out.println("<media:content url='http://i.juick.com/photos-1024/" + msg.MID + ".png' type='image/png'/>");
+                        out.println("<media:thumbnail url='http://i.juick.com/ps/" + msg.MID + ".png'/>");
+                    }
+                }
+                out.println("<juick:user uid='" + msg.User.UID + "'/>");
+                out.println("</item>");
+            }
+
+            out.println("</channel></rss>");
+        } finally {
+            out.close();
+        }
+    }
+
+    private void responseReplies(HttpServletResponse response, int hours) throws IOException {
+        response.setContentType("application/rss+xml; charset=UTF-8");
+
+        PrintWriter out = response.getWriter();
+        try {
+
+            out.println("<?xml version=\"1.0\" encoding=\"utf-8\"?>");
+            out.println("<rss version=\"2.0\" xmlns:rdf=\"http://www.w3.org/1999/02/22-rdf-syntax-ns#\" xmlns:ya=\"http://blogs.yandex.ru/yarss/\" xmlns:media=\"http://search.yahoo.com/mrss/\">");
+            out.println("<channel>");
+            out.println("<title>Juick</title>");
+            out.println("<link>http://juick.com/</link>");
+            out.println("<description>The latest comments at Juick</description>");
+
+            PreparedStatement stmt = null;
+            ResultSet rs = null;
+            try {
+                stmt = sql.prepareStatement("SELECT users2.nick,replies.message_id,replies.reply_id,users.nick,replies.txt,DATE_FORMAT(replies.ts,'%a, %d %b %Y %T UT'),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");
+                stmt.setInt(1, -hours);
+                rs = stmt.executeQuery();
+                rs.beforeFirst();
+                while (rs.next()) {
+                    String muname = rs.getString(1);
+                    int mid = rs.getInt(2);
+                    int rid = rs.getInt(3);
+                    String uname = rs.getString(4);
+
+                    out.println("<item>");
+                    out.println("<ya:post>http://juick.com/" + muname + "/" + mid + "</ya:post>");
+                    out.println("<link>http://juick.com/" + muname + "/" + mid + "#" + rid + "</link>");
+                    out.println("<guid>http://juick.com/" + muname + "/" + mid + "#" + rid + "</guid>");
+                    out.println("<author>http://juick.com/" + uname + "/</author>");
+                    out.println("<title>@" + uname + ":</title>");
+                    out.println("<description><![CDATA[" + formatMessage(rs.getString(5)) + "]]></description>");
+                    out.println("<pubDate>" + rs.getString(6) + "</pubDate>");
+                    String attachment = rs.getString(7);
+                    if (attachment != null && !attachment.isEmpty()) {
+                        if (attachment.equals("jpg")) {
+                            out.println("<media:content url='http://i.juick.com/photos-1024/" + mid + "-" + rid + ".jpg' type='image/jpeg'/>");
+                            out.println("<media:thumbnail url='http://i.juick.com/ps/" + mid + "-" + rid + ".jpg'/>");
+                        } else if (attachment.equals("png")) {
+                            out.println("<media:content url='http://i.juick.com/photos-1024/" + mid + "-" + rid + ".png' type='image/png'/>");
+                            out.println("<media:thumbnail url='http://i.juick.com/ps/" + mid + "-" + rid + ".png'/>");
+                        }
+                    }
+                    out.println("</item>");
+                }
+            } catch (SQLException e) {
+                System.err.println(e);
+            } finally {
+                Utils.finishSQL(rs, stmt);
+            }
+
+            out.println("</channel></rss>");
+        } finally {
+            out.close();
+        }
+    }
+
+    private ArrayList<Integer> getLastMessages(int hours) {
+        ArrayList<Integer> mids = new ArrayList<Integer>(20);
+
+        PreparedStatement stmt = null;
+        ResultSet rs = null;
+        try {
+            stmt = sql.prepareStatement("SELECT message_id FROM messages WHERE messages.ts>TIMESTAMPADD(HOUR,?,NOW())");
+            stmt.setInt(1, -hours);
+            rs = stmt.executeQuery();
+            rs.beforeFirst();
+            while (rs.next()) {
+                mids.add(rs.getInt(1));
+            }
+        } catch (SQLException e) {
+            System.err.println(e);
+        } finally {
+            Utils.finishSQL(rs, stmt);
+        }
+        return mids;
+    }
+    private static Pattern regexLinks2 = Pattern.compile("((?<=\\s)|(?<=\\A))([\\[\\{]|&lt;)((?:ht|f)tps?://(?:www\\.)?([^\\/\\s\\\"\\)\\!]+)/?(?:[^\\]\\}](?<!&gt;))*)([\\]\\}]|&gt;)");
+
+    public static String formatMessage(String msg) {
+        msg = msg.replaceAll("&", "&amp;");
+        msg = msg.replaceAll("<", "&lt;");
+        msg = msg.replaceAll(">", "&gt;");
+
+        // --
+        // &mdash;
+        msg = msg.replaceAll("((?<=\\s)|(?<=\\A))\\-\\-?((?=\\s)|(?=\\Z))", "$1&mdash;$2");
+
+        // http://juick.com/last?page=2
+        // <a href="http://juick.com/last?page=2" rel="nofollow">juick.com</a>
+        msg = msg.replaceAll("((?<=\\s)|(?<=\\A))((?:ht|f)tps?://(?:www\\.)?([^\\/\\s\\n\\\"]+)/?[^\\s\\n\\\"]*)", "$1<a href=\"$2\" rel=\"nofollow\">$3</a>");
+
+        // #12345
+        // <a href="http://juick.com/12345">#12345</a>
+        msg = msg.replaceAll("((?<=\\s)|(?<=\\A)|(?<=\\p{Punct}))#(\\d+)((?=\\s)|(?=\\Z)|(?=\\))|(?=\\.)|(?=\\,))", "$1<a href=\"http://juick.com/$2\">#$2</a>$3");
+
+        // #12345/65
+        // <a href="http://juick.com/12345#65">#12345/65</a>
+        msg = msg.replaceAll("((?<=\\s)|(?<=\\A)|(?<=\\p{Punct}))#(\\d+)/(\\d+)((?=\\s)|(?=\\Z)|(?=\\p{Punct}))", "$1<a href=\"http://juick.com/$2#$3\">#$2/$3</a>$4");
+
+        // *bold*
+        // <b>bold</b>
+        msg = msg.replaceAll("((?<=\\s)|(?<=\\A)|(?<=\\p{Punct}))\\*([^\\*\\n<>]+)\\*((?=\\s)|(?=\\Z)|(?=\\p{Punct}))", "$1<b>$2</b>$3");
+
+        // /italic/
+        // <i>italic</i>
+        msg = msg.replaceAll("((?<=\\s)|(?<=\\A))/([^\\/\\n<>]+)/((?=\\s)|(?=\\Z)|(?=\\p{Punct}))", "$1<i>$2</i>$3");
+
+        // _underline_
+        // <span class="u">underline</span>
+        msg = msg.replaceAll("((?<=\\s)|(?<=\\A))_([^\\_\\n<>]+)_((?=\\s)|(?=\\Z)|(?=\\p{Punct}))", "$1<span class=\"u\">$2</span>$3");
+
+        // /12
+        // <a href="#12">/12</a>
+        msg = msg.replaceAll("((?<=\\s)|(?<=\\A))\\/(\\d+)((?=\\s)|(?=\\Z)|(?=\\p{Punct}))", "$1<a href=\"#$2\">/$2</a>$3");
+
+        // @username@jabber.org
+        // <a href="http://juick.com/username@jabber.org/">@username@jabber.org</a>
+        msg = msg.replaceAll("((?<=\\s)|(?<=\\A))@([\\w\\-\\.]+@[\\w\\-\\.]+)((?=\\s)|(?=\\Z)|(?=\\p{Punct}))", "$1<a href=\"http://juick.com/$2/\">@$2</a>$3");
+
+        // @username
+        // <a href="http://juick.com/username/">@username</a>
+        msg = msg.replaceAll("((?<=\\s)|(?<=\\A))@([\\w\\-]{2,16})((?=\\s)|(?=\\Z)|(?=\\p{Punct}))", "$1<a href=\"http://juick.com/$2/\">@$2</a>$3");
+
+        // (http://juick.com/last?page=2)
+        // (<a href="http://juick.com/last?page=2" rel="nofollow">juick.com</a>)
+        Matcher m = regexLinks2.matcher(msg);
+        StringBuffer sb = new StringBuffer();
+        while (m.find()) {
+            String url = m.group(3).replace(" ", "%20").replaceAll("\\s+", "");
+            m.appendReplacement(sb, "$1$2<a href=\"" + url + "\" rel=\"nofollow\">$4</a>$5");
+        }
+        m.appendTail(sb);
+        msg = sb.toString();
+
+        // > citate
+        msg = msg.replaceAll("(?:(?<=\\n)|(?<=\\A))&gt; *(.*)?(\\n|(?=\\Z))", "<blockquote>$1</blockquote>");
+        msg = msg.replaceAll("</blockquote><blockquote>", "\n");
+
+        msg = msg.replaceAll("\n", "<br/>\n");
+        return msg;
+    }
+}
diff --git a/src/java/com/juick/rss/Utils.java b/src/java/com/juick/rss/Utils.java
new file mode 100644
index 00000000..a6842fe1
--- /dev/null
+++ b/src/java/com/juick/rss/Utils.java
@@ -0,0 +1,74 @@
+/*
+ * Juick
+ * Copyright (C) 2008-2011, 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 <http://www.gnu.org/licenses/>.
+ */
+package com.juick.rss;
+
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
+import java.util.ArrayList;
+
+/**
+ *
+ * @author ugnich
+ */
+public class Utils {
+
+    public static void finishSQL(ResultSet rs, Statement stmt) {
+        if (rs != null) {
+            try {
+                rs.close();
+            } catch (SQLException e) {
+            }
+        }
+        if (stmt != null) {
+            try {
+                stmt.close();
+            } catch (SQLException e) {
+            }
+        }
+    }
+
+    public static String convertArray2String(ArrayList<Integer> mids) {
+        String q = "";
+        for (int i = 0; i < mids.size(); i++) {
+            if (i > 0) {
+                q += ",";
+            }
+            q += mids.get(i);
+        }
+        return q;
+    }
+
+    public static String encodeHTML(String str) {
+        String ret = str;
+        ret = ret.replaceAll("<", "&lt;");
+        ret = ret.replaceAll(">", "&gt;");
+        return str;
+    }
+
+    public static int parseInt(String str, int def) {
+        int ret = def;
+        if (str != null) {
+            try {
+                ret = Integer.parseInt(str);
+            } catch (Exception e) {
+            }
+        }
+        return ret;
+    }
+}