mirror of
https://github.com/IntellectualSites/PlotSquared.git
synced 2025-03-13 13:29:44 +01:00
245 lines
12 KiB
Java
245 lines
12 KiB
Java
package com.plotsquared.bukkit.uuid;
|
|
|
|
import com.google.common.collect.HashBiMap;
|
|
import com.google.common.io.ByteSource;
|
|
import com.google.common.io.InputSupplier;
|
|
import com.intellectualcrafters.plot.PS;
|
|
import com.intellectualcrafters.plot.config.C;
|
|
import com.intellectualcrafters.plot.config.Settings;
|
|
import com.intellectualcrafters.plot.database.DBFunc;
|
|
import com.intellectualcrafters.plot.object.OfflinePlotPlayer;
|
|
import com.intellectualcrafters.plot.object.RunnableVal;
|
|
import com.intellectualcrafters.plot.object.StringWrapper;
|
|
import com.intellectualcrafters.plot.util.ExpireManager;
|
|
import com.intellectualcrafters.plot.util.StringMan;
|
|
import com.intellectualcrafters.plot.util.TaskManager;
|
|
import com.intellectualcrafters.plot.util.UUIDHandler;
|
|
import com.intellectualcrafters.plot.util.UUIDHandlerImplementation;
|
|
import com.intellectualcrafters.plot.uuid.UUIDWrapper;
|
|
import com.plotsquared.bukkit.util.NbtFactory;
|
|
import org.bukkit.Bukkit;
|
|
import org.bukkit.World;
|
|
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.io.FilenameFilter;
|
|
import java.io.IOException;
|
|
import java.nio.charset.StandardCharsets;
|
|
import java.nio.file.Files;
|
|
import java.util.HashMap;
|
|
import java.util.HashSet;
|
|
import java.util.List;
|
|
import java.util.UUID;
|
|
|
|
public class FileUUIDHandler extends UUIDHandlerImplementation {
|
|
|
|
public FileUUIDHandler(UUIDWrapper wrapper) {
|
|
super(wrapper);
|
|
}
|
|
|
|
@Override
|
|
public boolean startCaching(Runnable whenDone) {
|
|
return super.startCaching(whenDone) && cache(whenDone);
|
|
}
|
|
|
|
public boolean cache(final Runnable whenDone) {
|
|
final File container = Bukkit.getWorldContainer();
|
|
List<World> worlds = Bukkit.getWorlds();
|
|
final String world;
|
|
if (worlds.isEmpty()) {
|
|
world = "world";
|
|
} else {
|
|
world = worlds.get(0).getName();
|
|
}
|
|
TaskManager.runTaskAsync(new Runnable() {
|
|
@Override
|
|
public void run() {
|
|
PS.debug(C.PREFIX + "&6Starting player data caching for: " + world);
|
|
File uuidFile = new File(PS.get().IMP.getDirectory(), "uuids.txt");
|
|
if (uuidFile.exists()) {
|
|
try {
|
|
List<String> lines = Files.readAllLines(uuidFile.toPath(), StandardCharsets.UTF_8);
|
|
for (String line : lines) {
|
|
try {
|
|
line = line.trim();
|
|
if (line.isEmpty()) {
|
|
continue;
|
|
}
|
|
line = line.replaceAll("[\\|][0-9]+[\\|][0-9]+[\\|]", "");
|
|
String[] split = line.split("\\|");
|
|
String name = split[0];
|
|
if (name.isEmpty() || (name.length() > 16) || !StringMan.isAlphanumericUnd(name)) {
|
|
continue;
|
|
}
|
|
UUID uuid = FileUUIDHandler.this.uuidWrapper.getUUID(name);
|
|
if (uuid == null) {
|
|
continue;
|
|
}
|
|
UUIDHandler.add(new StringWrapper(name), uuid);
|
|
} catch (Exception e2) {
|
|
e2.printStackTrace();
|
|
}
|
|
}
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
if (Settings.TWIN_MODE_UUID) {
|
|
HashBiMap<StringWrapper, UUID> toAdd = HashBiMap.create(new HashMap<StringWrapper, UUID>());
|
|
toAdd.put(new StringWrapper("*"), DBFunc.everyone);
|
|
HashSet<UUID> all = UUIDHandler.getAllUUIDS();
|
|
PS.debug("&aFast mode UUID caching enabled!");
|
|
File playerDataFolder = new File(container, world + File.separator + "playerdata");
|
|
String[] dat = playerDataFolder.list(new FilenameFilter() {
|
|
@Override
|
|
public boolean accept(File f, String s) {
|
|
return s.endsWith(".dat");
|
|
}
|
|
});
|
|
boolean check = all.isEmpty();
|
|
if (dat != null) {
|
|
for (String current : dat) {
|
|
String s = current.replaceAll(".dat$", "");
|
|
try {
|
|
UUID uuid = UUID.fromString(s);
|
|
if (check || all.remove(uuid)) {
|
|
File file = new File(playerDataFolder + File.separator + current);
|
|
InputSupplier<FileInputStream> is = com.google.common.io.Files.newInputStreamSupplier(file);
|
|
NbtFactory.NbtCompound compound = NbtFactory.fromStream(is, NbtFactory.StreamOptions.GZIP_COMPRESSION);
|
|
NbtFactory.NbtCompound bukkit = (NbtFactory.NbtCompound) compound.get("bukkit");
|
|
String name = (String) bukkit.get("lastKnownName");
|
|
long last = (long) bukkit.get("lastPlayed");
|
|
if (ExpireManager.IMP != null) {
|
|
ExpireManager.IMP.storeDate(uuid, last);
|
|
}
|
|
toAdd.put(new StringWrapper(name), uuid);
|
|
}
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
PS.debug(C.PREFIX + "Invalid playerdata: " + current);
|
|
}
|
|
}
|
|
}
|
|
add(toAdd);
|
|
if (all.isEmpty()) {
|
|
if (whenDone != null) {
|
|
whenDone.run();
|
|
}
|
|
return;
|
|
} else {
|
|
PS.debug("Failed to cache: " + all.size() + " uuids - slowly processing all files");
|
|
}
|
|
}
|
|
HashBiMap<StringWrapper, UUID> toAdd = HashBiMap.create(new HashMap<StringWrapper, UUID>());
|
|
toAdd.put(new StringWrapper("*"), DBFunc.everyone);
|
|
HashSet<String> worlds = new HashSet<>();
|
|
worlds.add(world);
|
|
worlds.add("world");
|
|
HashSet<UUID> uuids = new HashSet<>();
|
|
HashSet<String> names = new HashSet<>();
|
|
File playerDataFolder = null;
|
|
for (String worldName : worlds) {
|
|
// Getting UUIDs
|
|
playerDataFolder = new File(container, worldName + File.separator + "playerdata");
|
|
String[] dat = playerDataFolder.list(new FilenameFilter() {
|
|
@Override
|
|
public boolean accept(File f, String s) {
|
|
return s.endsWith(".dat");
|
|
}
|
|
});
|
|
if ((dat != null) && (dat.length != 0)) {
|
|
for (String current : dat) {
|
|
String s = current.replaceAll(".dat$", "");
|
|
try {
|
|
UUID uuid = UUID.fromString(s);
|
|
uuids.add(uuid);
|
|
} catch (Exception e) {
|
|
PS.debug(C.PREFIX + "Invalid PlayerData: " + current);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
// Getting names
|
|
File playersFolder = new File(worldName + File.separator + "players");
|
|
dat = playersFolder.list(new FilenameFilter() {
|
|
@Override
|
|
public boolean accept(File f, String s) {
|
|
return s.endsWith(".dat");
|
|
}
|
|
});
|
|
if ((dat != null) && (dat.length != 0)) {
|
|
for (String current : dat) {
|
|
names.add(current.replaceAll(".dat$", ""));
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
for (UUID uuid : uuids) {
|
|
try {
|
|
File file = new File(playerDataFolder + File.separator + uuid.toString() + ".dat");
|
|
if (!file.exists()) {
|
|
continue;
|
|
}
|
|
ByteSource is = com.google.common.io.Files.asByteSource(file);
|
|
NbtFactory.NbtCompound compound = NbtFactory.fromStream(is, NbtFactory.StreamOptions.GZIP_COMPRESSION);
|
|
NbtFactory.NbtCompound bukkit = (NbtFactory.NbtCompound) compound.get("bukkit");
|
|
String name = (String) bukkit.get("lastKnownName");
|
|
long last = (long) bukkit.get("lastPlayed");
|
|
if (Settings.OFFLINE_MODE) {
|
|
if (Settings.UUID_LOWERCASE && !name.toLowerCase().equals(name)) {
|
|
uuid = FileUUIDHandler.this.uuidWrapper.getUUID(name);
|
|
} else {
|
|
long most = (long) compound.get("UUIDMost");
|
|
long least = (long) compound.get("UUIDLeast");
|
|
uuid = new UUID(most, least);
|
|
}
|
|
}
|
|
if (ExpireManager.IMP != null) {
|
|
ExpireManager.IMP.storeDate(uuid, last);
|
|
}
|
|
toAdd.put(new StringWrapper(name), uuid);
|
|
} catch (Throwable e) {
|
|
PS.debug(C.PREFIX + "&6Invalid PlayerData: " + uuid.toString() + ".dat");
|
|
}
|
|
}
|
|
for (String name : names) {
|
|
UUID uuid = FileUUIDHandler.this.uuidWrapper.getUUID(name);
|
|
StringWrapper nameWrap = new StringWrapper(name);
|
|
toAdd.put(nameWrap, uuid);
|
|
}
|
|
|
|
if (getUUIDMap().isEmpty()) {
|
|
for (OfflinePlotPlayer op : FileUUIDHandler.this.uuidWrapper.getOfflinePlayers()) {
|
|
long last = op.getLastPlayed();
|
|
if (last != 0) {
|
|
String name = op.getName();
|
|
StringWrapper wrap = new StringWrapper(name);
|
|
UUID uuid = FileUUIDHandler.this.uuidWrapper.getUUID(op);
|
|
toAdd.put(wrap, uuid);
|
|
if (ExpireManager.IMP != null) {
|
|
ExpireManager.IMP.storeDate(uuid, last);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
add(toAdd);
|
|
if (whenDone != null) {
|
|
whenDone.run();
|
|
}
|
|
}
|
|
});
|
|
return true;
|
|
}
|
|
|
|
@Override
|
|
public void fetchUUID(final String name, final RunnableVal<UUID> ifFetch) {
|
|
TaskManager.runTaskAsync(new Runnable() {
|
|
@Override
|
|
public void run() {
|
|
ifFetch.value = FileUUIDHandler.this.uuidWrapper.getUUID(name);
|
|
TaskManager.runTask(ifFetch);
|
|
}
|
|
});
|
|
}
|
|
}
|