Mineplex2018-withcommit/Plugins/Nautilus.Game.Arcade/src/nautilus/game/arcade/managers/GameHostManager.java

package nautilus.game.arcade.managers;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;

import mineplex.core.common.Rank;
import mineplex.core.common.util.C;
import mineplex.core.common.util.F;
import mineplex.core.common.util.UtilGear;
import mineplex.core.common.util.UtilPlayer;
import mineplex.core.common.util.UtilServer;
import mineplex.core.common.util.UtilTextBottom;
import mineplex.core.common.util.UtilTime;
import mineplex.core.game.GameCategory;
import mineplex.core.itemstack.ItemStackFactory;
import mineplex.core.updater.UpdateType;
import mineplex.core.updater.event.UpdateEvent;
import nautilus.game.arcade.ArcadeManager;
import nautilus.game.arcade.GameType;
import nautilus.game.arcade.game.Game.GameState;
import nautilus.game.arcade.gui.privateServer.PrivateServerShop;
import nautilus.game.arcade.gui.privateServer.page.GameVotingPage;

import org.bukkit.Bukkit;
import org.bukkit.ChatColor;
import org.bukkit.GameMode;
import org.bukkit.Material;
import org.bukkit.Sound;
import org.bukkit.entity.Player;
import org.bukkit.event.EventHandler;
import org.bukkit.event.HandlerList;
import org.bukkit.event.Listener;
import org.bukkit.event.inventory.InventoryType;
import org.bukkit.event.player.PlayerCommandPreprocessEvent;
import org.bukkit.event.player.PlayerInteractEvent;
import org.bukkit.event.player.PlayerJoinEvent;
import org.bukkit.event.player.PlayerLoginEvent;
import org.bukkit.event.player.PlayerQuitEvent;
import org.bukkit.plugin.Plugin;

public class GameHostManager implements Listener
{
	private ArrayList<GameType> ultraGames = new ArrayList<GameType>();
	private ArrayList<GameType> heroGames = new ArrayList<GameType>();
	private ArrayList<GameType> legendGames = new ArrayList<GameType>();

	ArcadeManager Manager;
	
	private Player _host;
	private String _hostName;
	private Rank _hostRank;
	private long _serverStartTime = System.currentTimeMillis();
	private long _serverExpireTime = 21600000;
	private long _lastOnline = System.currentTimeMillis();
	private long _expireTime = 300000;
	private boolean _hostExpired = false;

	private HashSet<Player> _onlineAdmins = new HashSet<Player>();
	private HashSet<String> _adminList = new HashSet<String>();
	private HashSet<String> _whitelist = new HashSet<String>();
	private HashSet<String> _blacklist = new HashSet<String>();

	private PrivateServerShop _shop;
	
	private boolean _isEventServer = false;
	
	private HashMap<Player, Boolean> _permissionMap = new HashMap<Player, Boolean>();

	private boolean _voteInProgress = false;
	private HashMap<String, GameType> _votes = new HashMap<String, GameType>();
	private int _voteNotificationStage = 1;
	
