AdminCommands.java

package org.mctourney.autoreferee.commands;

import java.io.IOException;
import java.util.Collections;
import java.util.List;

import org.bukkit.Bukkit;
import org.bukkit.ChatColor;
import org.bukkit.Location;
import org.bukkit.World;
import org.bukkit.command.BlockCommandSender;
import org.bukkit.command.CommandSender;
import org.bukkit.entity.Player;
import org.bukkit.entity.minecart.CommandMinecart;
import org.bukkit.OfflinePlayer;
import org.bukkit.plugin.Plugin;
import org.apache.commons.lang.StringUtils;
import org.mctourney.autoreferee.AutoRefMap;
import org.mctourney.autoreferee.AutoRefMatch;
import org.mctourney.autoreferee.AutoRefTeam;
import org.mctourney.autoreferee.AutoReferee;
import org.mctourney.autoreferee.event.match.MatchUnloadEvent;
import org.mctourney.autoreferee.event.player.PlayerTeamJoinEvent.Reason;
import org.mctourney.autoreferee.util.commands.AutoRefCommand;
import org.mctourney.autoreferee.util.commands.AutoRefPermission;
import org.mctourney.autoreferee.util.commands.CommandHandler;
import org.apache.commons.cli.CommandLine;

import com.google.common.collect.Lists;

public class AdminCommands implements CommandHandler
{
	AutoReferee plugin;

	public AdminCommands(Plugin plugin)
	{
		this.plugin = (AutoReferee) plugin;
	}

	@AutoRefCommand(name={"autoref", "world"}, argmin=1, argmax=1,
		description="Specifies the world for console commands to modify.",
		usage="<command> <world or player name>")
	@AutoRefPermission(console=true, nodes={"autoreferee.admin"})

	public boolean setConsoleWorld(CommandSender sender, AutoRefMatch match, String[] args, CommandLine options)
	{
		plugin.setConsoleWorld(args[0]);
		World world = plugin.getConsoleWorld();

		sender.sendMessage("Selected world: " + (world == null ? "<none>" : world.getName()));
		return world != null;
	}

	@AutoRefCommand(name={"autoref", "setlobby"}, argmax=0,
		description="Sets the current world to be the AutoReferee lobby world.")
	@AutoRefPermission(console=false, nodes={"autoreferee.admin"})

	public boolean setLobbyWorld(CommandSender sender, AutoRefMatch match, String[] args, CommandLine options)
	{
		if (match != null) return false;
		World lobby = ((Player) sender).getWorld();

		plugin.setLobbyWorld(lobby);
		sender.sendMessage(ChatColor.GREEN + lobby.getName() +
			" is the new AutoReferee lobby world.");
		return true;
	}

	@AutoRefCommand(name={"autoref", "setname"}, argmin=1, argmax=1,
			description="Set the name of the current match used in the match report.",
			usage="<command> <match title>")
	@AutoRefPermission(console=true, nodes={"autoreferee.admin"})

	public boolean setMatchName(CommandSender sender, AutoRefMatch match, String[] args, CommandLine options)
	{
		if (match == null) return false;

		match.setMatchName(args[0]);
		return true;
	}

	@AutoRefCommand(name={"autoref", "load"}, argmin=1, options="c+x",
		description="Loads a map by name, case insensitive.",
		usage="<command> <map name>",
		opthelp=
		{
			"c", "specify a custom world name",
			"x", "load world without AutoReferee",
		})
	@AutoRefPermission(console=true, nodes={"autoreferee.admin"})

	public boolean loadMap(CommandSender sender, AutoRefMatch match, String[] args, CommandLine options)
	{
		// generate a map name from the args
		String mapName = StringUtils.join(args, " ");

		// may specify a custom world name
		String customName = options.getOptionValue('c');

		// get world setup for match
		sender.sendMessage(ChatColor.GREEN + "Please wait...");
		AutoRefMap.loadMap(sender, mapName, customName);

		return true;
	}

	@AutoRefCommand(name={"autoref", "loadurl"}, argmin=1, argmax=1, options="c+x",
		description="Loads a map from a remote zip file, taking the URL as a parameter.",
		usage="<command> <url>",
		opthelp=
		{
			"c", "specify a custom world name",
			"x", "load world without AutoReferee",
		})
	@AutoRefPermission(console=true, nodes={"autoreferee.admin"})

