ChromaCore/src/main/java/buttondevteam/lib/TBMCPlayer.java

package buttondevteam.lib;

import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map.Entry;
import java.util.UUID;

import org.bukkit.Bukkit;
import org.bukkit.OfflinePlayer;
import org.bukkit.configuration.file.YamlConfiguration;
import com.palmergames.bukkit.towny.Towny;
import com.palmergames.bukkit.towny.object.Resident;
import com.palmergames.bukkit.towny.object.TownyUniverse;

/**
 * <p>
 * The class for holding data common to all TBMC plugins
 * </p>
 * <p>
 * Use {@link #asPluginPlayer(Class)} to get plugin-specific data
 * </p>
 * 
 * @author Norbi
 *
 */
public class TBMCPlayer implements AutoCloseable {
	private static final String TBMC_PLAYERS_DIR = "TBMC/players";

	private HashMap<String, Object> data = new HashMap<>();

	/**
	 * <p>
	 * Gets a player data entry for the caller plugin returning the desired type.<br>
	 * <i>It will automatically determine the key and the return type.</i><br>
	 * Usage:
	 * </p>
	 * 
	 * <pre>
	 * {@code
	 * public String getPlayerName() {
	 * 	return getData();
	 * }
	 * </pre>
	 * 
	 * @return The value or null if not found
	 */
	@SuppressWarnings("unchecked")
	protected <T> T getData() {
		StackTraceElement st = new Exception().getStackTrace()[1];
		String mname = st.getMethodName();
		if (!mname.startsWith("get"))
			throw new UnsupportedOperationException("Can only use getData from a getXYZ method");
		Object ret = getLoadedPlayers().get(uuid).data.get(mname.substring("get".length()).toLowerCase());
		if (ret != null && Integer.class.isAssignableFrom(ret.getClass()))
			throw new UnsupportedOperationException("For integers use getIntData()");
		return (T) ret;
	}

	/**
	 * Sets a player data entry <i>based on the caller method</i><br>
	 * Usage:
	 * </p>
	 * 
	 * <pre>
	 * {@code
	 * public String setPlayerName(String value) {
	 * 	return setData(value);
	 * }
	 * </pre>
	 * 
	 * @param value
	 *            The value to set
	 */
	protected void setData(Object value) {
		StackTraceElement st = new Exception().getStackTrace()[1];
		String mname = st.getMethodName();
		if (!mname.startsWith("set"))
			throw new UnsupportedOperationException("Can only use setData from a setXYZ method");
		getLoadedPlayers().get(uuid).data.put(mname.substring("set".length()).toLowerCase(), value);
	}

	/**
	 * <p>
	 * Gets a player data entry for the caller plugin returning the desired type, <b>which is an enum</b><br>
	 * <i>It will automatically determine the key and the return type.</i><br>
	 * Usage:
	 * </p>
	 * 
	 * <pre>
	 * {@code
	 * public String getSomeEnum() {
	 * 	return getEnumData();
	 * }
	 * </pre>
	 * 
	 * @return The value or null if not found
	 */
	protected <T extends Enum<T>> T getEnumData(Class<T> cl) {
		StackTraceElement st = new Exception().getStackTrace()[1];
		String mname = st.getMethodName();
		if (!mname.startsWith("get"))
			throw new UnsupportedOperationException("Can only use getEnumData from a getXYZ method");
		final String retstr = (String) getLoadedPlayers().get(uuid).data
				.get(mname.substring("get".length()).toLowerCase());
		if (retstr != null)
			return Enum.valueOf(cl, retstr);
		else
			return null;
	}

	/**
	 * Sets a player data entry <i>based on the caller method</i><br>
	 * Usage:
	 * </p>
	 * 
	 * <pre>
	 * {@code
	 * public String setSomeEnum(SomeEnum value) {
	 * 	return setEnumData(value);
	 * }
	 * </pre>
	 * 
	 * @param value
	 *            The value to set
	 */
	protected void setEnumData(Enum<?> value) {
		StackTraceElement st = new Exception().getStackTrace()[1];
		String mname = st.getMethodName();
		if (!mname.startsWith("set"))
			throw new UnsupportedOperationException("Can only use setEnumData from a setXYZ method");
		getLoadedPlayers().get(uuid).data.put(mname.substring("set".length()).toLowerCase(), value.toString());
	}