	public GameHostManager(ArcadeManager manager)
	{
		Manager = manager;
		_shop = new PrivateServerShop(manager, manager.GetClients(), manager.GetDonation());
		Manager.getPluginManager().registerEvents(this, Manager.getPlugin());
		
		//Ultra Games
		ultraGames.add(GameType.Smash);
		
		ultraGames.add(GameType.BaconBrawl);
		ultraGames.add(GameType.DeathTag);
		ultraGames.add(GameType.DragonEscape);
		ultraGames.add(GameType.Dragons);
		ultraGames.add(GameType.Micro);
		ultraGames.add(GameType.Paintball);
		ultraGames.add(GameType.Quiver);
		ultraGames.add(GameType.Runner);
		ultraGames.add(GameType.Sheep);
		ultraGames.add(GameType.Snake);
		ultraGames.add(GameType.SneakyAssassins);
		ultraGames.add(GameType.TurfWars);
		ultraGames.add(GameType.Spleef);
		ultraGames.add(GameType.Lobbers);
		ultraGames.add(GameType.Evolution);
		ultraGames.add(GameType.MonsterMaze);

		//Hero Games
		heroGames.add(GameType.ChampionsDominate);
		heroGames.add(GameType.ChampionsTDM);
		heroGames.add(GameType.HideSeek);
		heroGames.add(GameType.Draw);

		//Legend Games
		legendGames.add(GameType.Bridge);
		legendGames.add(GameType.SurvivalGames);
		legendGames.add(GameType.CastleSiege);
		legendGames.add(GameType.WitherAssault);
		legendGames.add(GameType.Wizards);
		legendGames.add(GameType.Build);
		legendGames.add(GameType.UHC);
		legendGames.add(GameType.MineStrike);
		legendGames.add(GameType.Skywars);
		// Team variants - Currently being remade.
		/*
		legendGames.add(GameType.DragonEscapeTeams);
		legendGames.add(GameType.DragonsTeams);
		legendGames.add(GameType.QuiverTeams);
		legendGames.add(GameType.SmashTeams);
		legendGames.add(GameType.SpleefTeams);
		legendGames.add(GameType.SurvivalGamesTeams);
		*/
		//Rejected / Other
		legendGames.add(GameType.MilkCow);
		legendGames.add(GameType.SearchAndDestroy);
		legendGames.add(GameType.ZombieSurvival);
		legendGames.add(GameType.SurvivalGamesTeams);
		legendGames.add(GameType.SkywarsTeams);
		legendGames.add(GameType.SmashTeams);
		legendGames.add(GameType.SnowFight);
		legendGames.add(GameType.Gravity);
		legendGames.add(GameType.Barbarians);
		legendGames.add(GameType.SmashDomination);

		//Config Defaults
		if (Manager.GetHost() != null && Manager.GetHost().length() > 0)
		{
			setDefaultConfig();
		}
	}

	public ArrayList<GameType> hasWarning()
	{
		ArrayList<GameType> games = new ArrayList<>();
		games.add(GameType.MilkCow);
		games.add(GameType.SearchAndDestroy);
		games.add(GameType.ZombieSurvival);
		return games;
	}

	@EventHandler
	public void updateHost(UpdateEvent event)
	{
		if (event.getType() != UpdateType.FAST)
			return;
		
		//No Host - Not MPS
		if (Manager.GetHost() == null || Manager.GetHost().length() == 0)
			return;
		
		// Set as event server
		if (Manager.GetGame() != null && Manager.GetGame().GetType() == GameType.Event)
		{
			setEventServer(true);
		}
		
		// Admins update
		for (Player player : UtilServer.getPlayers())
		{
			if (player.equals(_host) || _adminList.contains(player.getName()) || Manager.GetClients().Get(player).GetRank().has(Rank.ADMIN))
			{
				if (Manager.GetGame() == null || Manager.GetGame().GetState() == GameState.Recruit)
					giveAdminItem(player);
			}
			
			if (player.equals(_host) ||	(isAdmin(player, false) && isEventServer()))
				_lastOnline = System.currentTimeMillis();
		}
	}

	@EventHandler
	public void voteNotification(UpdateEvent e)
	{
		if (e.getType() != UpdateType.FAST)
			return;

		if (!_voteInProgress)
			return;

		if (_voteNotificationStage == 1)
		{
			UtilTextBottom.display(C.cYellow + C.Bold + "Type " + C.cGold +  C.Bold + "/vote" + C.cYellow + C.Bold + " to vote for next game", UtilServer.getPlayers());
			_voteNotificationStage++;
			return;
		}
		else if (_voteNotificationStage == 2)
		{
			UtilTextBottom.display(C.cGold + C.Bold + "Type " + C.cYellow +  C.Bold + "/vote" + C.cGold + C.Bold + " to vote for next game", UtilServer.getPlayers());
			_voteNotificationStage = 1;
			return;
		}
	}

	@EventHandler
	public void whitelistJoin(PlayerLoginEvent event)
	{
		Player p = event.getPlayer();
		if (Manager.GetServerConfig().PlayerServerWhitelist){
			if (!getWhitelist().contains(p.getName())){
				if ((Manager.GetHost() != null) && (Manager.GetHost().equalsIgnoreCase(p.getName().toLowerCase())))
				{
					return;
				}
				event.disallow(PlayerLoginEvent.Result.KICK_WHITELIST, "This MPS is whitelisted.");
			}
		}
	}

