package com.juick.components; import com.juick.User; import com.juick.server.UserQueries; import com.juick.server.helpers.UserInfo; import com.juick.server.protocol.JuickProtocol; import com.juick.server.protocol.ProtocolReply; 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 org.springframework.jdbc.core.JdbcTemplate; 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.Closeable; import java.io.IOException; import java.lang.reflect.InvocationTargetException; import java.net.URL; /** * Created by vt on 12/11/2016. */ public class XMPPBot implements AutoCloseable{ private static final Logger logger = LoggerFactory.getLogger(XMPPBot.class); @Inject JdbcTemplate jdbc; 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)); Jid juickJid = Jid.of(env.getProperty("xmppbot_jid", "juick@juick.com/Juick")); 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); JuickProtocol protocol = new JuickProtocol(jdbc, "http://juick.com/"); 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 = UserQueries.getUserByJID(jdbc, 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()); UserQueries.updateUserInfo(jdbc, 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 = protocol.getReply(user, text); Message replyMessage = new Message(message.getFrom(), Message.Type.CHAT); replyMessage.setBody(reply.getDescription()); replyMessage.setFrom(juickJid); component.send(replyMessage); } catch (InvocationTargetException | IllegalAccessException | NoSuchMethodException ex) { logger.warn("unhandled error", ex); } } }); component.connect(); } catch (XmppException e) { logger.error("bot connection error", e); } catch (IOException e) { logger.error("bot initialization error", e); } } @Override public void close() throws Exception { if (component != null) component.close(); logger.info("ExternalComponent on xmpp-bot destroyed"); } }