	public boolean loadMapFromURL(CommandSender sender, AutoRefMatch match, String[] args, CommandLine options)
	{
		// may specify a custom world name
		String customName = options.getOptionValue('c');

		// get world setup for match
		sender.sendMessage(ChatColor.GREEN + "Please wait...");
		AutoRefMap.loadMapFromURL(sender, args[0], customName);

		return true;
	}

	@AutoRefCommand(name={"autoref", "unload"}, argmin=0, argmax=0,
		description="Unloads the current map. Connected players are either moved to the lobby or kicked.")
	@AutoRefPermission(console=true)

	public boolean unloadMap(CommandSender sender, AutoRefMatch match, String[] args, CommandLine options)
	{
		if (match == null)
		{
			sender.sendMessage(ChatColor.GRAY + "No world to unload.");
			return false;
		}
		
		if ((!match.isPracticeMode() || match.getCurrentState().isBeforeMatch())
				&& !sender.hasPermission("autoreferee.admin")) return false;
		
		if (match != null)
			match.destroy(MatchUnloadEvent.Reason.COMMAND);

		return true;
	}

	@AutoRefCommand(name={"autoref", "reload"}, options="x",
		description="Reloads the current map (or specified map) to its original, unmodified state. Players are migrated to the new copy.",
		usage="<command> [<map name>]",
		opthelp=
		{
			"x", "transfer players to same teams and location",
		})
	@AutoRefPermission(console=true)

	public boolean reloadMap(CommandSender sender, AutoRefMatch match, String[] args, CommandLine options)
		throws IOException
	{
		if (match == null) return false;

		// in practice mode, this command is available to anyone
		if ((!match.isPracticeMode() || match.getCurrentState().isBeforeMatch())
			&& !sender.hasPermission("autoreferee.admin")) return false;

		String name = match.getMapName();
		if (args.length > 0) name = StringUtils.join(args, ' ');

		AutoRefMap map = AutoRefMap.getMap(name);
		if (map == null)
		{
			sender.sendMessage(ChatColor.DARK_GRAY +
				"No archive of this map exists " + name);
			return true;
		}

		sender.sendMessage(ChatColor.DARK_GRAY +
			"Preparing a new copy of " + map.getVersionString());
		AutoRefMatch newmatch = AutoRefMap.createMatch(map, null);

		if (map.equals(match.getMap()) && options.hasOption('x'))
		{
			newmatch.setStartTime(match.getStartTime());
			newmatch.setCurrentState(match.getCurrentState());

			for (Player p : match.getWorld().getPlayers())
			{
				AutoRefTeam team = match.getPlayerTeam(p);
				if (team != null)
				{
					AutoRefTeam newteam = newmatch.getTeam(team.getDefaultName());
					newteam.join(p, Reason.AUTOMATIC, true);
				}

				// teleport the player to location in new map
				Location loc = p.getLocation();
				loc.setWorld(newmatch.getWorld());
				p.teleport(loc);
			}
		}
		else for (Player p : match.getWorld().getPlayers()) newmatch.joinMatch(p);

		match.destroy(MatchUnloadEvent.Reason.COMMAND);
		return true;
	}

	@AutoRefCommand(name={"autoref", "maplist"}, argmin=0, argmax=0,
		description="List all maps available, both on this server and in the repository.")
	@AutoRefPermission(console=true, nodes={"autoreferee.admin"})

	public boolean getMapList(CommandSender sender, AutoRefMatch match, String[] args, CommandLine options)
	{
		List<AutoRefMap> maps = Lists.newArrayList(AutoRefMap.getAvailableMaps());
		Collections.sort(maps);

		sender.sendMessage(ChatColor.GOLD + String.format("Available Maps (%d):", maps.size()));
		for (AutoRefMap mapInfo : maps)
		{
			ChatColor color = mapInfo.isInstalled() ? ChatColor.WHITE : ChatColor.DARK_GRAY;
			sender.sendMessage("* " + color + mapInfo.getVersionString());
		}

		return true;
	}

	@AutoRefCommand(name={"autoref", "update"}, argmin=0, argmax=0, options="f",
		description="Updates maps installed on server. Use -f to force an update.",
		opthelp=
		{
			"f", "force an update of all maps",
		})
	@AutoRefPermission(console=true, nodes={"autoreferee.admin"})