	/**
	 * <p>
	 * Gets a player data entry for the caller plugin returning the desired type, <b>which is a number</b><br>
	 * <i>It will automatically determine the key and the return type.</i><br>
	 * Usage:
	 * </p>
	 * 
	 * <pre>
	 * {@code
	 * public short getNumber() {
	 * 	return getIntData();
	 * }
	 * </pre>
	 * 
	 * @return The value or null if not found
	 */
	@SuppressWarnings("unchecked")
	protected <T extends Number> T getIntData(Class<T> cl) {
		StackTraceElement st = new Exception().getStackTrace()[1];
		String mname = st.getMethodName();
		if (!mname.startsWith("get"))
			throw new UnsupportedOperationException("Can only use getIntData from a getXYZ method");
		Object obj = getLoadedPlayers().get(uuid).data.get(mname.substring("get".length()).toLowerCase());
		if (!(obj instanceof Integer))
			throw new UnsupportedOperationException("The retrieved object isn't a number: " + obj);
		Integer int_ = (Integer) obj;
		if (Short.class.isAssignableFrom(cl))
			return (T) (Object) int_.shortValue();
		else
			return (T) (Object) int_;
	}

	/**
	 * Sets a player data entry <i>based on the caller method</i><br>
	 * Usage:
	 * </p>
	 * 
	 * <pre>
	 * {@code
	 * public String setNumber(short value) {
	 * 	return setIntData(value);
	 * }
	 * </pre>
	 * 
	 * @param value
	 *            The value to set
	 */
	protected void setIntData(Number value) {
		StackTraceElement st = new Exception().getStackTrace()[1];
		String mname = st.getMethodName();
		if (!mname.startsWith("set"))
			throw new UnsupportedOperationException("Can only use setIntData from a setXYZ method");
		getLoadedPlayers().get(uuid).data.put(mname.substring("set".length()).toLowerCase(), value);
	}

	/**
	 * Gets the player's Minecraft name
	 * 
	 * @return The player's Minecraft name
	 */
	public String getPlayerName() {
		return getData();
	}

	/**
	 * Sets the player's Minecraft name
	 * 
	 * @param playerName
	 *            the new name
	 */
	public void setPlayerName(String playerName) {
		setData(playerName);
	}

	private UUID uuid; // Do not save it in the file

	/**
	 * Get the player's UUID
	 * 
	 * @return The Minecraft UUID of the player
	 */
	public UUID getUuid() {
		return uuid;
	}

	/**
	 * Gets the TBMCPlayer object as a specific plugin player, keeping it's data *
	 * 
	 * @param p
	 *            Player to get
	 * @param cl
	 *            The TBMCPlayer subclass
	 */
	public <T extends TBMCPlayer> T asPluginPlayer(Class<T> cl) {
		T obj = null;
		try {
			obj = cl.newInstance();
			((TBMCPlayer) obj).uuid = uuid;
			((TBMCPlayer) obj).data.putAll(data);
		} catch (Exception e) {
			e.printStackTrace();
		}
		return obj;
	}

	private static HashMap<UUID, TBMCPlayer> LoadedPlayers = new HashMap<>();

	/**
	 * This method returns a TBMC player from their name. Calling this method may return an offline player which will load it, therefore it's highly recommended to use {@link #close()} to unload the
	 * player data. Using try-with-resources may be the easiest way to achieve this. Example:
	 * 
	 * <pre>
	 * {@code
	 * try(TBMCPlayer player = getFromName(p))
	 * {
	 * 	...
	 * }
	 * </pre>
	 * 
	 * @param name
	 *            The player's name
	 * @return The {@link TBMCPlayer} object for the player
	 */
	public static TBMCPlayer getFromName(String name) {
		@SuppressWarnings("deprecation")
		OfflinePlayer p = Bukkit.getOfflinePlayer(name);
		if (p != null)
			return getPlayer(p);
		else
			return null;
	}

