VotingManager.java

/*******************************************************************************
 * Copyright 2014 stuntguy3000 (Luke Anderson) and coasterman10.
 *  
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA 02110-1301, USA.
 ******************************************************************************/
package org.eodsteven.CrafterNexus.manager;

import java.util.HashMap;
import org.bukkit.Bukkit;
import org.bukkit.ChatColor;
import org.bukkit.command.CommandSender;
import org.eodsteven.CrafterNexus.CrafterNexus;

public class VotingManager {
    private final CrafterNexus plugin;
    private final HashMap<Integer, String> maps = new HashMap<Integer, String>();
    private final HashMap<String, String> votes = new HashMap<String, String>();
    private boolean running = false;

    public VotingManager(CrafterNexus plugin) {
        this.plugin = plugin;
    }

    public void start() {
        maps.clear();
        votes.clear();
        int count = 0;
        int size = plugin.getMapManager().getRandomMaps().size();
        for (String map : plugin.getMapManager().getRandomMaps()) {
            count++;
            size--;
            maps.put(count, map);
            plugin.getScoreboardHandler().scores.put(map, plugin
                    .getScoreboardHandler().obj.getScore(Bukkit
                    .getOfflinePlayer(map)));
            plugin.getScoreboardHandler().scores.get(map).setScore(size);
            plugin.getScoreboardHandler().teams.put(map,
                    plugin.getScoreboardHandler().sb.registerNewTeam(map));
            plugin.getScoreboardHandler().teams.get(map).addPlayer(
                    Bukkit.getOfflinePlayer(map));
            plugin.getScoreboardHandler().teams.get(map).setPrefix(
                    ChatColor.AQUA + "[" + count + "] " + ChatColor.GRAY);
            plugin.getScoreboardHandler().teams.get(map).setSuffix(
                    ChatColor.RED + " » " + ChatColor.GREEN + "0 Votes");
        }

        running = true;

        plugin.getScoreboardHandler().update();
    }

    public boolean vote(CommandSender voter, String vote) {
        try {
            int val = Integer.parseInt(vote);

            if (maps.containsKey(val)) {
                vote = maps.get(val);
                for (String map : maps.values()) {
                    if (vote.equalsIgnoreCase(map)) {
                        votes.put(voter.getName(), map);
                        voter.sendMessage(ChatColor.GOLD + "You voted for "
                                + ChatColor.WHITE + map);
                        updateScoreboard();
                        return true;
                    }
                }
            }
        } catch (Exception ex) {
            for (String map : maps.values()) {
                if (vote.equalsIgnoreCase(map)) {
                    votes.put(voter.getName(), map);
                    voter.sendMessage(ChatColor.GOLD + "You voted for "
                            + ChatColor.WHITE + map);
                    updateScoreboard();
                    return true;
                }
            }
        }

        voter.sendMessage(vote + ChatColor.RED + " is not a valid map");
        return false;
    }

    public String getWinner() {
        String winner = null;
        Integer highest = -1;
        for (String map : maps.values()) {
            int totalVotes = countVotes(map);
            if (totalVotes > highest) {
                winner = map;
                highest = totalVotes;
            }
        }
        return winner;
    }

    public void end() {
        running = false;
    }

    public boolean isRunning() {
        return running;
    }

    public HashMap<Integer, String> getMaps() {
        return maps;
    }

    private int countVotes(String map) {
        int total = 0;
        for (String vote : votes.values())
            if (vote.equals(map))
                total++;
        return total;
    }

    private void updateScoreboard() {
        for (String map : maps.values())
            plugin.getScoreboardHandler().teams.get(map).setSuffix(
                    ChatColor.RED + " » " + ChatColor.GREEN + countVotes(map)
                            + " Vote" + (countVotes(map) == 1 ? "" : "s"));
    }
}