TF-PlotSquared/PlotSquared/src/main/java/com/intellectualcrafters/plot/util/UUIDHandler.java

package com.intellectualcrafters.plot.util;

import java.io.File;
import java.io.FilenameFilter;
import java.util.HashMap;
import java.util.HashSet;
import java.util.UUID;

import org.bukkit.Bukkit;
import org.bukkit.OfflinePlayer;
import org.bukkit.entity.Player;

import com.google.common.collect.BiMap;
import com.google.common.collect.HashBiMap;
import com.intellectualcrafters.plot.PlotMain;
import com.intellectualcrafters.plot.config.C;
import com.intellectualcrafters.plot.config.Settings;
import com.intellectualcrafters.plot.object.StringWrapper;
import com.intellectualcrafters.plot.uuid.DefaultUUIDWrapper;
import com.intellectualcrafters.plot.uuid.OfflineUUIDWrapper;
import com.intellectualcrafters.plot.uuid.UUIDWrapper;

public class UUIDHandler {
    
    public static boolean CACHED = false;
    public static UUIDWrapper uuidWrapper = null;
    
    /**
     * Map containing names and UUIDs
     *
     * @see com.google.common.collect.BiMap
     */
    private final static BiMap<StringWrapper, UUID> uuidMap = HashBiMap.create(new HashMap<StringWrapper, UUID>());

    public static void add(final StringWrapper name, final UUID uuid) {
        if (uuid == null || name == null) {
            return;
        }
        if (!uuidMap.containsKey(name) && !uuidMap.inverse().containsKey(uuid)) {
            uuidMap.put(name, uuid);
        }
    }
    
    /**
     * Get the map containing all names/uuids
     *
     * @return map with names + uuids
     *
     * @see com.google.common.collect.BiMap
     */
    public static BiMap<StringWrapper, UUID> getUuidMap() {
        return uuidMap;
    }
    
    /**
     * Check if a uuid is cached
     *
     * @param uuid to check
     *
     * @return true of the uuid is cached
     *
     * @see com.google.common.collect.BiMap#containsValue(Object)
     */
    public static boolean uuidExists(final UUID uuid) {
        return uuidMap.containsValue(uuid);
    }

    /**
     * Check if a name is cached
     *
     * @param name to check
     *
     * @return true of the name is cached
     *
     * @see com.google.common.collect.BiMap#containsKey(Object)
     */
    public static boolean nameExists(final StringWrapper name) {
        return uuidMap.containsKey(name);
    }
    
    public static void cacheAll() {
        UUIDHandler.CACHED = true;
        HashSet<String> worlds = new HashSet<>();
        worlds.add(Bukkit.getWorlds().get(0).getName());
        worlds.add("world");

        HashSet<UUID> uuids = new HashSet<>();
        HashSet<String> names = new HashSet<>();
        
        for (String worldname : worlds) {
            // Getting UUIDs
            File playerdataFolder = new File(worldname + File.separator + "playerdata");
            String[] dat = playerdataFolder.list(new FilenameFilter() {
                public boolean accept(File f, String s) {
                    return s.endsWith(".dat");
                }
            });
            if (dat != null) {
                for (String current : dat) {
                    String s = current.replaceAll(".dat$", "");
                    try {
                        UUID uuid = UUID.fromString(s);
                        uuids.add(uuid);
                    }
                    catch (Exception e) {
                        PlotMain.sendConsoleSenderMessage(C.PREFIX.s() + "Invalid playerdata: "+current);
                    }
                }
            }
            
            // Getting names
            File playersFolder = new File(worldname + File.separator + "players");
            dat = playersFolder.list(new FilenameFilter() {
                public boolean accept(File f, String s) {
                    return s.endsWith(".dat");
                }
            });
            if (dat != null) {
                for (String current : dat) {
                    names.add(current.replaceAll(".dat$", ""));
                }
            }
        }
        
        
        UUIDWrapper wrapper = new DefaultUUIDWrapper();
        for (UUID uuid : uuids) {
            try {
                OfflinePlayer player = wrapper.getOfflinePlayer(uuid);
                uuid = UUIDHandler.uuidWrapper.getUUID(player);
                StringWrapper name = new StringWrapper(player.getName());
                add(name, uuid);
            }
            catch (Throwable e) {
                PlotMain.sendConsoleSenderMessage(C.PREFIX.s() + "&6Invalid playerdata: "+uuid.toString() + ".dat");
            }
        }
        for (String name : names) {
            OfflinePlayer player = Bukkit.getOfflinePlayer(name);
            UUID uuid = UUIDHandler.uuidWrapper.getUUID(player);
            StringWrapper nameWrap = new StringWrapper(name);
            add(nameWrap, uuid);
        }
        
        PlotMain.sendConsoleSenderMessage(C.PREFIX.s() + "&6Cached a total of: " + UUIDHandler.uuidMap.size() + " UUIDs");
    }
    
    public static UUID getUUID(Player player) {
        return UUIDHandler.uuidWrapper.getUUID(player);
    }
    
    public static UUID getUUID(OfflinePlayer player) {
        return UUIDHandler.uuidWrapper.getUUID(player);
    }
    
    public static String getName(UUID uuid) {
        if (uuid == null) {
            return null;
        }
        
        // check online
        for (Player player : Bukkit.getOnlinePlayers()) {
            UUID u2 = UUIDHandler.uuidWrapper.getUUID(player);
            if (uuid.equals(u2)) {
                return player.getName();
            }
        }
        
        // check cache
        StringWrapper name = UUIDHandler.uuidMap.inverse().get(uuid);
        if (name != null) {
            return name.value;
        }
        
        // check drive
        if (Settings.UUID_FROM_DISK) {
            OfflinePlayer op = UUIDHandler.uuidWrapper.getOfflinePlayer(uuid);
            String string = op.getName();
            StringWrapper sw = new StringWrapper(string);
            add(sw, uuid);
            return string;
        }
        
        return null;
    }

    public static UUID getUUID(final String name) {
        if (name == null) {
            return null;
        }
        
        // check online
        Player player = Bukkit.getPlayer(name);
        if (player != null) {
            UUID uuid = UUIDHandler.uuidWrapper.getUUID(player);
            add(new StringWrapper(name), uuid);
            return uuid;
        }
        
        // check cache
        StringWrapper wrap = new StringWrapper(name);
        UUID uuid = UUIDHandler.uuidMap.get(wrap);
        if (uuid != null) {
            return uuid;
        }
        
        // Read from disk OR convert directly to offline UUID
        if (Settings.UUID_FROM_DISK || uuidWrapper instanceof OfflineUUIDWrapper) {
            OfflinePlayer op = Bukkit.getOfflinePlayer(name);
            uuid = UUIDHandler.uuidWrapper.getUUID(op);
            add(new StringWrapper(name), uuid);
            return uuid;
        }
        
        return null;
    }
}