diff options
| -rw-r--r-- | juick-api/src/main/java/com/juick/api/controllers/PM.java | 20 | ||||
| -rw-r--r-- | juick-www/src/main/java/com/juick/www/controllers/NewMessage.java | 19 |
2 files changed, 22 insertions, 17 deletions
diff --git a/juick-api/src/main/java/com/juick/api/controllers/PM.java b/juick-api/src/main/java/com/juick/api/controllers/PM.java index b72e0a0f..87858e8c 100644 --- a/juick-api/src/main/java/com/juick/api/controllers/PM.java +++ b/juick-api/src/main/java/com/juick/api/controllers/PM.java @@ -19,6 +19,7 @@ package com.juick.api.controllers; import com.juick.User; import com.juick.api.ApiServer; +import com.juick.server.helpers.AnonymousUser; import com.juick.server.util.HttpBadRequestException; import com.juick.server.util.HttpForbiddenException; import com.juick.service.PMQueriesService; @@ -77,33 +78,34 @@ public class PM { if (vuid == 0) { throw new HttpForbiddenException(); } - int uid = 0; + User userTo = AnonymousUser.INSTANCE; if (WebUtils.isUserName(uname)) { - uid = userService.getUIDbyName(uname); + userTo = userService.getUserByName(uname); } - if (uid == 0 || body == null || body.length() < 1 || body.length() > 10240) { + if (userTo.getUid() == 0 || body == null || body.length() < 1 || body.length() > 10240) { throw new HttpBadRequestException(); } - if (userService.isInBLAny(uid, vuid)) { + if (userService.isInBLAny(userTo.getUid(), vuid)) { throw new HttpForbiddenException(); } - if (pmQueriesService.createPM(vuid, uid, body)) { + if (pmQueriesService.createPM(vuid, userTo.getUid(), body)) { Message msg = new Message(); msg.setFrom(Jid.of("juick@juick.com")); - msg.setTo(Jid.of(String.format("%d@push.juick.com", uid))); + msg.setTo(Jid.of(String.format("%d@push.juick.com", userTo.getUid()))); com.juick.Message jmsg = new com.juick.Message(); - jmsg.setUser(userService.getUserByUID(vuid).get()); + jmsg.setUser(visitor); jmsg.setText(body); + jmsg.setTo(userTo); msg.addExtension(jmsg); apiServer.sendMessage(msg); - msg.setTo(Jid.of(String.format("%d@ws.juick.com", uid))); + msg.setTo(Jid.of(String.format("%d@ws.juick.com", userTo.getUid()))); apiServer.sendMessage(msg); - List<String> jids = userService.getJIDsbyUID(uid); + List<String> jids = userService.getJIDsbyUID(userTo.getUid()); for (String jid : jids) { Message mm = new Message(); mm.setTo(Jid.of(jid)); 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 5a73d983..c9a1d572 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 @@ -18,6 +18,8 @@ package com.juick.www.controllers; import com.juick.Status; import com.juick.Tag; +import com.juick.User; +import com.juick.server.helpers.AnonymousUser; import com.juick.server.util.*; import com.juick.service.*; import com.juick.www.WebApp; @@ -342,34 +344,35 @@ public class NewMessage { if (uname.startsWith("@")) { uname = uname.substring(1); } - int uid = 0; + User userTo = AnonymousUser.INSTANCE; if (WebUtils.isUserName(uname)) { - uid = userService.getUIDbyName(uname); + userTo = userService.getUserByName(uname); } - if (uid == 0 || body.length() > 10240) { + if (userTo.getUid() == 0 || body.length() > 10240) { throw new HttpBadRequestException(); } - if (userService.isInBLAny(uid, visitor.getUid())) { + if (userService.isInBLAny(userTo.getUid(), visitor.getUid())) { throw new HttpForbiddenException(); } - if (pmQueriesService.createPM(visitor.getUid(), uid, body)) { + if (pmQueriesService.createPM(visitor.getUid(), userTo.getUid(), 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))); + msg.setTo(Jid.of(String.format("%d@push.juick.com", userTo.getUid()))); com.juick.Message jmsg = new com.juick.Message(); jmsg.setUser(visitor); jmsg.setText(body); + jmsg.setTo(userTo); msg.addExtension(jmsg); webApp.getXmpp().send(msg); - msg.setTo(Jid.of(String.format("%d@ws.juick.com", uid))); + msg.setTo(Jid.of(String.format("%d@ws.juick.com", userTo.getUid()))); webApp.getXmpp().send(msg); - List<String> jids = userService.getJIDsbyUID(uid); + List<String> jids = userService.getJIDsbyUID(userTo.getUid()); for (String jid : jids) { Message mm = new Message(); mm.setTo(Jid.of(jid)); |