1 files changed, 105 insertions, 0 deletions
diff --git a/juick-www/src/main/java/com/juick/www/controllers/NewMessage.java b/juick-www/src/main/java/com/juick/www/controllers/NewMessage.java
index b0a2b4af..a87c7136 100644
--- a/juick-www/src/main/java/com/juick/www/controllers/NewMessage.java
+++ b/juick-www/src/main/java/com/juick/www/controllers/NewMessage.java
@@ -62,6 +62,8 @@ public class NewMessage {
     @Inject
     private CrosspostService crosspostService;
     @Inject
+    private PMQueriesService pmQueriesService;
+    @Inject
     private WebApp webApp;
 
     private static final Logger logger = LoggerFactory.getLogger(NewMessage.class);
@@ -332,4 +334,107 @@ public class NewMessage {
             throw new HttpBadRequestException();
         }
     }
+    @PostMapping("/pm/send")
+    public String doPostPM(@RequestParam(name = "uname", required = false) String unameParam,
+                           @RequestParam String body) throws IOException {
+        com.juick.User visitor = UserUtils.getCurrentUser();
+        if (visitor.getUid() == 0 || visitor.isBanned()) {
+            throw new HttpForbiddenException();
+        }
+        String uname = unameParam;
+        if (uname.startsWith("@")) {
+            uname = uname.substring(1);
+        }
+        int uid = 0;
+        if (WebUtils.isUserName(uname)) {
+            uid = userService.getUIDbyName(uname);
+        }
+
+        if (uid == 0 || body.length() > 10240) {
+            throw new HttpBadRequestException();
+        }
+
+        if (userService.isInBLAny(uid, visitor.getUid())) {
+            throw new HttpForbiddenException();
+        }
+
+        if (pmQueriesService.createPM(visitor.getUid(), uid, body)) {
+            if (webApp.getXmpp() != null) {
+                Message msg = new Message();
+                msg.setFrom(Jid.of("juick@juick.com"));
+                msg.setTo(Jid.of(String.format("%d@push.juick.com", uid)));
+                com.juick.Message jmsg = new com.juick.Message();
+                jmsg.setUser(visitor);
+                jmsg.setText(body);
+                msg.addExtension(jmsg);
+                webApp.getXmpp().send(msg);
+
+                msg.setTo(Jid.of(String.format("%d@ws.juick.com", uid)));
+                webApp.getXmpp().send(msg);
+
+                List<String> jids = userService.getJIDsbyUID(uid);
+                for (String jid : jids) {
+                    Message mm = new Message();
+                    mm.setTo(Jid.of(jid));
+                    mm.setType(Message.Type.CHAT);
+                    if (pmQueriesService.havePMinRoster(visitor.getUid(), jid)) {
+                        mm.setFrom(Jid.of(jmsg.getUser().getName(), "juick.com", "Juick"));
+                        mm.setBody(body);
+                    } else {
+                        mm.setFrom(Jid.of("juick", "juick.com", "Juick"));
+                        mm.setBody("Private message from @" + jmsg.getUser().getName() + ":\n" + body);
+                    }
+                    webApp.getXmpp().send(mm);
+                }
+            } else {
+                logger.warn("XMPP unavailable");
+            }
+            return "redirect:/pm/sent";
+        } else {
+            throw new HttpBadRequestException();
+        }
+    }
+    @PostMapping("/post2")
+    public String doPostMessage(@RequestParam(name = "body") String bodyParam,
+                                @RequestParam(required = false) String img,
+                                @RequestParam(required = false) String referer,
+                                @RequestParam(required = false) MultipartFile attach) throws IOException {
+
+        com.juick.User visitor = UserUtils.getCurrentUser();
+        if (visitor.getUid() == 0 || visitor.isBanned()) {
+            throw new HttpForbiddenException();
+        }
+        String body = bodyParam.replace("\r", StringUtils.EMPTY);
+
+        String attachmentFName = HttpUtils.receiveMultiPartFile(attach, webApp.getTmpDir());
+
+        if (StringUtils.isBlank(attachmentFName) && img != null && img.length() > 10) {
+            try {
+                URL imgUrl = new URL(img);
+                attachmentFName = HttpUtils.downloadImage(imgUrl, webApp.getTmpDir());
+            } catch (Exception e) {
+                logger.error("DOWNLOAD ERROR", e);
+                throw new HttpBadRequestException();
+            }
+        }
+        Message msg = new Message();
+        msg.setType(Message.Type.CHAT);
+        msg.setFrom(Jid.of(String.valueOf(visitor.getUid()), "uid.juick.com", "perl"));
+        msg.setTo(Jid.of("juick@juick.com/Juick"));
+        msg.setBody(body);
+        try {
+            if (StringUtils.isNotEmpty(attachmentFName)) {
+                String attachmentUrl = String.format("juick://%s", attachmentFName);
+                msg.addExtension(new OobX(new URI(attachmentUrl), "!!!!Juick!!"));
+            }
+            webApp.getXmpp().sendMessage(msg);
+        } catch (URISyntaxException e1) {
+            logger.warn("attachment error", e1);
+        }
+        if (StringUtils.isBlank(referer) || referer.substring(0, 21).equals("http://juick.com/post")
+                || referer.substring(0, 22).equals("https://juick.com/post")) {
+            return "redirect:/?show=my";
+        }
+        return "redirect:" + referer;
+    }
 }