	/**
	 * This method returns a TBMC player from a Bukkit player. Calling this method may return an offline player, therefore it's highly recommended to use {@link #close()} to unload the player data.
	 * Using try-with-resources may be the easiest way to achieve this. Example:
	 * 
	 * <pre>
	 * {@code
	 * try(TBMCPlayer player = getPlayer(p))
	 * {
	 * 	...
	 * }
	 * </pre>
	 * 
	 * @param p
	 *            The Player object
	 * @return The {@link TBMCPlayer} object for the player
	 */
	public static TBMCPlayer getPlayer(OfflinePlayer p) {
		if (TBMCPlayer.getLoadedPlayers().containsKey(p.getUniqueId()))
			return TBMCPlayer.getLoadedPlayers().get(p.getUniqueId());
		else
			return TBMCPlayer.loadPlayer(p);
	}

	/**
	 * This method returns a TBMC player from a player UUID. Calling this method may return an offline player, therefore it's highly recommended to use {@link #close()} to unload the player data.
	 * Using try-with-resources may be the easiest way to achieve this. Example:
	 * 
	 * <pre>
	 * {@code
	 * try(TBMCPlayer player = getPlayer(p))
	 * {
	 * 	...
	 * }
	 * </pre>
	 * 
	 * @param p
	 *            The Player object
	 * @return The {@link TBMCPlayer} object for the player
	 */
	public static TBMCPlayer getPlayer(UUID uuid) {
		if (TBMCPlayer.getLoadedPlayers().containsKey(uuid))
			return TBMCPlayer.getLoadedPlayers().get(uuid);
		else
			return TBMCPlayer.loadPlayer(Bukkit.getOfflinePlayer(uuid));
	}

	/**
	 * This is a convenience method for {@link #getPlayer(OfflinePlayer)}.{@link #asPluginPlayer(Class)}.
	 * 
	 * See those methods for more information.
	 * 
	 * @param p
	 *            Player to get
	 * @param cl
	 *            The TBMCPlayer subclass
	 * @return The player as a subtype of TBMCPlayer
	 */
	public static <T extends TBMCPlayer> T getPlayerAs(OfflinePlayer p, Class<T> cl) {
		return getPlayer(p).asPluginPlayer(cl);
	}

	/**
	 * This is a convenience method for {@link #getPlayer(UUID)}.{@link #asPluginPlayer(Class)}
	 * 
	 * See those methods for more information.
	 * 
	 * @param uuid
	 *            The UUID of the player to get
	 * @param cl
	 *            The TBMCPlayer subclass
	 * @return The player as a subtype of TBMCPlayer
	 */
	public static <T extends TBMCPlayer> T getPlayerAs(UUID uuid, Class<T> cl) {
		return getPlayer(uuid).asPluginPlayer(cl);
	}

