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
105
106
107
108
109
|
package com.juick.api;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.juick.Message;
import com.juick.User;
import com.juick.server.MessagesQueries;
import com.juick.server.PushQueries;
import com.juick.server.SubscriptionsQueries;
import com.juick.server.UserQueries;
import com.juick.server.helpers.TokensList;
import org.apache.commons.lang3.math.NumberUtils;
import org.springframework.jdbc.core.JdbcTemplate;
import spark.utils.IOUtils;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.List;
import java.util.stream.Collectors;
/**
* Created by vitalyster on 24.10.2016.
*/
public class Notifications {
JdbcTemplate jdbc;
ObjectMapper mapper = new ObjectMapper();
public Notifications(JdbcTemplate jdbc) {
this.jdbc = jdbc;
}
public void doGet(HttpServletRequest request, HttpServletResponse response, int vuid) throws IOException {
User visitor = UserQueries.getUserByUID(jdbc, vuid).orElse(new User());
if ((visitor.getUid() == 0) || !(visitor.getName().equals("juick"))) {
response.sendError(HttpServletResponse.SC_FORBIDDEN);
return;
}
String type = request.getParameter("type");
int uid = NumberUtils.toInt(request.getParameter("uid"), 0);
int mid = NumberUtils.toInt(request.getParameter("mid"), 0);
if (uid > 0) {
switch (type) {
case "gcm":
List<String> tokens = PushQueries.getAndroidRegID(jdbc, uid);
Main.replyJSON(request, response, mapper.writeValueAsString(tokens));
break;
case "apns":
List<String> apnsTokens = PushQueries.getAPNSToken(jdbc, uid);
Main.replyJSON(request, response, mapper.writeValueAsString(apnsTokens));
break;
case "mpns":
List<String> mpnsTokens = PushQueries.getWinPhoneURL(jdbc, uid);
Main.replyJSON(request, response, mapper.writeValueAsString(mpnsTokens));
break;
default:
response.sendError(HttpServletResponse.SC_BAD_REQUEST);
}
} else {
if (mid > 0) {
Message msg = MessagesQueries.getMessage(jdbc, mid);
if (msg != null) {
List<User> users;
if (msg.getRid() > 0) {
users = SubscriptionsQueries.getUsersSubscribedToComments(jdbc, mid, msg.getUser().getUid());
} else {
users = SubscriptionsQueries.getSubscribedUsers(jdbc, msg.getUser().getUid(), mid);
}
List<Integer> uids = users.stream().map(User::getUid).collect(Collectors.toList());
switch (type) {
case "gcm":
List<String> tokens = PushQueries.getAndroidTokens(jdbc, uids);
Main.replyJSON(request, response, mapper.writeValueAsString(tokens));
break;
case "apns":
List<String> apnsTokens = PushQueries.getAPNSTokens(jdbc, uids);
Main.replyJSON(request, response, mapper.writeValueAsString(apnsTokens));
break;
case "mpns":
List<String> mpnsTokens = PushQueries.getWindowsTokens(jdbc, uids);
Main.replyJSON(request, response, mapper.writeValueAsString(mpnsTokens));
break;
default:
response.sendError(HttpServletResponse.SC_BAD_REQUEST);
}
}
} else {
response.sendError(HttpServletResponse.SC_BAD_REQUEST);
}
}
}
public void doDelete(HttpServletRequest request, HttpServletResponse response, int vuid) throws IOException {
User visitor = UserQueries.getUserByUID(jdbc, vuid).orElse(new User());
if ((visitor.getUid() == 0) || !(visitor.getName().equals("juick"))) {
response.sendError(HttpServletResponse.SC_FORBIDDEN);
return;
}
ObjectMapper mapper = new ObjectMapper();
mapper.setSerializationInclusion(JsonInclude.Include.NON_EMPTY);
mapper.setSerializationInclusion(JsonInclude.Include.NON_NULL);
mapper.setSerializationInclusion(JsonInclude.Include.NON_DEFAULT);
TokensList list = mapper.readValue(IOUtils.toString(request.getInputStream()), TokensList.class);
list.getTokens().forEach(t -> PushQueries.deleteAPNSToken(jdbc, t));
}
}
|