	@EventHandler
	public void adminJoin(PlayerJoinEvent event)
	{
		if (!isPrivateServer())
			return;

		if (Manager.GetHost().equals(event.getPlayer().getName()))
		{
			_host = event.getPlayer();
			_hostRank = Manager.GetClients().Get(_host).GetRank();
			System.out.println("Game Host Joined.");
			
			//Limit player count!
			if (Manager.GetServerConfig().MaxPlayers > getMaxPlayerCap())
				Manager.GetServerConfig().MaxPlayers = getMaxPlayerCap();
			
			if (isEventServer())
				worldeditPermissionSet(event.getPlayer(), true);
		}
		else if (isAdmin(event.getPlayer(), false))
		{
			System.out.println("Admin Joined.");
			_onlineAdmins.add(event.getPlayer());
			
			if (isEventServer())
				worldeditPermissionSet(event.getPlayer(), true);
		}
	}
	
	@EventHandler
	public void adminQuit(PlayerQuitEvent event)
	{
		if (!isPrivateServer())
			return;
		
		if (isHost(event.getPlayer()))
		{
			System.out.println("Game Host Quit.");
			_host = null;
			
			if (isEventServer())
				worldeditPermissionSet(event.getPlayer(), false);
		}
		else if (isAdmin(event.getPlayer(), false))
		{
			_onlineAdmins.remove(event.getPlayer());
			
			if (isEventServer())
				worldeditPermissionSet(event.getPlayer(), false);
		}
	}
		
	public void worldeditPermissionSet(Player player, boolean hasPermission)
	{
		if (!_permissionMap.containsKey(player) || _permissionMap.get(player) != hasPermission)
		{
			for (Plugin plugin : Bukkit.getPluginManager().getPlugins())
			{
				player.addAttachment(plugin, "worldedit.*", hasPermission);
			}
			
			_permissionMap.put(player, hasPermission);
			
			UtilPlayer.message(player, "World Edit Permissions: " + F.tf(hasPermission));
		}
	}
	
	
	@EventHandler
	public void updateHostExpired(UpdateEvent event)
	{
		if (!isPrivateServer())
			return;
		
		if (event.getType() != UpdateType.FAST)
			return;
	
		if (Manager.GetGame() != null && Manager.GetGame().GetState() != GameState.Recruit)
			return;
		
		if (_hostExpired)
			return;
		
		if (UtilTime.elapsed(_lastOnline, _expireTime))
			setHostExpired(true, Manager.GetServerConfig().HostName + " has abandoned the server. Thanks for playing!");
		
		else if (UtilTime.elapsed(_serverStartTime, _serverExpireTime))
			setHostExpired(true, "This server has expired! Thank you for playing!");
	}
	
	
	public boolean isHostExpired()
	{
		if (!isPrivateServer())
			return false;
		
		return _hostExpired;
	}

	public void setHostExpired(boolean expired, String string)
	{
		for (Player other : UtilServer.getPlayers())
		{
			UtilPlayer.message(other, C.cGold + C.Bold + string);
			other.playSound(other.getLocation(), Sound.ENDERDRAGON_GROWL, 10f, 1f);
			Manager.GetPortal().sendPlayerToServer(other, "Lobby");
		}
		
		_hostExpired = expired;
	}
	
	private void giveAdminItem(Player player)
	{
		if (Manager.GetGame() == null)
			return;

		if (UtilGear.isMat(player.getInventory().getItem(8), Material.SPECKLED_MELON))
			return;
		
		if (player.getOpenInventory().getType() != InventoryType.CRAFTING &&
			player.getOpenInventory().getType() != InventoryType.CREATIVE)
			return;
		
		player.getInventory().setItem(8, ItemStackFactory.Instance.CreateStack(Material.SPECKLED_MELON, (byte)0, 1, C.cGreen + C.Bold + "/menu"));
	}

	private void removeAdminItem(Player player)
	{
		if (player.getInventory().getItem(8) != null && player.getInventory().getItem(8).getType() == Material.SPECKLED_MELON)
		{
			player.getInventory().setItem(8, null);
		}
	}
	
	public HashSet<String> getWhitelist()
	{
		return _whitelist;
	}
	
	public HashSet<String> getBlacklist()
	{
		return _blacklist;
	}

	public HashSet<String> getAdminList()
	{
		return _adminList;
	}

