package com.juick.ws; import com.fasterxml.jackson.core.JsonProcessingException; import com.fasterxml.jackson.databind.ObjectMapper; import com.fasterxml.jackson.dataformat.xml.XmlMapper; import com.juick.User; import com.juick.service.MessagesService; import com.juick.service.SubscriptionService; import org.apache.commons.lang3.math.NumberUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.core.env.Environment; import org.springframework.util.Assert; import org.springframework.web.socket.TextMessage; import rocks.xmpp.core.XmppException; import rocks.xmpp.core.session.Extension; import rocks.xmpp.core.session.XmppSession; import rocks.xmpp.core.session.XmppSessionConfiguration; import rocks.xmpp.core.session.debug.LogbackDebugger; import rocks.xmpp.core.stanza.model.Message; import rocks.xmpp.extensions.component.accept.ExternalComponent; import javax.inject.Inject; import java.io.IOException; import java.util.List; import java.util.stream.Collectors; /** * @author ugnich */ public class XMPPConnection implements AutoCloseable { private static final Logger logger = LoggerFactory.getLogger(XMPPConnection.class); private final WebsocketComponent wsHandler; private final String xmppPassword; private final ObjectMapper ms; private final int xmppPort; private final String wsJid; private XmppSession xmpp; @Inject MessagesService messagesService; @Inject SubscriptionService subscriptionService; public XMPPConnection( final Environment env, final WebsocketComponent wsHandler) { Assert.notNull(env); Assert.notNull(wsHandler); this.wsHandler = wsHandler; xmppPassword = env.getProperty("xmpp_password"); xmppPort = NumberUtils.toInt(env.getProperty("xmpp_port"), 5347); wsJid = env.getProperty("ws_jid", "ws.juick.local"); ms = new XmlMapper(); XmppSessionConfiguration configuration = XmppSessionConfiguration.builder() .extensions(Extension.of(com.juick.Message.class)) .debugger(LogbackDebugger.class) .build(); xmpp = ExternalComponent.create(wsJid, xmppPassword, configuration, "localhost", xmppPort); xmpp.addInboundMessageListener(e -> { try { Message msg = e.getMessage(); com.juick.Message jmsg = msg.getExtension(com.juick.Message.class); if (jmsg != null) { if (logger.isInfoEnabled()) // prevent writeValueAsString execution if log is disabled logger.info("got msg: {}", ms.writeValueAsString(jmsg)); if (jmsg.getMid() == 0) { int uid_to = NumberUtils.toInt(msg.getTo().getLocal(), 0); if (uid_to > 0) { onJuickPM(uid_to, jmsg); } } else if (jmsg.getRid() == 0) { // to get full message with attachment, etc. onJuickMessagePost(messagesService.getMessage(jmsg.getMid())); } else { // to get quote and attachment com.juick.Message reply = messagesService.getReply(jmsg.getMid(), jmsg.getRid()); onJuickMessageReply(reply); } } } catch (JsonProcessingException ex) { logger.error("mapper exception", ex); } }); try { xmpp.connect(); } catch (XmppException e) { logger.warn("xmpp extension", e); } } @Override public void close() throws Exception { if (xmpp != null) xmpp.close(); logger.info("XmppSession on WS destroyed"); } private void onJuickPM(final int uid_to, final com.juick.Message jmsg) throws JsonProcessingException { String json = ms.writeValueAsString(jmsg); synchronized (wsHandler.getClients()) { wsHandler.getClients().stream().filter(c -> !c.legacy && c.visitor.getUid() == uid_to).forEach(c -> { try { logger.info("sending pm to {}", c.visitor.getUid()); c.session.sendMessage(new TextMessage(json)); } catch (IOException e) { logger.warn("ws error", e); } }); } } private void onJuickMessagePost(final com.juick.Message jmsg) throws JsonProcessingException { String json = ms.writeValueAsString(jmsg); List uids = subscriptionService.getSubscribedUsers(jmsg.getUser().getUid(), jmsg.getMid()) .stream().map(User::getUid).collect(Collectors.toList()); synchronized (wsHandler.getClients()) { wsHandler.getClients().stream().filter(c -> (!c.legacy && c.visitor.getUid() == 0) // anonymous users || (!c.legacy && uids.contains(c.visitor.getUid()))) // subscriptions .forEach(c -> { try { logger.info("sending message to {}", c.visitor.getUid()); c.session.sendMessage(new TextMessage(json)); } catch (IOException e) { logger.warn("ws error", e); } }); wsHandler.getClients().stream().filter(c -> c.legacy && c.allMessages) // legacy all posts .forEach(c -> { try { logger.info("sending message to legacy client {}", c.visitor.getUid()); c.session.sendMessage(new TextMessage(json)); } catch (IOException e) { logger.warn("ws error", e); } }); } } private void onJuickMessageReply(final com.juick.Message jmsg) throws JsonProcessingException { String json = ms.writeValueAsString(jmsg); List threadUsers = subscriptionService.getUsersSubscribedToComments(jmsg.getMid(), jmsg.getUser().getUid()) .stream().map(User::getUid).collect(Collectors.toList()); synchronized (wsHandler.getClients()) { wsHandler.getClients().stream().filter(c -> (!c.legacy && c.visitor.getUid() == 0) // anonymous users || (!c.legacy && threadUsers.contains(c.visitor.getUid()))) // subscriptions .forEach(c -> { try { logger.info("sending reply to {}", c.visitor.getUid()); c.session.sendMessage(new TextMessage(json)); } catch (IOException e) { logger.warn("ws error", e); } }); wsHandler.getClients().stream().filter(c -> (c.legacy && c.allReplies) || (c.legacy && c.MID == jmsg.getMid())) // legacy replies .forEach(c -> { try { logger.info("sending reply to legacy client {}", c.visitor.getUid()); c.session.sendMessage(new TextMessage(json)); } catch (IOException e) { logger.warn("ws error", e); } }); } } }