merge router and component

1 files changed, 221 insertions, 23 deletions

diff --git a/src/main/java/com/juick/xmpp/s2s/XMPPComponent.java b/src/main/java/com/juick/xmpp/s2s/XMPPComponent.java
index ecfe1671..9b8bb07e 100644
--- a/src/main/java/com/juick/xmpp/s2s/XMPPComponent.java
+++ b/src/main/java/com/juick/xmpp/s2s/XMPPComponent.java
@@ -1,8 +1,13 @@
 package com.juick.xmpp.s2s;
 
-import com.juick.xmpp.Stanza;
-import com.juick.xmpp.StanzaChild;
+import com.juick.User;
+import com.juick.server.MessagesQueries;
+import com.juick.server.SubscriptionsQueries;
+import com.juick.server.UserQueries;
+import com.juick.xmpp.*;
 import com.juick.xmpp.extensions.JuickMessage;
+import com.juick.xmpp.extensions.Nickname;
+import com.juick.xmpp.extensions.XOOB;
 import org.springframework.jdbc.core.JdbcTemplate;
 import org.springframework.jdbc.datasource.DriverManagerDataSource;
 import org.xmlpull.v1.XmlPullParserException;
@@ -17,9 +22,6 @@ import java.security.KeyStoreException;
 import java.security.NoSuchAlgorithmException;
 import java.security.UnrecoverableKeyException;
 import java.security.cert.CertificateException;
-import java.sql.Driver;
-import java.sql.DriverManager;
-import java.sql.SQLException;
 import java.util.*;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
@@ -30,19 +32,20 @@ import java.util.logging.Logger;
  *
  * @author ugnich
  */
-public class XMPPComponent implements ServletContextListener {
+public class XMPPComponent implements ServletContextListener, Stream.StreamListener,
+        Message.MessageListener, Iq.IqListener, Presence.PresenceListener {
 
-    private static final Logger LOGGER = Logger.getLogger(XMPPComponent.class.getName());
+    private static final Logger logger = Logger.getLogger(XMPPComponent.class.getName());
 
     public final ExecutorService executorService = Executors.newCachedThreadPool();
-    ExecutorService xmppService = Executors.newSingleThreadExecutor();
+    StreamComponent router;
+    JuickBot bot;
 
-    public String HOSTNAME = null;
+    public String HOSTNAME, componentName;
     public String STATSFILE = null;
     public String keystore;
     public String keystorePassword;
     public List<String> brokenSSLhosts;
-    public ConnectionRouter connRouter;
     final List<ConnectionIn> inConnections = Collections.synchronizedList(new ArrayList<>());
     final List<ConnectionOut> outConnections = Collections.synchronizedList(new ArrayList<>());
     final List<CacheEntry> outCache = Collections.synchronizedList(new ArrayList<>());
@@ -135,7 +138,7 @@ public class XMPPComponent implements ServletContextListener {
             try {
                 connOut.sendStanza(xml);
             } catch (IOException e) {
-                LOGGER.warning("STREAM TO " + connOut.to + " " + connOut.streamID + " ERROR: " + e.toString());
+                logger.warning("STREAM TO " + connOut.to + " " + connOut.streamID + " ERROR: " + e.toString());
             }
             return;
         }
@@ -161,7 +164,7 @@ public class XMPPComponent implements ServletContextListener {
                 connectionOut = new ConnectionOut(this, hostname);
                 executorService.submit(connectionOut);
             } catch (CertificateException | UnrecoverableKeyException | NoSuchAlgorithmException | XmlPullParserException | KeyStoreException | KeyManagementException | IOException e) {
-                LOGGER.log(Level.SEVERE, "s2s out error", e);
+                logger.log(Level.SEVERE, "s2s out error", e);
             }
         }
     }