	@EventHandler
	public void getAdmin(PlayerCommandPreprocessEvent event)
	{
		if (!event.getMessage().equalsIgnoreCase("/getadmin"))
			return;

		event.setCancelled(true);

		if (!isPrivateServer())
			return;

		if (!Manager.GetClients().Get(event.getPlayer()).GetRank().has(event.getPlayer(), Rank.JNR_DEV, true)  && !event.getPlayer().isOp())
			return;

		giveAdmin(event.getPlayer());
	}

	@EventHandler
	public void broadcastCommand(PlayerCommandPreprocessEvent event)
	{
		if (!event.getMessage().toLowerCase().startsWith("/bc"))
			return;

		if (!isPrivateServer())
			return;

		if (!isAdmin(event.getPlayer(), true))
		{
			event.getPlayer().sendMessage(F.main("Broadcast", "Only MPS admins can use this command."));
			event.setCancelled(true);
			return;
		}

		event.setCancelled(true);

		if (event.getMessage().split(" ").length < 2)
		{
			event.getPlayer().sendMessage(F.main("Broadcast", "/bc <message>"));
			return;
		}

		String msg = "";
		for (int i = 1; i < event.getMessage().split(" ").length; i++)
		{
			msg += event.getMessage().split(" ")[i] + " ";
		}
		msg = msg.trim();

		Bukkit.broadcastMessage(C.cDGreen + C.Bold + event.getPlayer().getName() + " " + C.cGreen + msg);
	}

	@EventHandler
	public void voteCommand(PlayerCommandPreprocessEvent event)
	{
		if (!event.getMessage().toLowerCase().startsWith("/vote"))
			return;

		if (!isPrivateServer())
		{
			UtilPlayer.message(event.getPlayer(), F.main("Vote", "This command is only available on private servers."));
			event.setCancelled(true);
			return;
		}

		if (!_voteInProgress)
		{
			UtilPlayer.message(event.getPlayer(), F.main("Vote", "There is no vote in progress."));
			event.setCancelled(true);
			return;
		}

		event.setCancelled(true);
		_shop.openPageForPlayer(event.getPlayer(), new GameVotingPage(Manager, _shop, event.getPlayer()));
		return;
	}
	
	@EventHandler
	public void menuCommand(PlayerCommandPreprocessEvent event)
	{
		if (!event.getMessage().toLowerCase().startsWith("/menu"))
			return;

		if (!isPrivateServer())
			return;
		
		if (!isAdmin(event.getPlayer(), true))
			return;
		
		event.setCancelled(true);
		openMenu(event.getPlayer());
	}
	
	@EventHandler
	public void menuInteract(PlayerInteractEvent event)
	{
		if (!isPrivateServer())
			return;
		
		if (!isAdmin(event.getPlayer(), true))
			return;
		
		if (!UtilGear.isMat(event.getPlayer().getItemInHand(), Material.SPECKLED_MELON))
			return;
		
		openMenu(event.getPlayer());
		event.setCancelled(true);
	}
	
	private void openMenu(Player player)
	{
		_shop.attemptShopOpen(player);
	}

	public boolean isAdmin(Player player, boolean includeStaff)
	{
		return player.equals(_host) || _adminList.contains(player.getName()) || (includeStaff && Manager.GetClients().Get(player).GetRank().has(Rank.ADMIN));
	}

	public boolean isHost(Player player)
	{
		return player.getName().equals(Manager.GetHost());
	}

	public boolean isPrivateServer()
	{
		return Manager.GetHost() != null && Manager.GetHost().length() > 0;
	}

	@EventHandler
	public void whitelistCommand(PlayerCommandPreprocessEvent event)
	{
		if (_host == null || !event.getPlayer().equals(_host))
			return;
		
		if (!event.getMessage().toLowerCase().startsWith("/whitelist"))
			return;

		if (!isPrivateServer())
			return;

		event.setCancelled(true);
		
		String[] args = event.getMessage().split(" ");
		
		for (int i=1 ; i<args.length ; i++)
		{
			String name = args[i].toLowerCase();
			
			if (_whitelist.add(name))
			{
				UtilPlayer.message(event.getPlayer(), F.main("Host", "Added " + F.elem(args[i]) + " to the whitelist."));
			}
		}
	}
	
