aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/com/juick/api
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java/com/juick/api')
-rw-r--r--src/main/java/com/juick/api/Main.java2
-rw-r--r--src/main/java/com/juick/api/TelegramBotHook.java17
2 files changed, 15 insertions, 4 deletions
diff --git a/src/main/java/com/juick/api/Main.java b/src/main/java/com/juick/api/Main.java
index 9b327e83..c632d8fc 100644
--- a/src/main/java/com/juick/api/Main.java
+++ b/src/main/java/com/juick/api/Main.java
@@ -81,7 +81,7 @@ public class Main extends HttpServlet implements Stream.StreamListener {
users = new Users(jdbc);
pm = new PM(jdbc);
others = new Others(jdbc);
- tgb = new TelegramBotHook(conf.getProperty("telegram_token", ""));
+ tgb = new TelegramBotHook(jdbc, conf.getProperty("telegram_token", ""));
setupXmppComponent(conf.getProperty("xmpp_host", "localhost"), Integer.parseInt(conf.getProperty("xmpp_port", "5347")),
conf.getProperty("xmpp_jid", "api.localhost"), conf.getProperty("xmpp_password"));
diff --git a/src/main/java/com/juick/api/TelegramBotHook.java b/src/main/java/com/juick/api/TelegramBotHook.java
index ed3e8942..d06473a3 100644
--- a/src/main/java/com/juick/api/TelegramBotHook.java
+++ b/src/main/java/com/juick/api/TelegramBotHook.java
@@ -10,6 +10,7 @@ import com.pengrad.telegrambot.model.Message;
import com.pengrad.telegrambot.request.SendMessage;
import com.pengrad.telegrambot.response.SendResponse;
import org.json.JSONObject;
+import org.springframework.jdbc.core.JdbcTemplate;
import javax.servlet.http.HttpServletRequest;
import java.io.BufferedReader;
@@ -27,12 +28,13 @@ public class TelegramBotHook {
private static final Logger logger = Logger.getLogger(TelegramBotHook.class.getName());
TelegramBot bot;
- List<Long> chats = new ArrayList<>();
WebSocket ws;
MessageSerializer ms = new MessageSerializer();
+ JdbcTemplate jdbc;
- public TelegramBotHook(String token) {
+ public TelegramBotHook(JdbcTemplate jdbc, String token) {
+ this.jdbc = jdbc;
bot = TelegramBotAdapter.build(token);
bot.setWebhook("https://api.juick.com/tlgmbtwbhk");
try {
@@ -62,6 +64,7 @@ public class TelegramBotHook {
}
sb.append("https://juick.com/").append(jmsg.getMID());
String msg = sb.toString();
+ List<Long> chats = getChats();
chats.stream().forEach(c -> {
bot.execute(new SendMessage(c, msg), new Callback<SendMessage, SendResponse>() {
@Override
@@ -83,12 +86,20 @@ public class TelegramBotHook {
}
}
+ List<Long> getChats() {
+ return jdbc.queryForList("SELECT chat_id FROM telegram_chats", Long.class);
+ }
+ void addChat(Long id) {
+ jdbc.update("INSERT IGNORE TO telegram_chats(chat_id) VALUES(?)", id);
+ }
+
public void doPost(HttpServletRequest request) throws IOException {
try (BufferedReader reader = request.getReader()) {
Message message = BotUtils.parseUpdate(reader).message();
logger.info("got telegram msg: " + message.toString());
+ List<Long> chats = getChats();
if (!chats.contains(message.chat().id())) {
- chats.add(message.chat().id());
+ addChat(message.chat().id());
logger.info("added chat with " + message.from().firstName());
bot.execute(new SendMessage(message.chat().id(), "Subscribed!"), new Callback<SendMessage, SendResponse>() {
@Override