NewMessage complete rewrite

1 files changed, 209 insertions, 2 deletions

diff --git a/src/java/com/juick/http/www/NewMessage.java b/src/java/com/juick/http/www/NewMessage.java
index 895bfe91..79b087eb 100644
--- a/src/java/com/juick/http/www/NewMessage.java
+++ b/src/java/com/juick/http/www/NewMessage.java
@@ -18,7 +18,16 @@
 package com.juick.http.www;
 
 import com.juick.Tag;
+import com.juick.server.MessagesQueries;
+import com.juick.server.SubscriptionsQueries;
 import com.juick.server.TagQueries;
+import com.juick.server.UserQueries;
+import com.juick.xmpp.JID;
+import com.juick.xmpp.Message;
+import com.juick.xmpp.Stream;
+import com.juick.xmpp.extensions.JuickMessage;
+import com.juick.xmpp.extensions.Nickname;
+import com.juick.xmpp.extensions.XOOB;
 import java.io.IOException;
 import java.io.PrintWriter;
 import java.io.UnsupportedEncodingException;
@@ -49,7 +58,7 @@ public class NewMessage {
             }
 
             out.println("<section id=\"content\" class=\"pagetext\">");
-            out.println("<form action=\"/post\" method=\"post\" id=\"postmsg\" enctype=\"multipart/form-data\">");
+            out.println("<form action=\"/post2\" method=\"post\" id=\"postmsg\" enctype=\"multipart/form-data\">");
             out.println("<p style=\"text-align: left\"><b>Место: <span id=\"location\"></span></b> <span id=\"locationclear\">&mdash; <a href=\"#\" onclick=\"clearLocation()\">Отменить</a></span></p>");
             out.println("<p style=\"text-align: left\"><b>Фото:</b> <span id=\"attachmentfile\"><input type=\"file\" name=\"attach\"/> <i>(JPG, PNG, до 10Мб)</i></span></p>");
 
@@ -127,6 +136,204 @@ public class NewMessage {
         out.println("</p>");
     }
 