	public boolean updateMaps(CommandSender sender, AutoRefMatch match, String[] args, CommandLine options)
	{
		AutoRefMap.getUpdates(sender, options.hasOption('f'));
		return true;
	}

	@AutoRefCommand(name={"autoref", "autoinvite"}, argmin=1,
		description="Invite player(s) to participate in match. Works for offline players as well.",
		usage="<command> <player names...>")
	@AutoRefPermission(console=true, nodes={"autoreferee.admin"})

	public boolean autoInvite(CommandSender sender, AutoRefMatch match, String[] args, CommandLine options)
	{
		if (match != null) for (int i = 0; i < args.length; ++i)
		{
			// first, remove this player from all expected player lists
			OfflinePlayer opl = plugin.getServer().getOfflinePlayer(args[i]);
			for (AutoRefMatch m : plugin.getMatches()) m.removeExpectedPlayer(opl);

			// add them to the expected players list
			match.addExpectedPlayer(opl);

			// if this player cannot be found, skip
			Player invited = plugin.getServer().getPlayer(args[i]);
			if (invited == null) continue;

			// if this player is already busy competing in a match, skip
			AutoRefMatch m = plugin.getMatch(invited.getWorld());
			if (m != null && m.isPlayer(invited) && m.getCurrentState().inProgress()) continue;

			// otherwise, let's drag them in (no asking)
			match.joinMatch(invited);
		}

		return true;
	}

	@AutoRefCommand(name={"autoref", "pmsend"}, argmin=1,
		description="Send plugin message (for debugging client mods).",
		usage="<command> <plugin message>")
	@AutoRefPermission(console=false, nodes={"autoreferee.admin"})

	public boolean sendPluginMessage(CommandSender sender, AutoRefMatch match, String[] args, CommandLine options)
	{
		if (!(sender instanceof Player) || !sender.isOp()) return false;
		AutoRefMatch.messageReferee((Player) sender, StringUtils.join(args, " "));

		return true;
	}
	
	
	/*
	 * Replaces the vanilla /time command so time is set only in the sender's world.
	 */
	
	@AutoRefCommand(name={"time"}, argmin=2,
		description="Set the world time.",
		usage="§cUsage: /time <set|add> <value>")
	@AutoRefPermission(console=false, nodes={"bukkit.command.time"})

	public boolean setTime(CommandSender sender, AutoRefMatch match, String[] args, CommandLine options)
	{
		if (args[0].equalsIgnoreCase("set")) {
			int time = 0;
			
			if (args[1].equalsIgnoreCase("day")) {
				time = 1000;
			} else if (args[1].equalsIgnoreCase("night")) {
				time = 13000;
			} else if (args[1].matches("\\d+")) {
				time = Integer.parseInt(args[1]);
			} else {
				sender.sendMessage(ChatColor.RED + "'" + args[1] +"' is not a valid number");
				return true;
			}
			
			if (sender instanceof Player) {
				((Player) sender).getWorld().setTime(time);
				sender.sendMessage("Set the time to " + time);
			} else if (sender instanceof BlockCommandSender) {
				((BlockCommandSender) sender).getBlock().getWorld().setTime(time);
			} else if (sender instanceof CommandMinecart) {
				((CommandMinecart) sender).getWorld().setTime(time);	
			}
		}
		
		if (args[0].equalsIgnoreCase("add")) {
			if (args[1].matches("\\d+")) {
				int time = Integer.parseInt(args[1]);
				
				if (sender instanceof Player) {
					World world = ((Player) sender).getWorld();
					world.setTime(time + world.getTime());
					sender.sendMessage("Added " + time + " to the time");
				} else if (sender instanceof BlockCommandSender) {
					World world = ((BlockCommandSender) sender).getBlock().getWorld();
					world.setTime(time + world.getTime());
				} else if (sender instanceof CommandMinecart) {
					World world = ((CommandMinecart) sender).getWorld();
					world.setTime(time + world.getTime());
				}

			} else {
				sender.sendMessage(ChatColor.RED + "'" + args[1] +"' is not a valid number");
				return true;
			}
		}
		
		return true;
	}
}