server: XMPPBot -> CommandsManager

1 files changed, 309 insertions, 13 deletions

diff --git a/juick-server/src/main/java/com/juick/server/XMPPConnection.java b/juick-server/src/main/java/com/juick/server/XMPPConnection.java
index e4d2db04..eb459ee1 100644
--- a/juick-server/src/main/java/com/juick/server/XMPPConnection.java
+++ b/juick-server/src/main/java/com/juick/server/XMPPConnection.java
@@ -17,14 +17,17 @@
 
 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.PingEvent;
+import com.juick.server.component.SubscribeEvent;
+import com.juick.server.util.TagUtils;
 import com.juick.server.xmpp.s2s.BasicXmppSession;
 import com.juick.server.helpers.UserInfo;
-import com.juick.service.MessagesService;
-import com.juick.service.SubscriptionService;
-import com.juick.service.UserService;
+import com.juick.server.xmpp.s2s.StanzaListener;
+import com.juick.service.*;
 import com.juick.util.MessageUtils;
 import org.apache.commons.codec.digest.DigestUtils;
 import org.apache.commons.io.FilenameUtils;
@@ -38,10 +41,9 @@ import org.springframework.stereotype.Component;
 import rocks.xmpp.addr.Jid;
 import rocks.xmpp.core.XmppException;
 import rocks.xmpp.core.stanza.AbstractIQHandler;
-import rocks.xmpp.core.stanza.model.IQ;
-import rocks.xmpp.core.stanza.model.Message;
-import rocks.xmpp.core.stanza.model.Stanza;
+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 rocks.xmpp.extensions.caps.model.EntityCapabilities;
 import rocks.xmpp.extensions.component.accept.ExternalComponent;
@@ -56,12 +58,14 @@ import rocks.xmpp.extensions.version.model.SoftwareVersion;
 import rocks.xmpp.util.XmppUtils;
 
 import javax.annotation.PostConstruct;
+import javax.annotation.PreDestroy;
 import javax.inject.Inject;
 import javax.xml.bind.JAXBException;
 import javax.xml.stream.XMLStreamException;
 import javax.xml.stream.XMLStreamWriter;
 import java.io.IOException;
 import java.io.StringWriter;
+import java.lang.reflect.InvocationTargetException;
 import java.net.MalformedURLException;
 import java.net.URI;
 import java.net.URISyntaxException;
@@ -70,6 +74,7 @@ import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Optional;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.ExecutorService;
 
@@ -77,13 +82,15 @@ import java.util.concurrent.ExecutorService;
  * @author ugnich
  */
 @Component