@@ -169,12 +172,13 @@ public class XMPPComponent implements ServletContextListener {
     @Override
     public void contextInitialized(ServletContextEvent sce) {
 
-        LOGGER.info("component initialized");
+        logger.info("component initialized");
         Properties conf = new Properties();
         try {
             conf.load(sce.getServletContext().getResourceAsStream("/WEB-INF/juick.conf"));
             HOSTNAME = conf.getProperty("hostname");
-            String componentName = conf.getProperty("componentname");
+            componentName = conf.getProperty("componentname");
+            JID Jid = new JID(conf.getProperty("xmppbot_jid"));
             STATSFILE = conf.getProperty("statsfile");
             keystore = conf.getProperty("keystore");
             keystorePassword = conf.getProperty("keystore_password");
@@ -183,27 +187,34 @@ public class XMPPComponent implements ServletContextListener {
             dataSource.setDriverClassName(conf.getProperty("datasource_driver", "com.mysql.jdbc.Driver"));
             dataSource.setUrl(conf.getProperty("datasource_url"));
             sql = new JdbcTemplate(dataSource);
+            bot = new JuickBot(this, Jid);
 
             childParsers.put(JuickMessage.XMLNS, new JuickMessage());
-            xmppService.submit(() -> {
-                connRouter = new ConnectionRouter(this, componentName, conf.getProperty("xmpp_password"));
+            Socket routerSocket = new Socket("localhost", 5347);
+            router = new StreamComponent(new JID("s2s"), routerSocket.getInputStream(), routerSocket.getOutputStream(), conf.getProperty("xmpp_password"));
+            router.addChildParser(new JuickMessage());
+            router.addListener((Stream.StreamListener) this);
+            router.addListener((Message.MessageListener) this);
+            router.addListener((Iq.IqListener) this);            
+            executorService.submit(() -> {
+                router.startParsing();
                 executorService.submit(new CleaningUp(this));
                 final ServerSocket listener = new ServerSocket(5269);
-                LOGGER.info("s2s listener ready");
+                logger.info("s2s listener ready");
                 while (true) {
                     try {
                         Socket socket = listener.accept();
-                        ConnectionIn client = new ConnectionIn(this, new JuickBot(this), socket);
+                        ConnectionIn client = new ConnectionIn(this, bot, socket);
                         addConnectionIn(client);
                         executorService.submit(client);
                     } catch (Exception e) {
-                        LOGGER.log(Level.SEVERE, "s2s error", e);
+                        logger.log(Level.SEVERE, "s2s error", e);
                     }
                 }
             });
 
         } catch (Exception e) {
-            LOGGER.log(Level.SEVERE, "XMPPComponent error", e);
+            logger.log(Level.SEVERE, "XMPPComponent error", e);
         }
     }
 
@@ -228,11 +239,198 @@ public class XMPPComponent implements ServletContextListener {
         }
 
         try {
-            connRouter.closeConnection();
+            closeRouterConnection();
         } catch (IOException e) {
-            LOGGER.log(Level.WARNING, "router warning", e);
+            logger.log(Level.WARNING, "router warning", e);
         }
         executorService.shutdown();
-        LOGGER.info("component destroyed");
+        logger.info("component destroyed");
+    }
+    public void closeRouterConnection() throws IOException {
+        router.logoff();
+    }
+
+    public void sendJuickMessage(JuickMessage jmsg) {
+        List<String> jids = new ArrayList<>();
+
+        if (jmsg.FriendsOnly) {
+            jids = SubscriptionsQueries.getJIDSubscribedToUser(sql, jmsg.getUser().getUID(), jmsg.FriendsOnly);
+        } else {
+            List<User> users = SubscriptionsQueries.getSubscribedUsers(sql, jmsg.getUser().getUID(), jmsg.getMID());
+            for (User user : users) {
+                for (String jid : UserQueries.getJIDsbyUID(sql, user.getUID())) {
+                    jids.add(jid);
+                }
+            }
+        }
+
+        String txt = "@" + jmsg.getUser().getUName() + ":" + jmsg.getTagsString() + "\n";
+        String attachment = jmsg.getAttachmentURL();
+        if (attachment != null) {
+            txt += attachment + "\n";
+        }
+        txt += jmsg.getText() + "\n\n";
+        txt += "#" + jmsg.getMID() + " http://juick.com/" + jmsg.getMID();
+
+        Nickname nick = new Nickname();
+        nick.Nickname = "@" + jmsg.getUser().getUName();
+
+        com.juick.xmpp.Message msg = new com.juick.xmpp.Message();
+        msg.from = bot.JuickJID;
+        msg.body = txt;
+        msg.type = Message.Type.chat;
+        msg.thread = "juick-" + jmsg.getMID();
+        msg.addChild(jmsg);
+        msg.addChild(nick);
+        if (attachment != null) {
+            XOOB oob = new XOOB();
+            oob.URL = attachment;
+            msg.addChild(oob);
+        }
+
+        for (String jid : jids) {
+            msg.to = new JID(jid);
+            sendOut(msg);
+        }
+    }
+
+    public void sendJuickComment(JuickMessage jmsg) {
+        List<User> users;
+        String replyQuote;
+        String replyTo;
+
+        users = SubscriptionsQueries.getUsersSubscribedToComments(sql, jmsg.getMID(), jmsg.getUser().getUID());
+        com.juick.Message replyMessage = jmsg.ReplyTo > 0 ? MessagesQueries.getReply(sql, jmsg.getMID(), jmsg.ReplyTo)
+                : MessagesQueries.getMessage(sql, jmsg.getMID());
+        replyTo = replyMessage.getUser().getUName();
+        replyQuote = getReplyQuote(replyMessage);
+
+        String txt = "Reply by @" + jmsg.getUser().getUName() + ":\n" + replyQuote + "\n@" + replyTo + " ";
+        String attachment = jmsg.getAttachmentURL();
+        if (attachment != null) {
+            txt += attachment + "\n";
+        }
+        txt += jmsg.getText() + "\n\n" + "#" + jmsg.getMID() + "/" + jmsg.getRID() + " http://juick.com/" + jmsg.getMID() + "#" + jmsg.getRID();
+
+        com.juick.xmpp.Message msg = new com.juick.xmpp.Message();
+        msg.from = bot.JuickJID;
+        msg.body = txt;
+        msg.type = Message.Type.chat;
+        msg.addChild(jmsg);
+        for (User user : users) {
+            for (String jid : UserQueries.getJIDsbyUID(sql, user.getUID())) {
+                msg.to = new JID(jid);
+                sendOut(msg);
+            }
+        }
+    }
+
+    private String getReplyQuote(com.juick.Message q) {
+        String quote = q.getText();
+        if (quote.length() > 50) {
+            quote = ">" + quote.substring(0, 47).replace('\n', ' ') + "...\n";
+        } else if (quote.length() > 0) {
+            quote = ">" + quote.replace('\n', ' ') + "\n";
+        }
+        return quote;
+    }
+
+    public void sendJuickRecommendation(JuickMessage recomm) {
+        List<User> users;
+        JuickMessage jmsg;
+        jmsg = new JuickMessage(MessagesQueries.getMessage(sql, recomm.getMID()));
+        users = SubscriptionsQueries.getUsersSubscribedToUserRecommendations(sql,
+                recomm.getUser().getUID(), recomm.getMID(), jmsg.getUser().getUID());
+
+        String txt = "Recommended by @" + recomm.getUser().getUName() + ":\n";
+        txt += "@" + jmsg.getUser().getUName() + ":" + jmsg.getTagsString() + "\n";
+        String attachment = jmsg.getAttachmentURL();
+        if (attachment != null) {
+            txt += attachment + "\n";
+        }
+        txt += jmsg.getText() + "\n\n";
+        txt += "#" + jmsg.getMID();
+        if (jmsg.Replies > 0) {
+            if (jmsg.Replies % 10 == 1 && jmsg.Replies % 100 != 11) {
+                txt += " (" + jmsg.Replies + " reply)";
+            } else {
+                txt += " (" + jmsg.Replies + " replies)";
+            }
+        }
+        txt += " http://juick.com/" + jmsg.getMID();
+
+        Nickname nick = new Nickname();
+        nick.Nickname = "@" + jmsg.getUser().getUName();
+
+        com.juick.xmpp.Message msg = new com.juick.xmpp.Message();
+        msg.from = bot.JuickJID;
+        msg.body = txt;
+        msg.type = Message.Type.chat;
+        msg.thread = "juick-" + jmsg.getMID();
+        msg.addChild(jmsg);
+        msg.addChild(nick);
+        if (attachment != null) {
+            XOOB oob = new XOOB();
+            oob.URL = attachment;
+            msg.addChild(oob);
+        }
+
+        for (User user : users) {
+            for (String jid : UserQueries.getJIDsbyUID(sql, user.getUID())) {
+                msg.to = new JID(jid);
+                sendOut(msg);
+            }
+        }
+    }
+
+    @Override
+    public boolean onIq(Iq iq) {
+        JID jid = iq.to;
+        if (!jid.Host.equals(componentName)) {
+            logger.info("STREAM ROUTER (IQ): " + iq.toString());
+            sendOut(iq);
+        }
+        return false;
+    }
+
+    @Override
+    public void onMessage(Message xmsg) {
+        logger.info("STREAM ROUTER (PROCESS): " + xmsg.toString());
+        JuickMessage jmsg = (JuickMessage) xmsg.getChild(JuickMessage.XMLNS);
+        JID jid = xmsg.to;
+        if (jid.Host.equals(componentName)) {
+            if (jmsg != null) {
+                if (jid.Username != null && jid.Username.equals("recomm")) {
+                    sendJuickRecommendation(jmsg);
+                } else {
+                    if (jmsg.getRID() > 0) {
+                        sendJuickComment(jmsg);
+                    } else if (jmsg.getMID() > 0) {
+                        sendJuickMessage(jmsg);
+                    }
+                }
+            }
+        } else {
+            sendOut(xmsg);
+        }
+    }
+
+    @Override
+    public void onPresence(Presence presence) {
+        JID jid = presence.to;
+        if (!jid.Host.equals(componentName)) {
+            logger.info("STREAM ROUTER (PRESENCE): " + presence.toString());
+            sendOut(presence);
+        }
+    }
+
+    @Override
+    public void onStreamReady() {
+        logger.info("STREAM ROUTER (READY)");
+    }
+
+    @Override
+    public void onStreamFail(Exception ex) {
+        logger.log(Level.SEVERE, "STREAM ROUTER (FAIL)", ex);
     }
 }