PartyUpListener

package me.rk.partyup;

import java.util.Iterator;
import java.util.TreeMap;

import org.bukkit.ChatColor;
import org.bukkit.entity.Player;

import com.garbagemule.MobArena.Arena;
import com.garbagemule.MobArena.MobArenaListener;
import com.herocraftonline.dev.heroes.hero.HeroManager;
import com.herocraftonline.dev.heroes.party.HeroParty;

public class PartyUpArenaListener extends MobArenaListener{
    private PartyUp core;
    private static final TreeMap<String, HeroParty> manager = new TreeMap<String,HeroParty>();

    public PartyUpArenaListener(PartyUp core){
        this.core=core;
    }

    @Override
    public void onArenaStart(Arena arena){// may need to do OnWave and check for wave 1
        Iterator<Player> combatants= arena.getAllPlayers().iterator();
        HeroManager hm = core.getHeroes().getHeroManager();
        Player nextPlayer = combatants.next();
        prompt(nextPlayer);
        HeroParty party = new HeroParty(hm.getHero(nextPlayer), core.getHeroes());
        while(combatants.hasNext()){
            nextPlayer = combatants.next();
            party.addMember(hm.getHero(nextPlayer));
            prompt(nextPlayer);
        }
        manager.put(arena.arenaName(), party);
    }

    @Override
    public void onArenaEnd(Arena arena){
        manager.remove(arena.arenaName());
    }

    @Override
    public void onPlayerDeath(Arena arena, Player player){
        manager.get(arena.arenaName()).removeMember(core.getHeroes().getHeroManager().getHero(player));
    }

    @Override
    public void onPlayerLeave(Arena arena, Player player){
        manager.get(arena.arenaName()).removeMember(core.getHeroes().getHeroManager().getHero(player));
    }

    private void prompt(Player player){
        player.sendMessage(ChatColor.BLUE+"[PU] "+
            ChatColor.GREEN+"You have been added to the MobArena Party!");
    }
}