/*
* Copyright (C) 2008-2017, Juick
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see .
*/
package com.juick.api;
import com.juick.User;
import com.juick.server.component.MessageEvent;
import com.juick.service.MessagesService;
import com.juick.service.SubscriptionService;
import com.juick.service.TelegramService;
import com.juick.service.UserService;
import com.pengrad.telegrambot.Callback;
import com.pengrad.telegrambot.TelegramBot;
import com.pengrad.telegrambot.model.request.ParseMode;
import com.pengrad.telegrambot.request.SendMessage;
import com.pengrad.telegrambot.request.SetWebhook;
import com.pengrad.telegrambot.response.SendResponse;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.ApplicationListener;
import org.springframework.stereotype.Component;
import javax.annotation.Nonnull;
import javax.annotation.PostConstruct;
import javax.inject.Inject;
import java.io.IOException;
import java.util.List;
import static com.juick.formatters.PlainTextFormatter.formatPost;
import static com.juick.formatters.PlainTextFormatter.formatUrl;
/**
* Created by vt on 12/05/16.
*/
@Component
public class TelegramBotManager implements ApplicationListener {
private static final Logger logger = LoggerFactory.getLogger(TelegramBotManager.class);
private TelegramBot bot;
@Value("${telegram_token}")
private String telegramToken;
@Inject
private TelegramService telegramService;
@Inject
private MessagesService messagesService;
@Inject
private SubscriptionService subscriptionService;
@Inject
private UserService userService;
private static final String MSG_LINK = "🔗";
@PostConstruct
public void init() {
if (StringUtils.isBlank(telegramToken)) {
logger.info("telegram token is not set, exiting");
return;
}
bot = new TelegramBot(telegramToken);
try {
SetWebhook webhook = new SetWebhook().url("https://api.juick.com/tlgmbtwbhk");
if (!bot.execute(webhook).isOk()) {
logger.error("error setting webhook");
}
} catch (Exception e) {
logger.warn("couldn't initialize telegram bot", e);
}
}
@Override
public void onApplicationEvent(@Nonnull MessageEvent event) {
com.juick.Message jmsg = event.getMessage();
String msgUrl = formatUrl(jmsg);
if (jmsg.getMid() > 0 && jmsg.getRid() == 0) {
String msg = String.format("[%s](%s) %s", MSG_LINK, msgUrl, formatPost(jmsg, true));
List users = telegramService.getTelegramIdentifiers(subscriptionService.getSubscribedUsers(jmsg.getUser().getUid(), jmsg.getMid()));
List chats = telegramService.getChats();
// registered subscribed users
users.forEach(c -> telegramNotify(c, msg));
// anonymous
chats.stream().filter(u -> telegramService.getUser(u) == 0).forEach(c -> telegramNotify(c, msg));
} else {
// get quote
com.juick.Message msg = messagesService.getReply(jmsg.getMid(), jmsg.getRid());
String fmsg = String.format("[%s](%s) %s", MSG_LINK, msgUrl, formatPost(msg, true));
telegramService.getTelegramIdentifiers(
subscriptionService.getUsersSubscribedToComments(jmsg.getMid(), jmsg.getUser().getUid())
).forEach(c -> telegramNotify(c, fmsg));
}
}
public void telegramNotify(Long chatId, String msg) {
telegramNotify(chatId, msg, 0);
}
public void telegramNotify(Long chatId, String msg, Integer replyTo) {
SendMessage telegramMessage = new SendMessage(chatId, msg);
if (replyTo > 0) {
telegramMessage.replyToMessageId(replyTo);
}
telegramMessage.parseMode(ParseMode.Markdown).disableWebPagePreview(true);
bot.execute(telegramMessage, new Callback() {
@Override
public void onResponse(SendMessage request, SendResponse response) {
if (!response.isOk()) {
if (response.errorCode() == 403) {
// remove from anonymous chat
telegramService.getChats().stream().filter(c -> c.equals(chatId)).findFirst().ifPresent(
d -> {
telegramService.deleteChat(d);
logger.info("deleted {} chat", d);
}
);
int userId = telegramService.getUser(chatId);
if (userId > 0) {
telegramService.deleteTelegramUser(chatId.intValue());
logger.info("deleted telegram id of @{}", userService.getUserByUID(userId)
.orElse(new User()).getName());
}
} else {
logger.warn("error response, isOk: {}, errorCode: {}, description: {}",
response.isOk(), response.errorCode(), response.description());
}
}
}
@Override
public void onFailure(SendMessage request, IOException e) {
logger.warn("telegram failure", e);
}
});
}
public void telegramSignupNotify(Long telegramId, String hash) {
bot.execute(new SendMessage(telegramId,
String.format("You are subscribed to all Juick messages. " +
"[Create or link](http://juick.com/signup?type=durov&hash=%s) " +
"an existing Juick account to get your subscriptions and ability to post messages", hash))
.parseMode(ParseMode.Markdown), new Callback() {
@Override
public void onResponse(SendMessage request, SendResponse response) {
logger.info("got response: {}", response.message());
}
@Override
public void onFailure(SendMessage request, IOException e) {
logger.warn("telegram failure", e);
}
});
}
public TelegramBot getBot() {
return bot;
}
}