-public class XMPPConnection implements AutoCloseable {
+public class XMPPConnection implements StanzaListener, NotificationListener {
 
     private static final Logger logger = LoggerFactory.getLogger(XMPPConnection.class);
 
     private ExternalComponent router;
     @Inject
     private XMPPServer xmpp;
+    @Inject
+    private CommandsManager commandsManager;
     @Value("${xmppbot_jid:juick@localhost}")
     private Jid jid;
 
@@ -105,6 +112,10 @@ public class XMPPConnection implements AutoCloseable {
     @Inject
     private SubscriptionService subscriptionService;
     @Inject
+    private PMQueriesService pmQueriesService;
+    @Inject
+    private TagService tagService;
+    @Inject
     private BasicXmppSession session;
     @Inject
     private ExecutorService service;
@@ -114,6 +125,8 @@ public class XMPPConnection implements AutoCloseable {
     @PostConstruct
     public void init() {
         logger.info("stream router start connecting to {}", componentPort);
+        xmpp.addStanzaListener(this);
+        broadcastPresence(null);
         router = ExternalComponent.create(componentName, password, session.getConfiguration(), "localhost",
                 componentPort);
         PingManager pingManager = router.getManager(PingManager.class);
@@ -166,7 +179,6 @@ public class XMPPConnection implements AutoCloseable {
                 com.juick.Message jmsg = message.getExtension(com.juick.Message.class);
                 if (jmsg != null) {
                     if (jid.getLocal().equals("recomm")) {
-                        sendJuickRecommendation(jmsg);
                         User fromUser = jmsg.getUser();
                         com.juick.Message msg = messagesService.getMessage(jmsg.getMid());
                         applicationEventPublisher.publishEvent(new LikeEvent(this, fromUser, msg));
@@ -377,13 +389,57 @@ public class XMPPConnection implements AutoCloseable {
         }
     }
 
-    public void sendJuickRecommendation(com.juick.Message recomm) {
+    @Override
+    public void processMessageEvent(MessageEvent event) {
+        com.juick.Message msg = event.getMessage();
+        boolean isPM = msg.getMid() == 0;
+        boolean isReply = msg.getRid() > 0;
+        if (isPM) {
+            userService.getJIDsbyUID(msg.getTo().getUid())
+                    .forEach(userJid -> {
+                        Message mm = new Message();
+                        mm.setTo(Jid.of(userJid));
+                        mm.setType(Message.Type.CHAT);
+                        boolean inroster = pmQueriesService.havePMinRoster(msg.getUser().getUid(), userJid);
+                        if (inroster) {
+                            mm.setFrom(Jid.of(msg.getUser().getName(), "juick.com", "Juick"));
+                            mm.setBody(msg.getText());
+                        } else {
+                            mm.setFrom(jid);
+                            mm.setBody("Private message from @" + msg.getUser().getName() + ":\n" + msg.getText());
+                        }
+                        xmpp.sendOut(ClientMessage.from(mm));
+                    });
+        } else if (!isReply) {
+            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));
+        }
+    }
+
+    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));
+    }
+
+    @Override
+    public void processSubscribeEvent(SubscribeEvent subscribeEvent) {
+
+    }
+
+    @Override
+    public void processLikeEvent(LikeEvent likeEvent) {
         List<User> users;
-        com.juick.Message jmsg = messagesService.getMessage(recomm.getMid());
-        users = subscriptionService.getUsersSubscribedToUserRecommendations(recomm.getUser().getUid(),
-                recomm.getMid(), jmsg.getUser().getUid());
+        com.juick.Message jmsg = likeEvent.getMessage();
+        User liker = likeEvent.getUser();
+        users = subscriptionService.getUsersSubscribedToUserRecommendations(liker.getUid(),
+                jmsg.getMid(), jmsg.getUser().getUid());
 
-        String txt = "Recommended by @" + recomm.getUser().getName() + ":\n";
+        String txt = "Recommended by @" + liker.getName() + ":\n";
         txt += "@" + jmsg.getUser().getName() + ":" + MessageUtils.getTagsString(jmsg) + "\n";
         String attachmentUrl = MessageUtils.attachmentUrl(jmsg);
         if (StringUtils.isNotEmpty(attachmentUrl)) {
@@ -427,7 +483,247 @@ public class XMPPConnection implements AutoCloseable {
     }
 
     @Override
+    public void ProcessPingEvent(PingEvent pingEvent) {
+        userService.getJIDsbyUID(pingEvent.getPinger().getUid())
+                .forEach(userJid -> {
+                    Presence p = new Presence(Jid.of(userJid));
+                    p.setFrom(jid);
+                    p.setPriority((byte) 10);
+                    xmpp.sendOut(ClientPresence.from(p));
+                });
+    }
+
+    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) {
+            com.juick.Message jmsg = new com.juick.Message();
+            jmsg.setUser(user_from);
+            jmsg.setTo(userService.getUserByUID(uid_to).get());
+            jmsg.setText(msg.getBody());
+            applicationEventPublisher.publishEvent(new MessageEvent(this, jmsg));
+        } 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 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 = commandsManager.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;
+    }
+
+    @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)) {
+                sendStanza(msg);
+            }
+        } else if (xmlValue instanceof IQ) {
+            IQ iq = (IQ) xmlValue;
+            sendStanza(iq);
+        }
+    }
+
+    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);
+            }
+        });
+    }
+
+    @PreDestroy
     public void close() throws Exception {
+        broadcastPresence(Presence.Type.UNAVAILABLE);
         if (router != null) {
             router.close();
         }