1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
package com.juick.api;
import com.juick.json.MessageSerializer;
import com.juick.server.PMQueries;
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 java.io.IOException;
import java.sql.Connection;
import java.util.ArrayList;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
/**
*
* @author ugnich
*/
public class PM {
Connection sql;
MessageSerializer messageSerializer = new MessageSerializer();
public PM(Connection sql) {
this.sql = sql;
}
public void doGetPM(HttpServletRequest request, HttpServletResponse response, int vuid) throws ServletException, IOException {
String uname = request.getParameter("uname");
int uid = 0;
if (uname != null && uname.matches("^[a-zA-Z0-9\\-]{2,16}$")) {
uid = UserQueries.getUIDbyName(sql, uname);
}
if (uid == 0) {
response.sendError(400);
return;
}
ArrayList<com.juick.Message> msgs = PMQueries.getPMMessages(sql, vuid, uid);
if (msgs != null && !msgs.isEmpty()) {
String json = messageSerializer.serializeList(msgs);
Main.replyJSON(request, response, json);
} else {
response.sendError(404);
}
}
public void doPostPM(HttpServletRequest request, HttpServletResponse response, Stream xmpp, int vuid) throws ServletException, IOException {
String uname = request.getParameter("uname");
int uid = 0;
if (UserQueries.checkUserNameValid(uname)) {
uid = UserQueries.getUIDbyName(sql, uname);
}
String body = request.getParameter("body");
if (uid == 0 || body == null || body.length() < 1 || body.length() > 10240) {
response.sendError(400);
return;
}
if (UserQueries.isInBLAny(sql, uid, vuid)) {
response.sendError(403);
return;
}
if (PMQueries.createPM(sql, vuid, uid, body)) {
Message msg = new Message();
msg.from = new JID("juick", "juick.com", null);
msg.to = new JID(Integer.toString(uid), "push.juick.com", null);
JuickMessage jmsg = new JuickMessage();
jmsg.User = UserQueries.getUserByUID(sql, vuid);
jmsg.Text = body;
msg.childs.add(jmsg);
xmpp.send(msg);
msg.to.Host = "ws.juick.com";
xmpp.send(msg);
Main.replyJSON(request, response, messageSerializer.serialize(jmsg).toString());
String jid = UserQueries.getJIDbyUID(sql, uid);
if (jid != null) {
Message mm = new Message();
mm.to = new JID(jid);
mm.type = Message.Type.chat;
if (PMQueries.havePMinRoster(sql, vuid, jid)) {
mm.from = new JID(jmsg.User.UName, "juick.com", "Juick");
mm.body = body;
} else {
mm.from = new JID("juick", "juick.com", "Juick");
mm.body = "Private message from @" + jmsg.User.UName + ":\n" + body;
}
xmpp.send(mm);
}
} else {
response.sendError(500);
}
}
}
|
