TFM-4.3-Reloaded/src/main/java/me/StevenLawson/TotalFreedomMod/commands/Command_lockup.java

package me.StevenLawson.TotalFreedomMod.commands;

import me.StevenLawson.TotalFreedomMod.player.PlayerData;
import me.StevenLawson.TotalFreedomMod.util.Utilities;
import org.bukkit.command.Command;
import org.bukkit.command.CommandSender;
import org.bukkit.entity.Player;
import org.bukkit.scheduler.BukkitRunnable;
import org.bukkit.scheduler.BukkitTask;

@CommandPermissions(level = AdminLevel.SENIOR, source = SourceType.ONLY_CONSOLE, blockHostConsole = true)
public class Command_lockup extends FreedomCommand {

	@Override
	public boolean run(CommandSender sender, org.bukkit.entity.Player sender_p, Command cmd,
		String commandLabel, String[] args, boolean senderIsConsole) {
		if (args.length == 1) {
			if (args[0].equalsIgnoreCase("all")) {
				Utilities.adminAction(sender.getName(), "Locking up all players", true);

				for (Player player : server.getOnlinePlayers()) {
					startLockup(player);
				}
				playerMsg(sender, "Locked up all players.");
			} else if (args[0].equalsIgnoreCase("purge")) {
				Utilities.adminAction(sender.getName(), "Unlocking all players", true);
				for (Player player : server.getOnlinePlayers()) {
					cancelLockup(player);
				}

				playerMsg(sender, "Unlocked all players.");
			} else {
				return false;
			}
		} else if (args.length == 2) {
			if (args[1].equalsIgnoreCase("on")) {
				final Player player = getPlayer(args[0]);

				if (player == null) {
					sender.sendMessage(FreedomCommand.PLAYER_NOT_FOUND);
					return true;
				}

				Utilities.adminAction(sender.getName(), "Locking up " + player.getName(), true);
				startLockup(player);
				playerMsg(sender, "Locked up " + player.getName() + ".");
			} else if ("off".equals(args[1])) {
				final Player player = getPlayer(args[0]);

				if (player == null) {
					sender.sendMessage(FreedomCommand.PLAYER_NOT_FOUND);
					return true;
				}

				Utilities.adminAction(sender.getName(), "Unlocking " + player.getName(), true);
				cancelLockup(player);
				playerMsg(sender, "Unlocked " + player.getName() + ".");
			} else {
				return false;
			}
		} else {
			return false;
		}

		return true;
	}

	private void cancelLockup(PlayerData playerdata) {
		BukkitTask lockupScheduleID = playerdata.getLockupScheduleID();
		if (lockupScheduleID != null) {
			lockupScheduleID.cancel();
			playerdata.setLockupScheduleID(null);
		}
	}

	private void cancelLockup(final Player player) {
		cancelLockup(PlayerData.getPlayerData(player));
	}

	private void startLockup(final Player player) {
		final PlayerData playerdata = PlayerData.getPlayerData(player);

		cancelLockup(playerdata);

		playerdata.setLockupScheduleID(new BukkitRunnable() {
			@Override
			public void run() {
				if (player.isOnline()) {
					player.openInventory(player.getInventory());
				} else {
					cancelLockup(playerdata);
				}
			}
		}.runTaskTimer(plugin, 0L, 5L));
	}
}