	/**
	 * Only intended to use from ButtonCore
	 */
	public static TBMCPlayer loadPlayer(OfflinePlayer p) {
		if (getLoadedPlayers().containsKey(p.getUniqueId()))
			return getLoadedPlayers().get(p.getUniqueId());
		File file = new File(TBMC_PLAYERS_DIR);
		file.mkdirs();
		file = new File(TBMC_PLAYERS_DIR, p.getUniqueId().toString() + ".yml");
		if (!file.exists())
			return addPlayer(p);
		else {
			final YamlConfiguration yc = new YamlConfiguration();
			try {
				yc.load(file);
			} catch (Exception e) {
				new Exception("Failed to load player data for " + p.getUniqueId(), e).printStackTrace();
				return null;
			}
			TBMCPlayer player = new TBMCPlayer();
			player.uuid = p.getUniqueId();
			player.data.putAll(yc.getValues(true));
			getLoadedPlayers().put(p.getUniqueId(), player); // Accessing any value requires it to be in the map
			Bukkit.getLogger().info("Loaded player: " + player.getPlayerName());
			if (player.getPlayerName() == null) {
				player.setPlayerName(p.getName());
				Bukkit.getLogger().info("Player name saved: " + player.getPlayerName());
			} else if (!p.getName().equals(player.getPlayerName())) {
				Bukkit.getLogger().info("Renaming " + player.getPlayerName() + " to " + p.getName());
				TownyUniverse tu = Towny.getPlugin(Towny.class).getTownyUniverse();
				Resident resident = tu.getResidentMap().get(player.getPlayerName());
				if (resident == null)
					Bukkit.getLogger().warning("Resident not found - couldn't rename in Towny.");
				else if (tu.getResidentMap().contains(p.getName()))
					Bukkit.getLogger().warning("Target resident name is already in use."); // TODO: Handle
				else
					resident.setName(p.getName());
				player.setPlayerName(p.getName());
				Bukkit.getLogger().info("Renaming done.");
			}

			// Load in other plugins
			Bukkit.getServer().getPluginManager().callEvent(new TBMCPlayerLoadEvent(yc, player));
			return player;
		}
	}

	/**
	 * Only intended to use from ButtonCore
	 */
	public static TBMCPlayer addPlayer(OfflinePlayer p) {
		TBMCPlayer player = new TBMCPlayer();
		player.uuid = p.getUniqueId();
		getLoadedPlayers().put(p.getUniqueId(), player); // Accessing any value requires it to be in the map
		player.setPlayerName(p.getName());
		Bukkit.getServer().getPluginManager().callEvent(new TBMCPlayerAddEvent(player));
		savePlayer(player);
		return player;
	}

	/**
	 * Only intended to use from ButtonCore
	 */
	public static void savePlayer(TBMCPlayer player) {
		YamlConfiguration yc = new YamlConfiguration();
		for (Entry<String, Object> item : player.data.entrySet())
			yc.set(item.getKey(), item.getValue());
		Bukkit.getServer().getPluginManager().callEvent(new TBMCPlayerSaveEvent(yc, player));
		try {
			yc.save(TBMC_PLAYERS_DIR + "/" + player.uuid + ".yml");
		} catch (IOException e) {
			new Exception("Failed to save player data for " + player.getPlayerName(), e).printStackTrace();
		}
	}

	/**
	 * Only intended to use from ButtonCore
	 */
	public static void joinPlayer(TBMCPlayer player) {
		getLoadedPlayers().put(player.uuid, player);
		Bukkit.getServer().getPluginManager().callEvent(new TBMCPlayerJoinEvent(player));
	}

	/**
	 * Only intended to use from ButtonCore
	 */
	public static void quitPlayer(TBMCPlayer player) {
		getLoadedPlayers().remove(player.uuid);
		Bukkit.getServer().getPluginManager().callEvent(new TBMCPlayerQuitEvent(player));
	}

	/**
	 * By default the player data will only get cleaned from memory when the player quits. Therefore this method must be called when accessing an offline player to clean the player data up. Calling
	 * this method will have no effect on online players.<br>
	 * Therefore, the recommended use is to call it when using {@link #GetPlayer} or use try-with-resources.
	 */
	@Override
	public void close() {
		if (!Bukkit.getPlayer(uuid).isOnline())
			getLoadedPlayers().remove(uuid);
	}

	public static HashMap<UUID, TBMCPlayer> getLoadedPlayers() {
		return LoadedPlayers;
	}

	/**
	 * Get player information. This method calls the {@link TBMCPlayerGetInfoEvent} to get all the player information across the TBMC plugins.
	 * 
	 * @param target
	 *            The {@link InfoTarget} to return the info for.
	 * @return The player information.
	 */
	public String getInfo(InfoTarget target) {
		TBMCPlayerGetInfoEvent event = new TBMCPlayerGetInfoEvent(this, target);
		Bukkit.getServer().getPluginManager().callEvent(event);
		return event.getResult();
	}

	public enum InfoTarget {
		MCHover, MCCommand, Discord
	}
}