/*
* 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.components;
import com.juick.Tag;
import com.juick.User;
import com.juick.components.s2s.StanzaListener;
import com.juick.formatters.PlainTextFormatter;
import com.juick.server.helpers.TagStats;
import com.juick.server.protocol.annotation.UserCommand;
import com.juick.service.*;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.math.NumberUtils;
import org.apache.commons.lang3.reflect.MethodUtils;
import org.ocpsoft.prettytime.PrettyTime;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import rocks.xmpp.addr.Jid;
import rocks.xmpp.core.stanza.model.*;
import rocks.xmpp.core.stanza.model.client.ClientMessage;
import rocks.xmpp.core.stanza.model.client.ClientPresence;
import rocks.xmpp.core.stanza.model.errors.Condition;
import javax.annotation.PostConstruct;
import javax.inject.Inject;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.*;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
/**
*
* @author ugnich
*/
@Component
public class JuickBot implements StanzaListener, AutoCloseable {
private static final Logger logger = LoggerFactory.getLogger(JuickBot.class);
@Inject
private XMPPServer xmpp;
@Inject
private XMPPConnection router;
@Value("${xmppbot_jid}")
private Jid jid;
private PrettyTime pt;
@Inject
private MessagesService messagesService;
@Inject
private UserService userService;
@Inject
private TagService tagService;
@Inject
private PMQueriesService pmQueriesService;
@Inject
private ShowQueriesService showQueriesService;
@Inject
private PrivacyQueriesService privacyQueriesService;
@Inject
private SubscriptionService subscriptionService;
@PostConstruct
public void init() {
xmpp.addStanzaListener(this);
broadcastPresence(null);
pt = new PrettyTime(new Locale("ru"));
}
public Jid getJid() {
return jid;
}
public boolean incomingPresence(Presence p) {
final String username = p.getTo().getLocal();
final boolean toJuick = username.equals(jid.getLocal());
if (p.getType() == null) {
Presence reply = new Presence();
reply.setFrom(p.getTo().asBareJid());
reply.setTo(p.getFrom().asBareJid());
reply.setType(Presence.Type.UNSUBSCRIBE);
xmpp.sendOut(ClientPresence.from(reply));
return true;
} else if (p.getType().equals(Presence.Type.PROBE)) {
int uid_to = 0;
if (!toJuick) {
uid_to = userService.getUIDbyName(username);
}
if (toJuick || uid_to > 0) {
Presence reply = new Presence();
reply.setFrom(p.getTo().withResource(jid.getResource()));
reply.setTo(p.getFrom());
reply.setPriority((byte)10);
if (!userService.getActiveJIDs().contains(p.getFrom().asBareJid().toEscapedString())) {
reply.setStatus("Send ON to enable notifications");
}
xmpp.sendOut(ClientPresence.from(reply));
} else {
Presence reply = new Presence();
reply.setFrom(p.getTo());
reply.setTo(p.getFrom());
reply.setType(Presence.Type.ERROR);
reply.setId(p.getId());
reply.setError(new StanzaError(StanzaError.Type.CANCEL, Condition.ITEM_NOT_FOUND));
xmpp.sendOut(ClientPresence.from(reply));
return true;
}
return true;
} else if (p.getType().equals(Presence.Type.SUBSCRIBE)) {
boolean canSubscribe = false;
if (toJuick) {
canSubscribe = true;
} else {
int uid_to = userService.getUIDbyName(username);
if (uid_to > 0) {
pmQueriesService.addPMinRoster(uid_to, p.getFrom().asBareJid().toEscapedString());
canSubscribe = true;
}
}
if (canSubscribe) {
Presence reply = new Presence();
reply.setFrom(p.getTo());
reply.setTo(p.getFrom());
reply.setType(Presence.Type.SUBSCRIBED);
xmpp.sendOut(ClientPresence.from(reply));
reply.setFrom(reply.getFrom().withResource(jid.getResource()));
reply.setPriority((byte) 10);
reply.setType(null);
xmpp.sendOut(ClientPresence.from(reply));
return true;
} else {
Presence reply = new Presence();
reply.setFrom(p.getTo());
reply.setTo(p.getFrom());
reply.setType(Presence.Type.ERROR);
reply.setId(p.getId());
reply.setError(new StanzaError(StanzaError.Type.CANCEL, Condition.ITEM_NOT_FOUND));
xmpp.sendOut(ClientPresence.from(reply));
return true;
}
} else if (p.getType().equals(Presence.Type.UNSUBSCRIBE)) {
if (!toJuick) {
int uid_to = userService.getUIDbyName(username);
if (uid_to > 0) {
pmQueriesService.removePMinRoster(uid_to, p.getFrom().asBareJid().toEscapedString());
}
}
Presence reply = new Presence();
reply.setFrom(p.getTo());
reply.setTo(p.getFrom());
reply.setType(Presence.Type.UNSUBSCRIBED);
xmpp.sendOut(ClientPresence.from(reply));
}
return false;
}
public boolean incomingMessage(Message msg) {
if (msg.getType() != null && msg.getType().equals(Message.Type.ERROR)) {
StanzaError error = msg.getError();
if (error != null && error.getCondition().equals(Condition.RESOURCE_CONSTRAINT)) {
// offline query is full, deactivating this jid
if (userService.setActiveStatusForJID(msg.getFrom().toEscapedString(), UserService.ActiveStatus.Inactive)) {
logger.info("{} is inactive now", msg.getFrom());
return true;
}
}
return false;
}
if (StringUtils.isBlank(msg.getBody())) {
return false;
}
String username = msg.getTo().getLocal();
User user_from;
String signuphash = StringUtils.EMPTY;
user_from = userService.getUserByJID(msg.getFrom().asBareJid().toEscapedString());
if (user_from == null) {
signuphash = userService.getSignUpHashByJID(msg.getFrom().asBareJid().toEscapedString());
}
if (user_from == null) {
Message reply = new Message();
reply.setFrom(msg.getTo());
reply.setTo(msg.getFrom());
reply.setType(Message.Type.CHAT);
if (username.equals(jid.getLocal())) {
reply.setBody("Для того, чтобы начать пользоваться сервисом, пожалуйста пройдите быструю регистрацию: http://juick.com/signup?type=xmpp&hash=" + signuphash + "\nЕсли у вас уже есть учетная запись на Juick, вы сможете присоединить этот JabberID к ней.\n\nTo start using Juick, please sign up: http://juick.com/signup?type=xmpp&hash=" + signuphash + "\nIf you already have an account on Juick, you will be proposed to attach this JabberID to your existing account.");
} else {
reply.setBody("Внимание, системное сообщение!\nВаш JabberID не обнаружен в списке доверенных. Для того, чтобы отправить сообщение пользователю " + username + "@juick.com, пожалуйста зарегистрируйте свой JabberID в системе: http://juick.com/signup?type=xmpp&hash=" + signuphash + "\nЕсли у вас уже есть учетная запись на Juick, вы сможете присоединить этот JabberID к ней.\n\nWarning, system message!\nYour JabberID is not found in our server's white list. To send a message to " + username + "@juick.com, please sign up: http://juick.com/signup?type=xmpp&hash=" + signuphash + "\nIf you already have an account on Juick, you will be proposed to attach this JabberID to your existing account.");
}
xmpp.sendOut(ClientMessage.from(reply));
return true;
}
if (username.equals(jid.getLocal())) {
try {
return incomingMessageJuick(user_from, msg);
} catch (IllegalAccessException | InvocationTargetException | NoSuchMethodException e) {
return false;
}
}
int uid_to = userService.getUIDbyName(username);
if (uid_to == 0) {
Message reply = new Message();
reply.setFrom(msg.getTo());
reply.setTo(msg.getFrom());
reply.setType(Message.Type.ERROR);
reply.setId(msg.getId());
reply.setError(new StanzaError(StanzaError.Type.CANCEL, Condition.ITEM_NOT_FOUND));
xmpp.sendOut(ClientMessage.from(reply));
return true;
}
boolean success = false;
if (!userService.isInBLAny(uid_to, user_from.getUid())) {
success = pmQueriesService.createPM(user_from.getUid(), uid_to, msg.getBody());
}
if (success) {
Message m = new Message();
m.setFrom(jid.asBareJid());
m.setTo(Jid.of(Integer.toString(uid_to), "push.juick.com", null));
com.juick.Message jmsg = new com.juick.Message();
jmsg.setUser(user_from);
jmsg.setText(msg.getBody());
m.addExtension(jmsg);
router.sendStanza(m);
m.setTo(Jid.of(Integer.toString(uid_to), "ws.juick.com", null));
router.sendStanza(m);
List jids;
boolean inroster = false;
jids = userService.getJIDsbyUID(uid_to);
for (String userJid : jids) {
Message mm = new Message();
mm.setTo(Jid.of(userJid));
mm.setType(Message.Type.CHAT);
inroster = pmQueriesService.havePMinRoster(user_from.getUid(), userJid);
if (inroster) {
mm.setFrom(Jid.of(jmsg.getUser().getName(), "juick.com", "Juick"));
mm.setBody(msg.getBody());
} else {
mm.setFrom(jid);
mm.setBody("Private message from @" + jmsg.getUser().getName() + ":\n" + msg.getBody());
}
xmpp.sendOut(ClientMessage.from(mm));
}
} else {
Message reply = new Message();
reply.setFrom(msg.getTo());
reply.setTo(msg.getFrom());
reply.setType(Message.Type.ERROR);
reply.setId(msg.getId());
reply.setError(new StanzaError(StanzaError.Type.CANCEL, Condition.NOT_ALLOWED));
xmpp.sendOut(ClientMessage.from(reply));
}
return false;
}
public Optional processCommand(User user, Jid from, String input) throws InvocationTargetException,
IllegalAccessException, NoSuchMethodException {
Optional cmd = MethodUtils.getMethodsListWithAnnotation(getClass(), UserCommand.class).stream()
.filter(m -> Pattern.compile(m.getAnnotation(UserCommand.class).pattern(),
m.getAnnotation(UserCommand.class).patternFlags()).matcher(input).matches())
.findFirst();
if (cmd.isPresent()) {
Matcher matcher = Pattern.compile(cmd.get().getAnnotation(UserCommand.class).pattern(),
cmd.get().getAnnotation(UserCommand.class).patternFlags()).matcher(input);
List groups = new ArrayList<>();
while (matcher.find()) {
for (int i = 1; i <= matcher.groupCount(); i++) {
groups.add(matcher.group(i));
}
}
return Optional.of((String) getClass().getMethod(cmd.get().getName(), User.class, Jid.class, String[].class)
.invoke(this, user, from, groups.toArray(new String[groups.size()])));
}
return Optional.empty();
}
public boolean incomingMessageJuick(User user_from, Message msg) throws IllegalAccessException, NoSuchMethodException, InvocationTargetException {
String command = msg.getBody().trim();
int commandlen = command.length();
// COMPATIBILITY
if (commandlen > 7 && command.substring(0, 3).equalsIgnoreCase("PM ")) {
command = command.substring(3).trim();
}
Optional result = processCommand(user_from, msg.getFrom(), command);
result.ifPresent(r -> sendReply(msg.getFrom(), r));
return result.isPresent();
}
@UserCommand(pattern = "^ping$", patternFlags = Pattern.CASE_INSENSITIVE,
help = "PING - returns you a PONG")
public String commandPing(User user, Jid from, String[] input) {
Presence p = new Presence(from);
p.setFrom(jid);
p.setPriority((byte) 10);
xmpp.sendOut(ClientPresence.from(p));
return "PONG";
}
@UserCommand(pattern = "^help$", patternFlags = Pattern.CASE_INSENSITIVE,
help = "HELP - returns this help message")
public String commandHelp(User user, Jid from, String[] input) {
return Arrays.stream(getClass().getDeclaredMethods())
.filter(m -> m.isAnnotationPresent(UserCommand.class))
.map(m -> m.getAnnotation(UserCommand.class).help())
.collect(Collectors.joining("\n"));
}
@UserCommand(pattern = "^login$", patternFlags = Pattern.CASE_INSENSITIVE,
help = "LOGIN - log in to Juick website")
public String commandLogin(User user_from, Jid from, String[] input) {
return "http://juick.com/login?hash=" + userService.getHashByUID(user_from.getUid());
}
@UserCommand(pattern = "^\\@(\\S+)\\s+([\\s\\S]+)$", help = "@username message - send PM to username")
public String commandPM(User user_from, Jid from, String... arguments) {
String user_to = arguments[0];
String body = arguments[1];
int ret = 0;
int uid_to = 0;
List jids_to = null;
boolean haveInRoster = false;
if (user_to.indexOf('@') > 0) {
uid_to = userService.getUIDbyJID(user_to);
} else {
uid_to = userService.getUIDbyName(user_to);
}
if (uid_to > 0) {
if (!userService.isInBLAny(uid_to, user_from.getUid())) {
if (pmQueriesService.createPM(user_from.getUid(), uid_to, body)) {
jids_to = userService.getJIDsbyUID(uid_to);
ret = 200;
} else {
ret = 500;
}
} else {
ret = 403;
}
} else {
ret = 404;
}
if (ret == 200) {
Message msg = new Message();
msg.setFrom(jid.asBareJid());
msg.setTo(Jid.of(Integer.toString(uid_to), "push.juick.com", null));
com.juick.Message jmsg = new com.juick.Message();
jmsg.setUser(user_from);
jmsg.setText(body);
msg.addExtension(jmsg);
router.sendStanza(msg);
msg.setTo(Jid.of(Integer.toString(uid_to), "ws.juick.com", null));
router.sendStanza(msg);
for (String userJid : jids_to) {
Message mm = new Message();
mm.setTo(Jid.of(userJid));
mm.setType(Message.Type.CHAT);
haveInRoster = pmQueriesService.havePMinRoster(user_from.getUid(), userJid);
if (haveInRoster) {
mm.setFrom(Jid.of(user_from.getName(), "juick.com", "Juick"));
mm.setBody(body);
} else {
mm.setFrom(jid);
mm.setBody("Private message from @" + user_from.getName() + ":\n" + body);
}
xmpp.sendOut(ClientMessage.from(mm));
}
}
if (ret == 200) {
return "Private message sent";
} else {
return "Error " + ret;
}
}
@UserCommand(pattern = "^bl$", patternFlags = Pattern.CASE_INSENSITIVE,
help = "BL - Show your blacklist")
public String commandBLShow(User user_from, Jid from, String... arguments) {
List blusers = userService.getUserBLUsers(user_from.getUid());
List bltags = tagService.getUserBLTags(user_from.getUid());
String txt = StringUtils.EMPTY;
if (bltags.size() > 0) {
for (String bltag : bltags) {
txt += "*" + bltag + "\n";
}
if (blusers.size() > 0) {
txt += "\n";
}
}
if (blusers.size() > 0) {
for (User bluser : blusers) {
txt += "@" + bluser.getName() + "\n";
}
}
if (txt.isEmpty()) {
txt = "You don't have any users or tags in your blacklist.";
}
return txt;
}
@UserCommand(pattern = "^#\\+$", help = "#+ - Show last Juick messages")
public String commandLast(User user_from, Jid from, String... arguments) {
return "Last messages:\n"
+ printMessages(messagesService.getAll(user_from.getUid(), 0), true);
}
@UserCommand(pattern = "@", help = "@ - Show recommendations and popular personal blogs")
public String commandUsers(User user_from, Jid from, String... arguments) {
StringBuilder msg = new StringBuilder();
msg.append("Recommended blogs");
List recommendedUsers = showQueriesService.getRecommendedUsers(user_from);
if (recommendedUsers.size() > 0) {
for (String user : recommendedUsers) {
msg.append("\n@").append(user);
}
} else {
msg.append("\nNo recommendations now. Subscribe to more blogs. ;)");
}
msg.append("\n\nTop 10 personal blogs:");
List topUsers = showQueriesService.getTopUsers();
if (topUsers.size() > 0) {
for (String user : topUsers) {
msg.append("\n@").append(user);
}
} else {
msg.append("\nNo top users. Empty DB? ;)");
}
return msg.toString();
}
@UserCommand(pattern = "^bl\\s+@([^\\s\\n\\+]+)", patternFlags = Pattern.CASE_INSENSITIVE,
help = "BL @username - add @username to your blacklist")
public String blacklistUser(User user_from, Jid from, String... arguments) {
User blUser = userService.getUserByName(arguments[0]);
if (blUser != null) {
PrivacyQueriesService.PrivacyResult result = privacyQueriesService.blacklistUser(user_from, blUser);
if (result == PrivacyQueriesService.PrivacyResult.Added) {
return "User added to your blacklist";
} else {
return "User removed from your blacklist";
}
}
return "User not found";
}
@UserCommand(pattern = "^bl\\s\\*(\\S+)$", patternFlags = Pattern.CASE_INSENSITIVE,
help = "BL *tag - add *tag to your blacklist")
public String blacklistTag(User user_from, Jid from, String... arguments) {
User blUser = userService.getUserByName(arguments[0]);
if (blUser != null) {
Tag tag = tagService.getTag(arguments[0], false);
if (tag != null) {
PrivacyQueriesService.PrivacyResult result = privacyQueriesService.blacklistTag(user_from, tag);
if (result == PrivacyQueriesService.PrivacyResult.Added) {
return "Tag added to your blacklist";
} else {
return "Tag removed from your blacklist";
}
}
}
return "Tag not found";
}
@UserCommand(pattern = "\\*", help = "* - Show your tags")
public String commandTags(User currentUser, Jid from, String... args) {
List tags = tagService.getUserTagStats(currentUser.getUid());
String msg = "Your tags: (tag - messages)\n" +
tags.stream()
.map(t -> String.format("\n*%s - %d", t.getTag().getName(), t.getUsageCount())).collect(Collectors.joining());
return msg;
}
@UserCommand(pattern = "S", help = "S - Show your subscriptions")
public String commandSubscriptions(User currentUser, Jid from, String... args) {
List friends = userService.getUserFriends(currentUser.getUid());
List tags = subscriptionService.getSubscribedTags(currentUser);
String msg = friends.size() > 0 ? "You are subscribed to users:" + friends.stream().map(u -> "\n@" + u.getName())
.collect(Collectors.joining())
: "You are not subscribed to any user.";
msg += tags.size() > 0 ? "\nYou are subscribed to tags:" + tags.stream().map(t -> "\n*" + t)
.collect(Collectors.joining())
: "\nYou are not subscribed to any tag.";
return msg;
}
@UserCommand(pattern = "!", help = "! - Show your favorite messages")
public String commandFavorites(User currentUser, Jid from, String... args) {
List mids = messagesService.getUserRecommendations(currentUser.getUid(), 0);
if (mids.size() > 0) {
List messages = messagesService.getMessages(mids);
return "Favorite messages: \n" + String.join("\n", messages.stream()
.sorted(Collections.reverseOrder())
.map(PlainTextFormatter::formatPost)
.collect(Collectors.toList()));
}
return "No favorite messages, try to \"like\" something ;)";
}
@UserCommand(pattern = "^(s|u)\\s+\\*(\\S+)$", help = "S *tag - subscribe to tag" +
"\nU *tag - unsubscribe from tag", patternFlags = Pattern.CASE_INSENSITIVE)
public String commandSubscribeTag(User user, Jid from, String... args) {
boolean subscribe = args[0].equalsIgnoreCase("s");
Tag tag = tagService.getTag(args[1], true);
if (subscribe) {
if (subscriptionService.subscribeTag(user, tag)) {
return "Subscribed";
}
} else {
if (subscriptionService.unSubscribeTag(user, tag)) {
return "Unsubscribed from " + tag.getName();
}
return "You was not subscribed to " + tag.getName();
}
return "Error";
}
@UserCommand(pattern = "^(s|u)\\s+#(\\d+)$", help = "S #1234 - subscribe to comments",
patternFlags = Pattern.CASE_INSENSITIVE)
public String commandSubscribeMessage(User user, String... args) {
boolean subscribe = args[0].equalsIgnoreCase("s");
int mid = NumberUtils.toInt(args[1], 0);
if (messagesService.getMessage(mid) != null) {
if (subscribe) {
if (subscriptionService.subscribeMessage(mid, user.getUid())) {
return "Subscribed";
}
} else {
if (subscriptionService.unSubscribeMessage(mid, user.getUid())) {
return "Unsubscribed from #" + mid;
}
return "You was not subscribed to #" + mid;
}
}
return "Error";
}
@UserCommand(pattern = "^(on|off)$", patternFlags = Pattern.CASE_INSENSITIVE,
help = "ON/OFF - Enable/disable subscriptions delivery")
public String commandOnOff(User user, String[] input) {
UserService.ActiveStatus newStatus;
String retValUpdated;
if (input[0].toLowerCase().equals("on")) {
newStatus = UserService.ActiveStatus.Active;
retValUpdated = "Notifications are activated for " + user.getJid();
} else {
newStatus = UserService.ActiveStatus.Inactive;
retValUpdated = "Notifications are disabled for " + user.getJid();
}
if (userService.setActiveStatusForJID(user.getJid(), newStatus)) {
return retValUpdated;
} else {
return String.format("Subscriptions status for %s was not changed", user.getJid());
}
}
@UserCommand(pattern = "^\\@([^\\s\\n\\+]+)(\\+?)$",
help = "@username+ - Show user's info and last 10 messages (@username++ - second page, ..)")
public String commandUser(User user, String... arguments) {
User blogUser = userService.getUserByName(arguments[0]);
int page = arguments[1].length();
if (blogUser.getUid() > 0) {
List mids = messagesService.getUserBlog(blogUser.getUid(), 0, page);
List messages = messagesService.getMessages(mids);
return String.format("Last messages from @%s:\n%s", arguments[0],
String.join("\n", messages.stream()
.map(PlainTextFormatter::formatPost).collect(Collectors.toList())));
}
return "User not found";
}
void sendReply(Jid jidTo, String txt) {
Message reply = new Message();
reply.setFrom(jid);
reply.setTo(jidTo);
reply.setType(Message.Type.CHAT);
reply.setBody(txt);
xmpp.sendOut(ClientMessage.from(reply));
}
void sendNotification(Stanza stanza) {
xmpp.sendOut(stanza);
}
@Override
public void stanzaReceived(Stanza xmlValue) {
if (xmlValue instanceof Presence) {
Presence p = (Presence) xmlValue;
if (p.getType() == null || !p.getType().equals(Presence.Type.ERROR)) {
incomingPresence(p);
}
} else if (xmlValue instanceof Message) {
Message msg = (Message) xmlValue;
if (!incomingMessage(msg)) {
router.sendStanza(msg);
}
} else if (xmlValue instanceof IQ) {
IQ iq = (IQ) xmlValue;
router.sendStanza(iq);
}
}
String printMessages(List mids, boolean crop) {
return messagesService.getMessages(mids).stream()
.sorted(Collections.reverseOrder())
.map(PlainTextFormatter::formatPostSummary).collect(Collectors.joining("\n\n"));
}
void broadcastPresence(Presence.Type type) {
Presence presence = new Presence();
presence.setFrom(jid);
if (type != null) {
presence.setType(type);
}
userService.getActiveJIDs().forEach(j -> {
try {
presence.setTo(Jid.of(j));
xmpp.sendOut(ClientPresence.from(presence));
} catch (IllegalArgumentException ex) {
logger.warn("Invalid jid: {}", j, ex);
}
});
}
@Override
public void close() throws Exception {
broadcastPresence(Presence.Type.UNAVAILABLE);
}
}