package jrummikub.control.network; import java.awt.Color; import java.util.UUID; import javax.swing.SwingUtilities; import jrummikub.model.GameSettings; import jrummikub.util.Event; import jrummikub.util.Event1; import jrummikub.util.Event2; import jrummikub.util.GameData; import jrummikub.util.IEvent; import jrummikub.util.IEvent1; import jrummikub.util.IEvent2; import jrummikub.util.LoginData; import org.jivesoftware.smack.Connection; import org.jivesoftware.smack.PacketListener; import org.jivesoftware.smack.XMPPConnection; import org.jivesoftware.smack.XMPPException; import org.jivesoftware.smack.filter.AndFilter; import org.jivesoftware.smack.filter.PacketExtensionFilter; import org.jivesoftware.smack.filter.PacketTypeFilter; import org.jivesoftware.smack.packet.DefaultPacketExtension; import org.jivesoftware.smack.packet.Message; import org.jivesoftware.smack.packet.Packet; import org.jivesoftware.smack.packet.PacketExtension; import org.jivesoftware.smack.packet.XMPPError; import org.jivesoftware.smack.packet.XMPPError.Type; import org.jivesoftware.smack.util.Base64; import org.jivesoftware.smackx.muc.DiscussionHistory; import org.jivesoftware.smackx.muc.MultiUserChat; public class ConnectionControl implements IConnectionControl { private final static String ELEMENT_NAME = "rummikub"; private final static String NAMESPACE = "http://home.universe-factory.net/rummikub/"; private final LoginData loginData; private volatile Connection connection; private volatile MultiUserChat muc; private Event connectedEvent = new Event(); private Event connectionFailedEvent = new Event(); private Event1 gameOfferEvent = new Event1(); private Event1 gameWithdrawalEvent = new Event1(); private Event1 gameJoinEvent = new Event1(); private Event1 gameLeaveEvent = new Event1(); private Event1 gameJoinAckEvent = new Event1(); private Event2 changeColorEvent = new Event2(); private GameData currentGame; private volatile GameData offeredGame; public ConnectionControl(LoginData loginData) { this.loginData = loginData; } @Override public String getNickname() { return muc.getNickname(); } @Override public void connect() { new Thread(new ConnectRunner()).start(); } @Override public void disconnect() { connectedEvent = new Event(); connectionFailedEvent = new Event(); new Thread(new Runnable() { @Override public void run() { synchronized (ConnectionControl.this) { if (connection != null) { connection.disconnect(); connection = null; } } } }).start(); } @Override public IEvent getConnectedEvent() { return connectedEvent; } @Override public IEvent getConnectionFailedEvent() { return connectionFailedEvent; } @Override public IEvent1 getGameOfferEvent() { return gameOfferEvent; } @Override public IEvent1 getGameWithdrawalEvent() { return gameWithdrawalEvent; } @Override public IEvent1 getGameJoinEvent() { return gameJoinEvent; } @Override public IEvent1 getGameLeaveEvent() { return gameLeaveEvent; } @Override public IEvent1 getGameJoinAckEvent() { return gameJoinAckEvent; } @Override public IEvent2 getChangeColorEvent() { return changeColorEvent; } @Override public void offerGame(GameData data) { offeredGame = data; currentGame = data; sendGameOffer(); } @Override public void withdrawGame() { offeredGame = null; final UUID uuid = currentGame.getGameID(); currentGame = null; new Thread(new SendRunner() { @Override public Message send() { DefaultPacketExtension extension = createJRummikubExtension(); extension.setValue("messageType", "game_withdrawal"); extension.setValue("uuid", uuid.toString()); return createMessage(extension); } }).start(); } @Override public GameData getCurrentGame() { return currentGame; } @Override public void setCurrentGame(GameData game) { this.currentGame = game; } @Override public void joinGame(final GameData game) { setCurrentGame(game); new Thread(new SendRunner() { @Override public Message send() { DefaultPacketExtension extension = createJRummikubExtension(); extension.setValue("messageType", "game_join"); extension.setValue("uuid", game.getGameID().toString()); return createMessage(extension); } }).start(); } @Override public void leaveGame() { final UUID uuid = currentGame.getGameID(); currentGame = null; new Thread(new SendRunner() { @Override public Message send() { DefaultPacketExtension extension = createJRummikubExtension(); extension.setValue("messageType", "game_leave"); extension.setValue("uuid", uuid.toString()); return createMessage(extension); } }).start(); } @Override public void ackJoinGame(final String recipient, final boolean ack) { final UUID uuid = currentGame.getGameID(); new Thread(new SendRunner() { @Override public Message send() { DefaultPacketExtension extension = createJRummikubExtension(); extension.setValue("messageType", "game_join_ack"); extension.setValue("uuid", uuid.toString()); extension.setValue("ack", Boolean.toString(ack)); Message message = createMessage(extension); message.setType(Message.Type.normal); message.setTo(muc.getRoom() + "/" + recipient); return message; } }).start(); } @Override public void changeColor(final Color color) { final UUID uuid = currentGame.getGameID(); new Thread(new SendRunner() { @Override public Message send() { DefaultPacketExtension extension = createJRummikubExtension(); extension.setValue("messageType", "change_color"); extension.setValue("uuid", uuid.toString()); extension.setValue("color", Base64.encodeObject(color, Base64.GZIP)); return createMessage(extension); } }).start(); } private void sendGameOffer() { final GameData data = offeredGame; new Thread(new SendRunner() { @Override public Message send() { DefaultPacketExtension extension = createJRummikubExtension(); extension.setValue("messageType", "game_offer"); extension.setValue("uuid", data.getGameID().toString()); extension.setValue("gameSettings", Base64.encodeObject( data.getGameSettings(), Base64.GZIP)); return createMessage(extension); } }).start(); } private void requestGames() { new Thread(new SendRunner() { @Override public Message send() { DefaultPacketExtension extension = createJRummikubExtension(); extension.setValue("messageType", "game_request"); return createMessage(extension); } }); } private static void emitLater(final Event event) { SwingUtilities.invokeLater(new Runnable() { @Override public void run() { event.emit(); } }); } private Message createMessage(PacketExtension extension) { Message message = muc.createMessage(); message.addExtension(extension); return message; } private static DefaultPacketExtension createJRummikubExtension() { return new DefaultPacketExtension(ELEMENT_NAME, NAMESPACE); } private void processPacket(Packet packet) { DefaultPacketExtension extension = (DefaultPacketExtension) packet .getExtension(ELEMENT_NAME, NAMESPACE); if (((Message) packet).getType() == Message.Type.error) { System.err.println("Received error message from '" + packet.getFrom() + "'"); return; } String sender = packet.getFrom(); sender = sender.substring(sender.indexOf('/') + 1); String messageType = extension.getValue("messageType"); handleMessageTypes(extension, sender, messageType); } private void handleMessageTypes(DefaultPacketExtension extension, String sender, String messageType) { if (messageType.equals("game_offer")) { UUID uuid = UUID.fromString(extension.getValue("uuid")); GameSettings settings = (GameSettings) Base64 .decodeToObject(extension.getValue("gameSettings")); GameData gameData = new GameData(uuid, settings, sender); gameOfferEvent.emit(gameData); } else if (messageType.equals("game_withdrawal")) { gameWithdrawalEvent .emit(UUID.fromString(extension.getValue("uuid"))); } else if (messageType.equals("game_request")) { if (offeredGame != null) { sendGameOffer(); } } else if (currentGame != null) { UUID uuid = UUID.fromString(extension.getValue("uuid")); if (!currentGame.getGameID().equals(uuid)) { return; } messagesDuringGame(extension, sender, messageType); } } private void messagesDuringGame(DefaultPacketExtension extension, String sender, String messageType) { if (messageType.equals("game_join")) { gameJoinEvent.emit(sender); } else if (messageType.equals("game_leave")) { gameLeaveEvent.emit(sender); } else if (messageType.equals("game_join_ack")) { gameJoinAckEvent .emit(Boolean.valueOf(extension.getValue("ack"))); } else if (messageType.equals("change_color")) { changeColorEvent.emit(sender, (Color) Base64 .decodeToObject(extension.getValue("color"))); } else { System.err.println("Received unrecognized message of type '" + messageType + "'"); } } private class ConnectRunner implements Runnable { @Override public void run() { synchronized (ConnectionControl.this) { connection = new XMPPConnection(loginData.getServerName()); connection.addPacketListener(new PacketListener() { @Override public void processPacket(final Packet packet) { SwingUtilities.invokeLater(new Runnable() { @Override public void run() { ConnectionControl.this.processPacket(packet); } }); } }, new AndFilter(new PacketTypeFilter(Message.class), new PacketExtensionFilter(ELEMENT_NAME, NAMESPACE))); try { doConnect(); } catch (XMPPException e) { connection.disconnect(); connection = null; // TODO Auto-generated catch block e.printStackTrace(); emitLater(connectionFailedEvent); } } } private void doConnect() throws XMPPException { connection.connect(); connection.login(loginData.getUserName(), loginData.getPassword(), "JRummikub"); muc = new MultiUserChat(connection, loginData.getChannelName()); DiscussionHistory history = new DiscussionHistory(); history.setMaxStanzas(0); String nickname = loginData.getUserName(); // Loop until a unused nickname is found while (true) { try { muc.join(nickname, null, history, 10000); break; // Join was successful, break the loop } catch (XMPPException e) { XMPPError error = e.getXMPPError(); if (error.getType() == Type.CANCEL && error.getCode() == 409) { // There was a conflict, try again with another // nickname nickname += "_"; continue; } else { // An unknown error has occurred, cancel connect throw e; } } } emitLater(connectedEvent); requestGames(); } } private abstract class SendRunner implements Runnable { @Override public void run() { synchronized (ConnectionControl.this) { if (connection != null) { connection.sendPacket(send()); } } } abstract public Message send(); } }