	public void setGame(GameType type)
	{
		if (_host == null)
			return;
		
		Manager.GetGameCreationManager().SetNextGameType(type);
		
		//End Current
		if (Manager.GetGame().GetState() == GameState.Recruit)
		{
			Manager.GetGame().SetState(GameState.Dead);
			
			Bukkit.broadcastMessage(C.cGreen + C.Bold + _host.getName() + " has changed game to " + type.GetName() + ".");
		}
		else
		{
			Bukkit.broadcastMessage(C.cGreen + C.Bold + _host.getName() + " set next game to " + type.GetName() + ".");
		}
	}
	
	public void startGame()
	{
		if (_host == null)
			return;
		
		Manager.GetGameManager().StateCountdown(Manager.GetGame(), 10, true);

		Manager.GetGame().Announce(C.cGreen + C.Bold + _host.getName() + " has started the game.");
	}
	
	public void stopGame()
	{
		if (_host == null)
			return;
		
		if (Manager.GetGame() == null)
			return; 

		HandlerList.unregisterAll(Manager.GetGame());
		
		if (Manager.GetGame().GetState() == GameState.End || Manager.GetGame().GetState() == GameState.End)
		{
			_host.sendMessage("Game is already ending..."); 
			return;
		}
		else if (Manager.GetGame().GetState() == GameState.Recruit)
		{
			Manager.GetGame().SetState(GameState.Dead);
		}
		else
		{
			Manager.GetGame().SetState(GameState.End);
		}


		Manager.GetGame().Announce(C.cGreen + C.Bold + _host.getName() + " has stopped the game.");
	}

	public boolean hasRank(Rank rank)
	{
		if (_hostRank == null)
			return false;
		
		return _hostRank.has(rank);
	}

	public ArrayList<GameType> getAvailableGames(Player player)
	{
		ArrayList<GameType> games = new ArrayList<GameType>();

		if (hasRank(Rank.ULTRA))
			games.addAll(ultraGames);
		if (hasRank(Rank.HERO))
			games.addAll(heroGames);
		if (hasRank(Rank.LEGEND))
			games.addAll(legendGames);

		return games;
	}

	public HashMap<GameCategory, ArrayList<GameType>> getGames(Player p)
	{
		HashMap<GameCategory, ArrayList<GameType>> games = new HashMap<GameCategory, ArrayList<GameType>>();
		for (GameCategory cat : GameCategory.values())
		{
			ArrayList<GameType> types = new ArrayList<>();
			for (GameType type : getAvailableGames(p))
			{
				if (type.getGameCategory().equals(cat))
				{
					types.add(type);
				}
			}
			games.put(cat, types);
		}
		return games;
	}

	public void ban(Player player)
	{
		_blacklist.add(player.getName());

		Manager.GetPortal().sendToHub(player, "You were removed from this Mineplex Private Server.");
	}
	
	@EventHandler
	public void kickBlacklist(UpdateEvent event)
	{
		if (event.getType() != UpdateType.SEC)
			return;
		
		for (Player player : UtilServer.getPlayers())
		{
			if (_blacklist.contains(player.getName()))
			{
				Manager.GetPortal().sendToHub(player, "You were removed from this Mineplex Private Server.");
			}
		}
	}

	public void giveAdmin(Player player)
	{
		_adminList.add(player.getName());
		_onlineAdmins.add(player);
		UtilPlayer.message(player, F.main("Server", "You were given Co-Host privileges."));
		
		if (isEventServer())
			worldeditPermissionSet(player, true);
	}

	public void removeAdmin(String playerName)
	{
		_adminList.remove(playerName);
		Player player = UtilPlayer.searchExact(playerName);
		if (player != null)
		{
			_onlineAdmins.remove(player);
			removeAdminItem(player);
			if (_shop.isPlayerInShop(player))
			{
				player.closeInventory();
			}
			UtilPlayer.message(player, F.main("Server", "Your Co-Host privileges were removed."));
			
			player.setGameMode(GameMode.SURVIVAL);
			
			if (isEventServer())
				worldeditPermissionSet(player, false);
		}
	}
	
	public boolean isAdminOnline()
	{
		return _onlineAdmins.isEmpty();
	}
	
