path: root/src/jrummikub/control/network/ConnectionControl.java

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;

class ConnectionControl {
	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<GameData> gameOfferEvent = new Event1<GameData>();
	private Event1<UUID> gameWithdrawalEvent = new Event1<UUID>();

	private Event1<String> gameJoinEvent = new Event1<String>();
	private Event1<String> gameLeaveEvent = new Event1<String>();

	private Event1<Boolean> gameJoinAckEvent = new Event1<Boolean>();

	private Event2<String, Color> changeColorEvent = new Event2<String, Color>();

	private GameData currentGame;

	private volatile GameData offeredGame;

	ConnectionControl(LoginData loginData) {
		this.loginData = loginData;
	}

	String getNickname() {
		return muc.getNickname();
	}

	void connect() {
		new Thread(new ConnectRunner()).start();
	}

	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();
	}

	IEvent getConnectedEvent() {
		return connectedEvent;
	}

	IEvent getConnectionFailedEvent() {
		return connectionFailedEvent;
	}

	IEvent1<GameData> getGameOfferEvent() {
		return gameOfferEvent;
	}

	IEvent1<UUID> getGameWithdrawalEvent() {
		return gameWithdrawalEvent;
	}

	IEvent1<String> getGameJoinEvent() {
		return gameJoinEvent;
	}

	IEvent1<String> getGameLeaveEvent() {
		return gameLeaveEvent;
	}

	IEvent1<Boolean> getGameJoinAckEvent() {
		return gameJoinAckEvent;
	}

	IEvent2<String, Color> getChangeColorEvent() {
		return changeColorEvent;
	}

	void offerGame(GameData data) {
		offeredGame = data;
		currentGame = data;
		sendGameOffer();
	}

	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();
	}

	GameData getCurrentGame() {
		return currentGame;
	}

	void setCurrentGame(GameData game) {
		this.currentGame = game;
	}

	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();
	}

	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();
	}

	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();
	}

	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");

		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;
			}
			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();
	}
}