-    protected void doPostNewMessage(Connection sql, HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
+    public void doPostMessage(Connection sql, HttpServletRequest request, HttpServletResponse response, Stream xmpp, com.juick.User visitor) throws ServletException, IOException {
+        String body = request.getParameter("body");
+        if (body == null || body.length() < 1 || body.length() > 4096) {
+            response.sendError(400);
+            return;
+        }
+
+        String tagsStr = request.getParameter("tags");
+        if (tagsStr == null || tagsStr.isEmpty()) {
+            response.sendError(400);
+            return;
+        }
+        String tagsArr[] = tagsStr.split("[ \\,]");
+        for (int i = 0; i < tagsArr.length; i++) {
+            if (tagsArr[i].startsWith("*")) {
+                tagsArr[i] = tagsArr[i].substring(1);
+            }
+            if (tagsArr[i].length() > 64) {
+                tagsArr[i] = tagsArr[i].substring(0, 64);
+            }
+        }
+        ArrayList<com.juick.Tag> tags = TagQueries.getTags(sql, tagsArr, true);
+        if (tags.isEmpty()) {
+            response.sendError(400);
+            return;
+        }
+        while (tags.size() > 5) {
+            tags.remove(5);
+        }
+
+        String attachmentFName = null;
+        try {
+            attachmentFName = Utils.receiveMultiPartFile(request, "attach");
+        } catch (Exception e) {
+            response.sendError(400);
+            return;
+        }
+
+        String paramImg = request.getParameter("img");
+        if (attachmentFName == null && paramImg != null && paramImg.length() > 12 && paramImg.startsWith("http://") && !paramImg.equals("http://")) {
+            try {
+                attachmentFName = Utils.downloadImage(paramImg);
+            } catch (Exception e) {
+                System.out.println("DOWNLOAD ERROR: " + e.toString());
+                response.sendError(500);
+                return;
+            }
+        }
+
+        String attachmentType = attachmentFName != null ? attachmentFName.substring(attachmentFName.length() - 3) : null;
+        int mid = MessagesQueries.createMessage(sql, visitor.UID, body, attachmentType, tags);
+        SubscriptionsQueries.subscribeMessage(sql, mid, visitor.UID);
+
+        Message xmsg = new Message();
+        xmsg.from = new JID("juick", "juick.com", null);
+        xmsg.type = Message.Type.chat;
+        xmsg.thread = "juick-" + mid;
+
+        JuickMessage jmsg = new JuickMessage(MessagesQueries.getMessage(sql, mid));
+        xmsg.addChild(jmsg);
+
+        Nickname nick = new Nickname();
+        nick.Nickname = "@" + jmsg.User.UName;
+        xmsg.addChild(nick);
+
+        if (attachmentFName != null) {
+            String fname = mid + "." + attachmentType;
+            String attachmentURL = "http://i.juick.com/photos-1024/" + fname;
+
+            Runtime.getRuntime().exec("/var/www/juick.com/cgi/p-convert.sh /var/www/juick.com/i/tmp/" + attachmentFName + " " + fname);
+
+            body = attachmentURL + "\n" + body;
+            XOOB xoob = new XOOB();
+            xoob.URL = attachmentURL;
+            xmsg.addChild(xoob);
+        }
+
+        String tagsStr2 = "";
+        for (int i = 0; i < tagsArr.length; i++) {
+            tagsStr2 += " *" + tagsArr[i];
+        }
+        xmsg.body = "@" + jmsg.User.UName + ":" + tagsStr2 + "\n" + body + "\n\n#" + mid + " http://juick.com/" + mid;
+
+        xmsg.to = new JID("juick", "s2s.juick.com", null);
+        xmpp.send(xmsg);
+
+        xmsg.to.Host = "ws.juick.com";
+        xmpp.send(xmsg);
+
+        xmsg.to.Host = "push.juick.com";
+        xmpp.send(xmsg);
+
+        xmsg.to.Host = "crosspost.juick.com";
+        xmsg.to.Username = "twitter";
+        xmpp.send(xmsg);
+        xmsg.to.Username = "fb";
+        xmpp.send(xmsg);
+
+        xmsg.to.Host = "nologin.ru";
+        xmsg.to.Username = "jubo";
+        xmpp.send(xmsg);
+
+        Utils.sendTemporaryRedirect(response, "/" + visitor.UName + "/" + mid);
+    }
+
+    public void doPostComment(Connection sql, HttpServletRequest request, HttpServletResponse response, Stream xmpp, com.juick.User visitor) throws ServletException, IOException {
+        int mid = Utils.parseInt(request.getParameter("mid"), 0);
+        if (mid == 0) {
+            response.sendError(400);
+            return;
+        }
+        com.juick.Message msg = MessagesQueries.getMessage(sql, mid);
+        if (msg == null) {
+            response.sendError(404);
+            return;
+        }
+
+        int rid = Utils.parseInt(request.getParameter("rid"), 0);
+        com.juick.Message reply = null;
+        if (rid > 0) {
+            reply = MessagesQueries.getReply(sql, mid, rid);
+            if (reply == null) {
+                response.sendError(404);
+                return;
+            }
+        }
+
+        String body = request.getParameter("body");
+        if (body == null || body.length() < 1 || body.length() > 4096) {
+            response.sendError(400);
+            return;
+        }
+
+        if ((msg.ReadOnly && msg.User.UID != visitor.UID) || UserQueries.isInBLAny(sql, msg.User.UID, visitor.UID) || (reply != null && UserQueries.isInBLAny(sql, reply.User.UID, visitor.UID))) {
+            response.sendError(403);
+            return;
+        }
+
+        String attachmentFName = null;
+        try {
+            attachmentFName = Utils.receiveMultiPartFile(request, "attach");
+        } catch (Exception e) {
+            response.sendError(400);
+            return;
+        }
+
+        String paramImg = request.getParameter("img");
+        if (attachmentFName == null && paramImg != null && paramImg.length() > 12 && paramImg.startsWith("http://") && !paramImg.equals("http://")) {
+            try {
+                attachmentFName = Utils.downloadImage(paramImg);
+            } catch (Exception e) {
+                System.out.println("DOWNLOAD ERROR: " + e.toString());
+                response.sendError(500);
+                return;
+            }
+        }
+
+        String attachmentType = attachmentFName != null ? attachmentFName.substring(attachmentFName.length() - 3) : null;
+        int ridnew = MessagesQueries.createReply(sql, mid, rid, visitor.UID, body, attachmentType);
+        SubscriptionsQueries.subscribeMessage(sql, mid, visitor.UID);
+
+        Message xmsg = new Message();
+        xmsg.from = new JID("juick", "juick.com", null);
+        xmsg.type = Message.Type.chat;
+        xmsg.thread = "juick-" + mid;
+
+        JuickMessage jmsg = new JuickMessage(MessagesQueries.getReply(sql, mid, ridnew));
+        xmsg.addChild(jmsg);
+
+        String quote = reply != null ? reply.Text : msg.Text;
+        if (quote.length() >= 50) {
+            quote = quote.substring(0, 47) + "...";
+        }
+
+        Nickname nick = new Nickname();
+        nick.Nickname = "@" + jmsg.User.UName;
+        xmsg.addChild(nick);
+
+        if (attachmentFName != null) {
+            String fname = mid + "-" + ridnew + "." + attachmentType;
+            String attachmentURL = "http://i.juick.com/photos-1024/" + fname;
+
+            Runtime.getRuntime().exec("/var/www/juick.com/cgi/p-convert.sh /var/www/juick.com/i/tmp/" + attachmentFName + " " + fname);
+
+            body = attachmentURL + "\n" + body;
+            XOOB xoob = new XOOB();
+            xoob.URL = attachmentURL;
+            xmsg.addChild(xoob);
+        }
+
+        xmsg.body = "Reply by @" + jmsg.User.UName + ":\n>" + quote + "\n" + body + "\n\n#" + mid + "/" + ridnew + " http://juick.com/" + mid + "#" + ridnew;
+
+        xmsg.to = new JID("juick", "s2s.juick.com", null);
+        xmpp.send(xmsg);
+
+        xmsg.to.Host = "ws.juick.com";
+        xmpp.send(xmsg);
+
+        Utils.sendTemporaryRedirect(response, "/" + msg.User.UName + "/" + mid + "#" + ridnew);
     }
 }