path: root/juick-server/src/main/java/com/juick/server/XMPPBot.java

/*
 * 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 <http://www.gnu.org/licenses/>.
 */

package com.juick.server;

import com.juick.Tag;
import com.juick.User;
import com.juick.server.component.LikeEvent;
import com.juick.server.component.MessageEvent;
import com.juick.server.component.SubscribeEvent;
import com.juick.server.util.TagUtils;
import com.juick.server.xmpp.s2s.StanzaListener;
import com.juick.formatters.PlainTextFormatter;
import com.juick.server.helpers.TagStats;
import com.juick.server.xmpp.helpers.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.context.ApplicationEventPublisher;
import org.springframework.context.ApplicationListener;
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 XMPPBot implements StanzaListener, AutoCloseable {

    private static final Logger logger = LoggerFactory.getLogger(XMPPBot.class);

    @Inject
    private XMPPServer xmpp;
    @Inject
    private XMPPConnection router;
    @Value("${xmppbot_jid:juick@localhost}")
    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;
    @Inject
    private ApplicationEventPublisher applicationEventPublisher;

    @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.getFrom(), msg.getBody().trim());
            } 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());
            com.juick.Message jmsg = new com.juick.Message();
            jmsg.setUser(user_from);
            jmsg.setTo(userService.getUserByUID(uid_to).get());
            jmsg.setText(msg.getBody());
            m.addExtension(jmsg);

            List<String> 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<String> processCommand(User user, Jid from, String input) throws InvocationTargetException,
            IllegalAccessException, NoSuchMethodException {
        Optional<Method> 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<String> 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, Jid from, String command) throws IllegalAccessException, NoSuchMethodException, InvocationTargetException {
        int commandlen = command.length();

        // COMPATIBILITY
        if (commandlen > 7 && command.substring(0, 3).equalsIgnoreCase("PM ")) {
            command = command.substring(3).trim();
        }

        Optional<String> result = processCommand(user_from, from, command);
        if (result.isPresent()) {
            sendReply(from, result.get());
        } else {
            // new message
            List<Tag> tags = tagService.fromString(command, false);
            String body = command.substring(TagUtils.toString(tags).length());
            int mid = messagesService.createMessage(user_from.getUid(), body, null, tags);
            subscriptionService.subscribeMessage(mid, user_from.getUid());
            applicationEventPublisher.publishEvent(new MessageEvent(this, messagesService.getMessage(mid)));
        }
        return true;
    }

    public void processMessage(com.juick.Message msg) {
        if (msg.getRid() == 0 && msg.getMid() > 0) {
            String notify = "New message posted.\n#" + msg.getMid() + " https://juick.com/" + msg.getMid();
            userService.getJIDsbyUID(msg.getUser().getUid())
                    .forEach(jid -> sendReply(Jid.of(jid), notify));
        }
    }

    @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 body = arguments[1];
        int ret = 0;

        User user_to = userService.getUserByName(arguments[0]);
        List<String> jids_to = null;
        boolean haveInRoster = false;

        if (user_to.getUid() > 0) {
            if (!userService.isInBLAny(user_to.getUid(), user_from.getUid())) {
                if (pmQueriesService.createPM(user_from.getUid(), user_to.getUid(), body)) {
                    jids_to = userService.getJIDsbyUID(user_to.getUid());
                    ret = 200;
                } else {
                    ret = 500;
                }
            } else {
                ret = 403;
            }
        } else {
            ret = 404;
        }

        if (ret == 200) {
            Message msg = new Message();
            msg.setFrom(jid.asBareJid());
            com.juick.Message jmsg = new com.juick.Message();
            jmsg.setUser(user_from);
            jmsg.setTo(user_to);
            jmsg.setText(body);
            msg.addExtension(jmsg);

            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<User> blusers = userService.getUserBLUsers(user_from.getUid());
        List<String> 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<String> 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<String> 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<TagStats> 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<User> friends = userService.getUserFriends(currentUser.getUid());
        List<String> 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<Integer> mids = messagesService.getUserRecommendations(currentUser.getUid(), 0);
        if (mids.size() > 0) {
            return "Favorite messages: \n" + printMessages(mids, false);
        }
        return "No favorite messages, try to \"like\" something ;)";
    }
    @UserCommand(pattern = "^\\!\\s+#(\\d+)", help = "! #12345 - recommend message")
    public String commandRecommend(User user, Jid from, String... arguments) {
        int mid = NumberUtils.toInt(arguments[0], 0);
        if (mid > 0) {
            com.juick.Message msg = messagesService.getMessage(mid);
            if (msg != null) {
                if (msg.getUser() == user) {
                    return "You can't recommend your own messages.";
                }
                MessagesService.RecommendStatus status = messagesService.recommendMessage(mid, user.getUid());
                switch (status) {
                    case Added:
                        applicationEventPublisher.publishEvent(new LikeEvent(this, user, msg));
                        return "Message is added to your recommendations";
                    case Deleted:
                        return "Message deleted from your recommendations.";
                }
            }
            return "Message not found";
        }
        return "Message not found";
    }
    // TODO: target notification
    @UserCommand(pattern = "^(s|u)\\s+\\@(\\S+)$", help = "S @username - subscribe to user" +
            "\nU @username - unsubscribe from user", patternFlags = Pattern.CASE_INSENSITIVE)
    public String commandSubscribeUser(User user, Jid from, String... args) {
        boolean subscribe = args[0].equalsIgnoreCase("s");
        User toUser = userService.getUserByName(args[1]);
        if (subscribe) {
            if (subscriptionService.subscribeUser(user, toUser)) {
                // TODO: already subscribed case
                applicationEventPublisher.publishEvent(new SubscribeEvent(this, user, toUser));
                return "Subscribed to @" + toUser.getName();
            }
        } else {
            if (subscriptionService.unSubscribeUser(user, toUser)) {
                return "Unsubscribed from @" + toUser.getName();
            }
            return "You was not subscribed to @" + toUser.getName();
        }
        return "Error";
    }
    @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" +
            "\nU #1234 - unsubscribe from comments", patternFlags = Pattern.CASE_INSENSITIVE)
    public String commandSubscribeMessage(User user, Jid from, 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, Jid from, String[] input) {
        UserService.ActiveStatus newStatus;
        String retValUpdated;
        if (input[0].toLowerCase().equals("on")) {
            newStatus = UserService.ActiveStatus.Active;
            retValUpdated = "Notifications are activated for " + from.asBareJid().toEscapedString();
        } else {
            newStatus = UserService.ActiveStatus.Inactive;
            retValUpdated = "Notifications are disabled for " + from.asBareJid().toEscapedString();
        }

        if (userService.setActiveStatusForJID(from.asBareJid().toEscapedString(), newStatus)) {
            return retValUpdated;
        } else {
            return String.format("Subscriptions status for %s was not changed", from.toEscapedString());
        }
    }
    @UserCommand(pattern = "^\\@([^\\s\\n\\+]+)(\\+?)$",
            help = "@username+ - Show user's info and last 20 messages")
    public String commandUser(User user, Jid from, String... arguments) {
        User blogUser = userService.getUserByName(arguments[0]);
        int page = arguments[1].length();
        if (blogUser.getUid() > 0) {
            List<Integer> mids = messagesService.getUserBlog(blogUser.getUid(), 0, 0);
            return String.format("Last messages from @%s:\n%s", arguments[0],
                    printMessages(mids, false));
        }
        return "User not found";
    }
    @UserCommand(pattern = "^#(\\d+)(\\+?)$", help = "#1234 - Show message (#1234+ - message with replies)")
    public String commandShow(User user, Jid from, String... arguments) {
        boolean showReplies = arguments[1].length() > 0;
        int mid = NumberUtils.toInt(arguments[0], 0);
        if (mid == 0) {
            return "Error";
        }
        com.juick.Message msg = messagesService.getMessage(mid);
        if (msg != null) {
            if (showReplies) {
                List<com.juick.Message> replies = messagesService.getReplies(mid);
                replies.add(0, msg);
                return String.join("\n",
                        replies.stream().map(PlainTextFormatter::formatPostSummary).collect(Collectors.toList()));
            }
            return PlainTextFormatter.formatPost(msg);
        }
        return "Message not found";
    }
    @UserCommand(pattern = "^#(\\d+)\\/(\\d+)$", help = "#1234/5 - Show reply")
    public String commandShowReply(User user, Jid from, String... arguments) {
        int mid = NumberUtils.toInt(arguments[0], 0);
        int rid = NumberUtils.toInt(arguments[1], 0);
        com.juick.Message reply = messagesService.getReply(mid, rid);
        if (reply != null) {
            return PlainTextFormatter.formatPost(reply);
        }
        return "Reply not found";
    }
    @UserCommand(pattern = "^D #(\\d+)$", help = "D #1234 - Delete post", patternFlags = Pattern.CASE_INSENSITIVE)
    public String commandDeletePost(User user, Jid from, String... args) {
        int mid = Integer.valueOf(args[0]);
        if (messagesService.deleteMessage(user.getUid(), mid)) {
            return "Message deleted";
        }
        return "This is not your message";
    }
    @UserCommand(pattern = "^D #(\\d+)(\\.|\\-|\\/)(\\d+)$", help = "D #1234/5 - Delete comment", patternFlags = Pattern.CASE_INSENSITIVE)
    public String commandDeleteReply(User user, Jid from, String... args) {
        int mid = Integer.valueOf(args[0]);
        int rid = Integer.valueOf(args[2]);
        if (messagesService.deleteReply(user.getUid(), mid, rid)) {
            return "Reply deleted";
        } else {
            return "This is not your reply";
        }
    }
    @UserCommand(pattern = "^(D L|DL|D LAST)$", help = "D L - Delete last message", patternFlags = Pattern.CASE_INSENSITIVE)
    public String commandDeleteLast(User user, Jid from, String... args) {
        return "Temporarily unavailable";
    }
    @UserCommand(pattern = "^\\?\\s+\\@([a-zA-Z0-9\\-\\.\\@]+)\\s+([\\s\\S]+)$", help = "? @user string - search in user messages")
    public String commandSearch(User user, Jid from, String... args) {
        return "Temporarily unavailable";
    }
    @UserCommand(pattern = "^\\?\\s+([\\s\\S]+)$", help = "? string - search in all messages")
    public String commandSearchAll(User user, Jid from, String... args) {
        return "Temporarily unavailable";
    }
    @UserCommand(pattern = "^(#+)$", help = "# - Show last messages from your feed (## - second page, ...)")
    public String commandMyFeed(User user, Jid from, String... arguments) {
        // number of # is the page count
        int page = arguments[0].length() - 1;
        List<Integer> mids = messagesService.getMyFeed(user.getUid(), page, false);
        if (mids.size() > 0) {
            return "Your feed: \n" + printMessages(mids, true);
        }
        return "Your feed is empty";
    }
    @UserCommand(pattern = "^(#|\\.)(\\d+)((\\.|\\-|\\/)(\\d+))?\\s([\\s\\S]+)",
            help = "#1234 *tag *tag2 - edit tags\n#1234 text - reply to message")
    public String EditOrReply(User user, Jid from, String... args) {
        int mid = NumberUtils.toInt(args[1]);
        int rid = NumberUtils.toInt(args[4], 0);
        String txt = args[5];
        List<Tag> messageTags = tagService.fromString(txt, true);
        if (messageTags.size() > 0) {
            if (user.getUid() != messagesService.getMessageAuthor(mid).getUid()) {
                return "It is not your message";
            }
            tagService.updateTags(mid, messageTags);
            return "Tags are updated";
        } else {
            int newrid = messagesService.createReply(mid, rid, user.getUid(), txt, null);
            applicationEventPublisher.publishEvent(new MessageEvent(this, messagesService.getReply(mid, newrid)));
            return "Reply posted.\n#" + mid + "/" + newrid + " "
                    + "https://juick.com/" + mid + "#" + newrid;
        }
    }

    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<Integer> 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);
    }
}