diff options
Diffstat (limited to 'EssentialsXMPP/src/com/earth2me/essentials/xmpp/XMPPManager.java')
| -rw-r--r-- | EssentialsXMPP/src/com/earth2me/essentials/xmpp/XMPPManager.java | 283 |
1 files changed, 283 insertions, 0 deletions
diff --git a/EssentialsXMPP/src/com/earth2me/essentials/xmpp/XMPPManager.java b/EssentialsXMPP/src/com/earth2me/essentials/xmpp/XMPPManager.java new file mode 100644 index 000000000..e9da61440 --- /dev/null +++ b/EssentialsXMPP/src/com/earth2me/essentials/xmpp/XMPPManager.java @@ -0,0 +1,283 @@ +package com.earth2me.essentials.xmpp; + +import com.earth2me.essentials.EssentialsConf; +import com.earth2me.essentials.IConf; +import java.io.File; +import java.util.ArrayList; +import java.util.Collections; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.logging.Handler; +import java.util.logging.Level; +import java.util.logging.LogRecord; +import java.util.logging.Logger; +import org.bukkit.command.ConsoleCommandSender; +import org.bukkit.craftbukkit.CraftServer; +import org.bukkit.entity.Player; +import org.bukkit.plugin.java.JavaPlugin; +import org.jivesoftware.smack.Chat; +import org.jivesoftware.smack.ChatManager; +import org.jivesoftware.smack.ChatManagerListener; +import org.jivesoftware.smack.ConnectionConfiguration; +import org.jivesoftware.smack.MessageListener; +import org.jivesoftware.smack.Roster.SubscriptionMode; +import org.jivesoftware.smack.XMPPConnection; +import org.jivesoftware.smack.XMPPException; +import org.jivesoftware.smack.packet.Message; +import org.jivesoftware.smack.packet.Presence; + + +public class XMPPManager extends Handler implements MessageListener, ChatManagerListener, IConf +{ + private static final Logger LOGGER = Logger.getLogger("Minecraft"); + private final transient EssentialsConf config; + private transient XMPPConnection connection; + private transient ChatManager chatManager; + private final transient Map<String, Chat> chats = Collections.synchronizedMap(new HashMap<String, Chat>()); + private final transient JavaPlugin parent; + private transient List<String> logUsers; + private transient Level logLevel; + private transient boolean ignoreLagMessages = true; + + public XMPPManager(final JavaPlugin parent) + { + super(); + this.parent = parent; + config = new EssentialsConf(new File(parent.getDataFolder(), "config.yml")); + config.setTemplateName("/config.yml", EssentialsXMPP.class); + reloadConfig(); + } + + public void sendMessage(final String address, final String message) + { + Chat chat = null; + try + { + if (address == null || address.isEmpty()) + { + return; + } + startChat(address); + chat = chats.get(address); + if (chat == null) + { + return; + } + chat.sendMessage(message.replaceAll("ยง[0-9a-f]", "")); + } + catch (XMPPException ex) + { + if (chat != null) + { + chat.removeMessageListener(this); + chats.remove(address); + LOGGER.log(Level.WARNING, "Failed to send xmpp message.", ex); + } + } + } + + @Override + public void processMessage(final Chat chat, final Message msg) + { + final String message = msg.getBody(); + if (message.length() > 0) + { + switch (message.charAt(0)) + { + case '@': + sendPrivateMessage(chat, message); + break; + case '/': + sendCommand(chat, message); + break; + default: + parent.getServer().broadcastMessage("<XMPP:" + chat.getParticipant() + "> " + message); + } + } + } + + private void connect() + { + final String server = config.getString("xmpp.server"); + if (server == null) + { + LOGGER.log(Level.WARNING, "config broken for xmpp"); + return; + } + final int port = config.getInt("xmpp.port", 5222); + final String serviceName = config.getString("xmpp.servicename", server); + final String xmppuser = config.getString("xmpp.user"); + final String password = config.getString("xmpp.password"); + final ConnectionConfiguration cc = new ConnectionConfiguration(server, port, serviceName); + final StringBuilder sb = new StringBuilder(); + sb.append("Connecting to xmpp server ").append(server).append(":").append(port); + sb.append(" as user ").append(xmppuser).append("."); + LOGGER.log(Level.INFO, sb.toString()); + cc.setSASLAuthenticationEnabled(config.getBoolean("xmpp.sasl-enabled", false)); + cc.setSendPresence(true); + cc.setReconnectionAllowed(true); + connection = new XMPPConnection(cc); + try + { + connection.connect(); + connection.login(xmppuser, password); + connection.getRoster().setSubscriptionMode(SubscriptionMode.accept_all); + chatManager = connection.getChatManager(); + chatManager.addChatListener(this); + } + catch (XMPPException ex) + { + LOGGER.log(Level.WARNING, "Failed to connect to server: " + server, ex); + } + } + + public final void disconnect() + { + if (connection != null) + { + connection.disconnect(new Presence(Presence.Type.unavailable)); + } + } + + @Override + public void chatCreated(final Chat chat, final boolean createdLocally) + { + if (!createdLocally) + { + chat.addMessageListener(this); + final Chat old = chats.put(chat.getParticipant(), chat); + if (old != null) + { + old.removeMessageListener(this); + } + } + } + + @Override + public final void reloadConfig() + { + config.load(); + synchronized (chats) + { + disconnect(); + chats.clear(); + connect(); + } + LOGGER.removeHandler(this); + if (config.getBoolean("log-enabled", false)) + { + LOGGER.addHandler(this); + logUsers = config.getStringList("log-users", new ArrayList<String>()); + final String level = config.getString("log-level", "info"); + try + { + logLevel = Level.parse(level.toUpperCase()); + } + catch (IllegalArgumentException e) + { + logLevel = Level.INFO; + } + ignoreLagMessages = config.getBoolean("ignore-lag-messages", true); + } + } + + @Override + public void publish(final LogRecord logRecord) + { + try + { + if (ignoreLagMessages && logRecord.getMessage().equals("Can't keep up! Did the system time change, or is the server overloaded?")) + { + return; + } + if (logRecord.getLevel().intValue() >= logLevel.intValue()) + { + for (String user : logUsers) + { + startChat(user); + final Chat chat = chats.get(user); + if (chat != null) + { + chat.sendMessage(String.format("[" + logRecord.getLevel().getLocalizedName() + "] " + logRecord.getMessage(), logRecord.getParameters())); + } + } + } + } + catch (Exception e) + { + // Ignore all exception and just print them to the console + // Otherwise we create a loop. + System.out.println(e.getMessage()); + e.printStackTrace(); + } + } + + @Override + public void flush() + { + } + + @Override + public void close() throws SecurityException + { + } + + private void startChat(final String address) throws XMPPException + { + if (chatManager == null) + { + return; + } + synchronized (chats) + { + if (!chats.containsKey(address)) + { + final Chat chat = chatManager.createChat(address, this); + if (chat == null) + { + throw new XMPPException("Could not start Chat with " + address); + } + chats.put(address, chat); + } + } + } + + private void sendPrivateMessage(final Chat chat, final String message) + { + final String[] parts = message.split(" ", 2); + if (parts.length == 2) + { + final List<Player> matches = parent.getServer().matchPlayer(parts[0].substring(1)); + + if (matches.isEmpty()) + { + try + { + chat.sendMessage("User " + parts[0] + " not found"); + } + catch (XMPPException ex) + { + LOGGER.log(Level.WARNING, "Failed to send xmpp message.", ex); + } + } else { + for (Player p : matches) + { + p.sendMessage("[" + chat.getParticipant() + ">" + p.getDisplayName() + "] " + message); + } + } + } + } + + private void sendCommand(final Chat chat, final String message) + { + if (config.getStringList("op-users", new ArrayList<String>()).contains(chat.getParticipant())) + { + final CraftServer craftServer = (CraftServer)parent.getServer(); + if (craftServer != null) + { + craftServer.dispatchCommand(new ConsoleCommandSender(craftServer), message.substring(1)); + } + } + } +} |