package org.hedgewars.hedgeroid.netplay;

import java.util.Map;

import java.util.TreeMap;

import org.hedgewars.hedgeroid.Datastructures.Room;

import org.hedgewars.hedgeroid.Datastructures.RoomWithId;

import org.hedgewars.hedgeroid.util.ObservableTreeMap;

public class Roomlist extends ObservableTreeMap<String, RoomWithId> {

	private long nextId = 1;

	public void updateList(Room[] newRooms) {

		Map<String, RoomWithId> newMap = new TreeMap<String, RoomWithId>();

		for(Room room : newRooms) {

			RoomWithId oldEntry = get(room.name);

			if(oldEntry == null) {

				newMap.put(room.name, new RoomWithId(room, nextId++));

			} else {

				newMap.put(room.name, new RoomWithId(room, oldEntry.id));

			}

		}

		replaceContent(newMap);

	}

	public void addRoomWithNewId(Room room) {

		put(room.name, new RoomWithId(room, nextId++));

	}

	public void updateRoom(String name, Room room) {

		RoomWithId oldEntry = get(name);

		if(oldEntry == null) {

			addRoomWithNewId(room);

		} else {

			remove(name);

			put(room.name, new RoomWithId(room, oldEntry.id));

		}

	}

}