package com.juick.components; import com.juick.User; import com.juick.server.helpers.UserInfo; import com.juick.server.protocol.JuickProtocol; import com.juick.server.protocol.ProtocolListener; import com.juick.server.protocol.ProtocolReply; import com.juick.service.PMQueriesService; import com.juick.service.UserService; import org.apache.commons.io.IOUtils; import org.apache.commons.lang3.math.NumberUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.core.env.Environment; import rocks.xmpp.addr.Jid; import rocks.xmpp.core.XmppException; import rocks.xmpp.core.stanza.model.Message; import rocks.xmpp.extensions.component.accept.ExternalComponent; import rocks.xmpp.extensions.vcard.temp.VCardManager; import rocks.xmpp.extensions.vcard.temp.model.VCard; import rocks.xmpp.extensions.version.SoftwareVersionManager; import rocks.xmpp.extensions.version.model.SoftwareVersion; import javax.inject.Inject; import java.io.IOException; import java.lang.reflect.InvocationTargetException; import java.net.URL; import java.util.List; /** * Created by vt on 12/11/2016. */ public class XMPPBot implements AutoCloseable, ProtocolListener { private static final Logger logger = LoggerFactory.getLogger(XMPPBot.class); @Inject UserService userService; @Inject PMQueriesService pmQueriesService; @Inject JuickProtocol juickProtocol; Jid juickJid; private ExternalComponent component; public XMPPBot(Environment env) { component = ExternalComponent.create(env.getProperty("component_name", "juick.com"), env.getProperty("component_password", "secret"), env.getProperty("component_host", "localhost"), NumberUtils.toInt(env.getProperty("component_port", "5347"), 5347)); juickJid = Jid.of(env.getProperty("xmppbot_jid", "juick@juick.com/Juick")); juickProtocol.setListener(this); try { SoftwareVersionManager softwareVersionManager = component.getManager(SoftwareVersionManager.class); softwareVersionManager.setSoftwareVersion(new SoftwareVersion("Juick", "git", System.getProperty("os.name", "generic"))); VCardManager vCardManager = component.getManager(VCardManager.class); vCardManager.setEnabled(true); VCard ownVCard = new VCard(); ownVCard.setNickname("Juick"); ownVCard.setUrl(new URL("http://juick.com/")); ownVCard.setPhoto(new VCard.Image("image/png", IOUtils.toByteArray(getClass().getClassLoader().getResourceAsStream("vCard.png")))); vCardManager.setVCard(ownVCard); component.addInboundMessageListener(e -> { Message message = e.getMessage(); if (message.getType().equals(Message.Type.ERROR) || message.getType().equals(Message.Type.GROUPCHAT)) { return; } String text = message.getBody().trim(); String command = text.toUpperCase(); User user = userService.getUserByJID(message.getFrom().asBareJid().toString()); if (command.equals("VCARD")) { try { VCard vCard = vCardManager.getVCard(message.getFrom().asBareJid()).getResult(); UserInfo info = new UserInfo(); info.setFullName(vCard.getFormattedName()); info.setCountry(vCard.getAddresses().get(0).getCountry()); info.setUrl(vCard.getUrl().toString()); userService.updateUserInfo(user, info); component.sendMessage(new Message(message.getFrom(), Message.Type.CHAT, "vCard updated")); } catch (XmppException vce) { logger.warn("vcard exception", vce); } } else { try { ProtocolReply reply = juickProtocol.getReply(user, text); Message replyMessage = new Message(message.getFrom(), Message.Type.CHAT); replyMessage.setBody(reply.getResult()); replyMessage.setFrom(juickJid); component.send(replyMessage); } catch (InvocationTargetException | IllegalAccessException | NoSuchMethodException ex) { logger.warn("unhandled error", ex); } } }); component.connect(); } catch (XmppException | IOException e) { logger.error("bot connection error", e); } } @Override public void close() throws Exception { if (component != null) component.close(); logger.info("ExternalComponent on xmpp-bot destroyed"); } @Override public void privateMessage(User from, User to, String body) { List toJids = userService.getJIDsbyUID(to.getUid()); toJids.forEach(jid -> { Message mm = new Message(); mm.setTo(Jid.of(jid)); mm.setType(Message.Type.CHAT); boolean haveInRoster = pmQueriesService.havePMinRoster(from.getUid(), jid); if (haveInRoster) { mm.setFrom(Jid.of(from.getName(), juickJid.getDomain(), "Juick")); mm.setBody(body); } else { mm.setFrom(Jid.of("juick", juickJid.getDomain(), "Juick")); mm.setBody("Private message from @" + from.getName() + ":\n" + body); } component.send(mm); }); } @Override public void userSubscribed(User from, User to) { String notification = String.format("%s subscribed to your blog", from.getName()); List toJids = userService.getJIDsbyUID(to.getUid()); toJids.forEach(jid -> { Message mm = new Message(); mm.setTo(Jid.of(jid)); mm.setType(Message.Type.CHAT); mm.setFrom(juickJid); mm.setBody(notification); component.send(mm); }); } }