	public void setDefaultConfig()
	{
		Manager.GetServerConfig().HotbarInventory = false;
		
		Manager.GetServerConfig().RewardAchievements = false;
		Manager.GetServerConfig().RewardGems = false;
		Manager.GetServerConfig().RewardItems = false;
		Manager.GetServerConfig().RewardStats = false;
		
		Manager.GetServerConfig().GameAutoStart = true;
		Manager.GetServerConfig().GameTimeout = true;
		Manager.GetServerConfig().PlayerKickIdle = true;
		Manager.GetServerConfig().TeamForceBalance = true;
	}

	public int getMaxPlayerCap()
	{
		if (hasRank(Rank.SNR_MODERATOR) || _hostRank == Rank.YOUTUBE || _hostRank == Rank.TWITCH)
			return 100;
		else if (hasRank(Rank.LEGEND))
			return 40;
		else if (hasRank(Rank.HERO))
			return 12;
		else
			return 4;
	}
		
	@EventHandler
	public void setEventGame(PlayerCommandPreprocessEvent event)
	{
		if (!isEventServer() || Manager.GetGame() == null)
			return;
		
		if (!isAdmin(event.getPlayer(), false))
			return;
		
		if (!event.getMessage().toLowerCase().startsWith("/e set ") && !event.getMessage().toLowerCase().equals("/e set"))
			return;
		
		Player caller = event.getPlayer();
		String[] args = event.getMessage().split(" ");
		
		String game = args[2].toLowerCase();
		
		if (args.length >= 4)
		{
			String map = "";
			String source = "";
			if(args.length == 5)
			{
				Manager.GetGameCreationManager().MapSource = args[3];
				Manager.GetGameCreationManager().MapPref = args[4];
				source = args[3];
				map = args[4]; 
			} 
			else 
			{
				Manager.GetGameCreationManager().MapSource = args[2];
				Manager.GetGameCreationManager().MapPref = args[3];
				source = args[2];
				map = args[3];
			}
			UtilPlayer.message(caller, C.cAqua + C.Bold + "Map Preference: " + ChatColor.RESET + source + ":" + map);
		}
		
		//Parse Game
		ArrayList<GameType> matches = new ArrayList<>();
		for (GameType type : GameType.values())
		{
			if (type.toString().toLowerCase().equals(game))
			{
				matches.clear();
				matches.add(type);
				break;
			}
			
			if (type.toString().toLowerCase().contains(game))
			{
				matches.add(type);
			}
		}
		
		if (matches.size() == 0)
		{
			caller.sendMessage("No results for: " + game);
			return;
		}
		
		if (matches.size() > 1)
		{
			caller.sendMessage("Matched multiple games;");
			for (GameType cur : matches)
				caller.sendMessage(cur.toString());
			return;
		}
		
		GameType type = matches.get(0);
		Manager.GetGame().setGame(type, event.getPlayer(), true);
		
		event.setCancelled(true);
	}

	@EventHandler
	public void playerJoin(PlayerJoinEvent event)
	{
		if (!isPrivateServer())
			return;

		if (_blacklist.contains(event.getPlayer().getName()))
		{
			Manager.GetPortal().sendToHub(event.getPlayer(), "You were kicked from this Mineplex Private Server.");
			return;
		}
		
		String serverName = Manager.getPlugin().getConfig().getString("serverstatus.name");
		UtilPlayer.message(event.getPlayer(), ChatColor.BOLD + "Welcome to Mineplex Private Servers!");
		UtilPlayer.message(event.getPlayer(), C.Bold + "Friends can connect with " + C.cGreen + C.Bold + "/server " + serverName);
	}
	
	public boolean isEventServer()
	{
		return _isEventServer;
	}
	
	public void setEventServer(boolean var)
	{
		_isEventServer = var;
	}

	public HashMap<String, GameType> getVotes()
	{
		return _votes;
	}

	public void setVoteInProgress(boolean voteInProgress)
	{
		_voteInProgress = voteInProgress;
	}

	public boolean isVoteInProgress()
	{
		return _voteInProgress;
	}
	
	public Rank getHostRank()
	{
		return _hostRank;
	}
	
	public void setHostRank(Rank rank)
	{
		_hostRank = rank;
	}
	
	public Player getHost()
	{
		return _host;
	}
	
	public void setHost(Player player)
	{
		_host = player;
	}
	
}