TF-EssentialsX/Essentials/src/com/earth2me/essentials/UserMap.java

package com.earth2me.essentials;

import com.earth2me.essentials.utils.StringUtil;
import com.google.common.cache.Cache;
import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.RemovalCause;
import com.google.common.cache.RemovalListener;
import com.google.common.cache.RemovalNotification;
import com.google.common.util.concurrent.UncheckedExecutionException;
import java.io.File;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Set;
import java.util.UUID;
import java.util.concurrent.ConcurrentSkipListMap;
import java.util.concurrent.ConcurrentSkipListSet;
import java.util.concurrent.ExecutionException;
import net.ess3.api.IEssentials;
import org.bukkit.entity.Player;


public class UserMap extends CacheLoader<String, User> implements IConf
{
	private final transient IEssentials ess;
	private final transient Cache<String, User> users;
	private final transient ConcurrentSkipListSet<UUID> keys = new ConcurrentSkipListSet<UUID>();
	private final transient ConcurrentSkipListMap<String, UUID> names = new ConcurrentSkipListMap<String, UUID>();
	private final transient ConcurrentSkipListMap<UUID, ArrayList<String>> history = new ConcurrentSkipListMap<UUID, ArrayList<String>>();
	private UUIDMap uuidMap;
	
	public UserMap(final IEssentials ess)
	{
		super();
		this.ess = ess;
		uuidMap = new UUIDMap(ess);
		//RemovalListener<UUID, User> remListener = new UserMapRemovalListener();
		//users = CacheBuilder.newBuilder().maximumSize(ess.getSettings().getMaxUserCacheCount()).softValues().removalListener(remListener).build(this);
		users = CacheBuilder.newBuilder().maximumSize(ess.getSettings().getMaxUserCacheCount()).softValues().build(this);
	}
	
	private void loadAllUsersAsync(final IEssentials ess)
	{
		ess.runTaskAsynchronously(new Runnable()
		{
			@Override
			public void run()
			{
				synchronized (users)
				{
					final File userdir = new File(ess.getDataFolder(), "userdata");
					if (!userdir.exists())
					{
						return;
					}
					keys.clear();
					users.invalidateAll();
					for (String string : userdir.list())
					{
						if (!string.endsWith(".yml"))
						{
							continue;
						}
						final String name = string.substring(0, string.length() - 4);
						try
						{
							keys.add(UUID.fromString(name));
						}
						catch (IllegalArgumentException ex)
						{
							//Ignore these users till they rejoin.
						}
					}
					uuidMap.loadAllUsers(names, history);
				}
			}
		});
	}
	
	public boolean userExists(final UUID uuid)
	{
		return keys.contains(uuid);
	}
	
	public User getUser(final String name)
	{
		try
		{
			final String sanitizedName = StringUtil.safeString(name);
			if (names.containsKey(sanitizedName))
			{
				final UUID uuid = names.get(sanitizedName);
				return getUser(uuid);
			}
			
			final File userFile = getUserFileFromString(sanitizedName);
			if (userFile.exists())
			{
				ess.getLogger().info("Importing user " + name + " to usermap.");
				User user = new User(new OfflinePlayer(sanitizedName, ess.getServer()), ess);
				trackUUID(user.getBase().getUniqueId(), user.getName());
				return user;
			}
			return null;
		}
		catch (UncheckedExecutionException ex)
		{
			return null;
		}
	}
	
	public User getUser(final UUID uuid)
	{
		try
		{
			return users.get(uuid.toString());
		}
		catch (ExecutionException ex)
		{
			return null;
		}
		catch (UncheckedExecutionException ex)
		{
			return null;
		}
	}
	
	public void trackUUID(final UUID uuid, final String name)
	{
		if (uuid != null)
		{
			keys.add(uuid);
			if (name != null && name.length() > 0)
			{
				final String keyName = StringUtil.safeString(name);
				if (!names.containsKey(keyName) || !names.get(keyName).equals(uuid))
				{
					names.put(keyName, uuid);
					uuidMap.writeUUIDMap();
				}
			}
		}
	}
	
	@Override
	public User load(final String stringUUID) throws Exception
	{
		UUID uuid = UUID.fromString(stringUUID);
		Player player = ess.getServer().getPlayer(uuid);
		if (player != null)
		{
			final User user = new User(player, ess);
			trackUUID(uuid, user.getName());
			return user;
		}
		
		final File userFile = getUserFileFromID(uuid);
		
		if (userFile.exists())
		{
			player = new OfflinePlayer(uuid, ess.getServer());
			final User user = new User(player, ess);
			((OfflinePlayer)player).setName(user.getLastAccountName());
			trackUUID(uuid, user.getName());
			return user;
		}
		
		throw new Exception("User not found!");
	}
	
	@Override
	public void reloadConfig()
	{
		getUUIDMap().forceWriteUUIDMap();
		loadAllUsersAsync(ess);
	}
	
	public void invalidateAll()
	{
		users.invalidateAll();
	}
	
	public void removeUser(final String name)
	{
		if (names == null)
		{
			ess.getLogger().warning("Name collection is null, cannot remove user.");
			return;
		}
		UUID uuid = names.get(name);
		if (uuid != null)
		{
			keys.remove(uuid);
			users.invalidate(uuid);
		}
		names.remove(name);
		names.remove(StringUtil.safeString(name));
	}
	
	public Set<UUID> getAllUniqueUsers()
	{
		return Collections.unmodifiableSet(keys.clone());
	}
	
	public int getUniqueUsers()
	{
		return keys.size();
	}
	
	protected ConcurrentSkipListMap<String, UUID> getNames()
	{
		return names;
	}
	
	protected ConcurrentSkipListMap<UUID, ArrayList<String>> getHistory()
	{
		return history;
	}
	
	public List<String> getUserHistory(final UUID uuid)
	{
		return history.get(uuid);
	}
	
	public UUIDMap getUUIDMap()
	{
		return uuidMap;
	}
	
	private File getUserFileFromID(final UUID uuid)
	{
		final File userFolder = new File(ess.getDataFolder(), "userdata");
		return new File(userFolder, uuid.toString() + ".yml");
	}
	
	public File getUserFileFromString(final String name)
	{
		final File userFolder = new File(ess.getDataFolder(), "userdata");
		return new File(userFolder, StringUtil.sanitizeFileName(name) + ".yml");
	}
//	class UserMapRemovalListener implements RemovalListener
//	{
//		@Override
//		public void onRemoval(final RemovalNotification notification)
//		{
//			Object value = notification.getValue();
//			if (value != null)
//			{
//				((User)value).cleanup();
//			}
//		}
//	}
}