package com.juick.api.controllers; import com.juick.User; import com.juick.api.ApiServer; import com.juick.api.util.HttpBadRequestException; import com.juick.api.util.HttpForbiddenException; import com.juick.service.PMQueriesService; import com.juick.service.UserService; import com.juick.util.UserUtils; import org.springframework.http.MediaType; import org.springframework.stereotype.Controller; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestMethod; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.ResponseBody; import rocks.xmpp.addr.Jid; import rocks.xmpp.core.stanza.model.Message; import javax.inject.Inject; import java.security.Principal; import java.util.List; /** * * @author ugnich */ @Controller @ResponseBody public class PM { @Inject UserService userService; @Inject PMQueriesService pmQueriesService; @Inject ApiServer apiServer; @RequestMapping(value = "/pm", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_UTF8_VALUE) public List doGetPM( Principal principal, @RequestParam(required = false) String uname) { String name = UserUtils.getUsername(principal, null); User visitor = userService.getUserByName(name); int vuid = visitor.getUid(); if (vuid == 0) { throw new HttpForbiddenException(); } int uid = 0; if (uname != null && uname.matches("^[a-zA-Z0-9\\-]{2,16}$")) { uid = userService.getUIDbyName(uname); } if (uid == 0) { throw new HttpBadRequestException(); } return pmQueriesService.getPMMessages(vuid, uid); } @RequestMapping(value = "/pm", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE) public com.juick.Message doPostPM( Principal principal, @RequestParam String uname, @RequestParam String body) { String name = UserUtils.getUsername(principal, null); User visitor = userService.getUserByName(name); int vuid = visitor.getUid(); if (vuid == 0) { throw new HttpForbiddenException(); } int uid = 0; if (UserUtils.checkUserNameValid(uname)) { uid = userService.getUIDbyName(uname); } if (uid == 0 || body == null || body.length() < 1 || body.length() > 10240) { throw new HttpBadRequestException(); } if (userService.isInBLAny(uid, vuid)) { throw new HttpForbiddenException(); } if (pmQueriesService.createPM(vuid, uid, body)) { 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(userService.getUserByUID(vuid).get()); jmsg.setText(body); msg.addExtension(jmsg); apiServer.getXmpp().send(msg); msg.setTo(Jid.of(String.format("%d@ws.juick.com", uid))); apiServer.getXmpp().send(msg); List 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(vuid, 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); } apiServer.getXmpp().send(mm); } return jmsg; } throw new HttpBadRequestException(); } }