package org.hedgewars.hedgeroid.netplay;

import static org.hedgewars.hedgeroid.netplay.ThreadedNetConnection.ToNetMsgType.*;

import static org.hedgewars.hedgeroid.util.ObjectUtils.equal;

import java.util.Arrays;

import java.util.Collections;

import java.util.Map;

import java.util.TreeMap;

import org.hedgewars.hedgeroid.BasicRoomState;

import org.hedgewars.hedgeroid.Datastructures.GameConfig;

import org.hedgewars.hedgeroid.Datastructures.MapRecipe;

import org.hedgewars.hedgeroid.Datastructures.Scheme;

import org.hedgewars.hedgeroid.Datastructures.Team;

import org.hedgewars.hedgeroid.Datastructures.TeamInGame;

import org.hedgewars.hedgeroid.Datastructures.TeamIngameAttributes;

import org.hedgewars.hedgeroid.Datastructures.Weaponset;

import org.hedgewars.hedgeroid.netplay.ThreadedNetConnection.ToNetMsgType;

/**

 * This class manages the room state in a network game.

 */

class NetRoomState extends BasicRoomState {

	final Map<String, TeamInGame> requestedTeams = new TreeMap<String, TeamInGame>();

	private Netplay netplay;

	public NetRoomState(Netplay netplay) {

		this.netplay = netplay;

		initRoomState(false);

	}

	public void changeWeaponset(Weaponset weaponset) {

		if(getChiefStatus() && !equal(weaponset, getWeaponset())) {

			sendToNet(MSG_SEND_WEAPONSET, weaponset);

			setWeaponset(weaponset);

		}

	}

	public void changeMapRecipe(MapRecipe mapRecipe) {

		if(getChiefStatus() && !equal(mapRecipe, getMapRecipe())) {

			sendToNet(MSG_SEND_MAP, mapRecipe);

			setMapRecipe(mapRecipe);

		}

	}

	public void changeMapNameAndGenerator(String mapName) {

		if(getChiefStatus() && !equal(mapName, getMapRecipe().name)) {

			int newGenerator = MapRecipe.generatorForMapname(mapName);

			if(newGenerator != getMapRecipe().mapgen) {

				sendToNet(MSG_SEND_MAP_GENERATOR, newGenerator, null);

			}

			sendToNet(MSG_SEND_MAP_NAME, mapName);

			setMapRecipe(getMapRecipe().withName(mapName).withMapgen(newGenerator));

		}

	}

	public void changeMapTemplate(int template) {

		if(getChiefStatus() && template != getMapRecipe().templateFilter) {

			sendToNet(MSG_SEND_MAP_TEMPLATE, template, null);

			setMapRecipe(getMapRecipe().withTemplateFilter(template));

		}

	}

	public void changeMazeSize(int mazeSize) {

		if(getChiefStatus() && mazeSize != getMapRecipe().mazeSize) {

			sendToNet(MSG_SEND_MAZE_SIZE, mazeSize, 0);

			setMapRecipe(getMapRecipe().withMazeSize(mazeSize));

		}

	}

	public void changeMapSeed(String seed) {

		if(getChiefStatus() && !equal(seed, getMapRecipe().seed)) {

			sendToNet(MSG_SEND_MAP_SEED, seed);

			setMapRecipe(getMapRecipe().withSeed(seed));

		}

	}

	public void changeMapTheme(String theme) {

		if(getChiefStatus() && !equal(theme, getMapRecipe().theme)) {

			sendToNet(MSG_SEND_MAP_THEME, theme);

			setMapRecipe(getMapRecipe().withTheme(theme));

		}

	}

	public void changeMapDrawdata(byte[] drawdata) {

		if(getChiefStatus() && !Arrays.equals(drawdata, getMapRecipe().getDrawData())) {

			sendToNet(MSG_SEND_MAP_DRAWDATA, drawdata);

			setMapRecipe(getMapRecipe().withDrawData(drawdata));

		}

	}

	public void changeGameStyle(String gameStyle) {

		if(getChiefStatus() && !equal(gameStyle, getGameStyle())) {

			sendToNet(MSG_SEND_GAMESTYLE, gameStyle);

			setGameStyle(gameStyle);

		}

	}

	public void changeScheme(Scheme scheme) {

		if(getChiefStatus() && !equal(scheme, getScheme())) {

			sendToNet(MSG_SEND_SCHEME, scheme);

			setScheme(scheme);

		}

	}

	void initRoomState(boolean chief) {

		setTeams(Collections.<String, TeamInGame>emptyMap());

		requestedTeams.clear();

		setChief(chief);

		setGameStyle(GameConfig.DEFAULT_STYLE);

		setMapRecipe(MapRecipe.makeRandomMap(0, "seed", GameConfig.DEFAULT_THEME));

		setScheme(netplay.defaultScheme);

		setWeaponset(netplay.defaultWeaponset);

		sendFullConfig();

	}

	void sendFullConfig() {

		if(getChiefStatus()) {

			sendToNet(MSG_SEND_GAMESTYLE, getGameStyle());

			sendToNet(MSG_SEND_SCHEME, getScheme());

			sendToNet(MSG_SEND_WEAPONSET, getWeaponset());

			sendToNet(MSG_SEND_MAP, getMapRecipe());

		}

	}

	private boolean sendToNet(ToNetMsgType what, Object obj) {

		return netplay.sendToNet(what, 0, obj);

	}

	private boolean sendToNet(ToNetMsgType what, int arg1, Object obj) {

		return netplay.sendToNet(what, arg1, obj);

	}

	public void requestAddTeam(Team team, int colorIndex) {

		TeamIngameAttributes tia = new TeamIngameAttributes(netplay.getPlayerName(), colorIndex, TeamIngameAttributes.DEFAULT_HOG_COUNT, false);

		TeamInGame newTeamInGame = new TeamInGame(team, tia);

		requestedTeams.put(team.name, newTeamInGame);

		sendToNet(MSG_SEND_ADD_TEAM, newTeamInGame);

	}

	public void requestRemoveTeam(String teamname) {

		sendToNet(MSG_SEND_REMOVE_TEAM, teamname);

	}

	public void changeTeamColorIndex(String teamname, int colorIndex) {

		if(getChiefStatus()) {

			TeamInGame team = getTeams().get(teamname);

			if(team.ingameAttribs.colorIndex != colorIndex) {

				sendToNet(MSG_SEND_TEAM_COLOR_INDEX, colorIndex, teamname);

				putTeam(team.withAttribs(team.ingameAttribs.withColorIndex(colorIndex)));

			}

		}

	}

	public void changeTeamHogCount(String teamname, int hogcount) {

		if(getChiefStatus()) {

			TeamInGame team = getTeams().get(teamname);

			if(team.ingameAttribs.hogCount != hogcount) {

				sendToNet(MSG_SEND_TEAM_HOG_COUNT, hogcount, teamname);

				putTeam(team.withAttribs(team.ingameAttribs.withHogCount(hogcount)));

			}

		}

	}

}