mirror of
https://github.com/KAMKEEL/CustomNPC-Plus.git
synced 2026-11-12 11:57:29 +00:00
361 lines
13 KiB
Java
361 lines
13 KiB
Java
package noppes.npcs.controllers;
|
|
|
|
import net.minecraft.command.ICommandSender;
|
|
import net.minecraft.entity.Entity;
|
|
import net.minecraft.entity.EntityList;
|
|
import net.minecraft.nbt.CompressedStreamTools;
|
|
import net.minecraft.nbt.NBTTagCompound;
|
|
import net.minecraft.nbt.NBTTagList;
|
|
import net.minecraft.nbt.NBTTagString;
|
|
import net.minecraft.util.ChatComponentText;
|
|
import noppes.npcs.CustomNpcs;
|
|
import noppes.npcs.LogWriter;
|
|
import noppes.npcs.NoppesUtilServer;
|
|
import noppes.npcs.api.IPos;
|
|
import noppes.npcs.api.IWorld;
|
|
import noppes.npcs.api.entity.IEntity;
|
|
import noppes.npcs.api.handler.ICloneHandler;
|
|
import noppes.npcs.controllers.data.TagMap;
|
|
import noppes.npcs.scripted.CustomNPCsException;
|
|
import noppes.npcs.scripted.NpcAPI;
|
|
import noppes.npcs.util.NBTJsonUtil;
|
|
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.util.*;
|
|
|
|
public class ServerCloneController implements ICloneHandler {
|
|
public static ServerCloneController Instance;
|
|
|
|
public ServerCloneController() {
|
|
loadClones();
|
|
}
|
|
|
|
private void loadClones() {
|
|
try {
|
|
File dir = new File(getDir(), "..");
|
|
File file = new File(dir, "clonednpcs.dat");
|
|
if (file.exists()) {
|
|
Map<Integer, Map<String, NBTTagCompound>> clones = loadOldClones(file);
|
|
file.delete();
|
|
file = new File(dir, "clonednpcs.dat_old");
|
|
if (file.exists())
|
|
file.delete();
|
|
|
|
for (int tab : clones.keySet()) {
|
|
Map<String, NBTTagCompound> map = clones.get(tab);
|
|
for (String name : map.keySet()) {
|
|
saveClone(tab, name, map.get(name));
|
|
}
|
|
}
|
|
}
|
|
} catch (Exception e) {
|
|
LogWriter.except(e);
|
|
}
|
|
}
|
|
|
|
public File getDir() {
|
|
File dir = new File(CustomNpcs.getWorldSaveDirectory(), "clones");
|
|
if (!dir.exists())
|
|
dir.mkdir();
|
|
return dir;
|
|
}
|
|
|
|
private Map<Integer, Map<String, NBTTagCompound>> loadOldClones(File file) throws Exception {
|
|
Map<Integer, Map<String, NBTTagCompound>> clones = new HashMap<Integer, Map<String, NBTTagCompound>>();
|
|
NBTTagCompound nbttagcompound1 = CompressedStreamTools.readCompressed(new FileInputStream(file));
|
|
NBTTagList list = nbttagcompound1.getTagList("Data", 10);
|
|
if (list == null) {
|
|
return clones;
|
|
}
|
|
for (int i = 0; i < list.tagCount(); i++) {
|
|
NBTTagCompound compound = list.getCompoundTagAt(i);
|
|
if (!compound.hasKey("ClonedTab")) {
|
|
compound.setInteger("ClonedTab", 1);
|
|
}
|
|
|
|
Map<String, NBTTagCompound> tab = clones.get(compound.getInteger("ClonedTab"));
|
|
if (tab == null)
|
|
clones.put(compound.getInteger("ClonedTab"), tab = new HashMap<String, NBTTagCompound>());
|
|
|
|
String name = compound.getString("ClonedName");
|
|
int number = 1;
|
|
while (tab.containsKey(name)) {
|
|
number++;
|
|
name = String.format("%s%s", compound.getString("ClonedName"), number);
|
|
}
|
|
compound.removeTag("ClonedName");
|
|
compound.removeTag("ClonedTab");
|
|
compound.removeTag("ClonedDate");
|
|
cleanTags(compound);
|
|
tab.put(name, compound);
|
|
}
|
|
return clones;
|
|
}
|
|
|
|
public NBTTagCompound getCloneData(ICommandSender player, String name, int tab) {
|
|
File file = new File(new File(getDir(), tab + ""), name + ".json");
|
|
if (!file.exists()) {
|
|
if (player != null)
|
|
player.addChatMessage(new ChatComponentText("Could not find clone file"));
|
|
return null;
|
|
}
|
|
try {
|
|
return NBTJsonUtil.LoadFile(file);
|
|
} catch (Exception e) {
|
|
LogWriter.error("Error loading: " + file.getAbsolutePath(), e);
|
|
if (player != null)
|
|
player.addChatMessage(new ChatComponentText(e.getMessage()));
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public void saveClone(int tab, String name, NBTTagCompound compound) {
|
|
try {
|
|
File dir = new File(getDir(), tab + "");
|
|
if (!dir.exists())
|
|
dir.mkdir();
|
|
String filename = name + ".json";
|
|
|
|
File file = new File(dir, filename + "_new");
|
|
File file2 = new File(dir, filename);
|
|
NBTJsonUtil.SaveFile(file, compound);
|
|
addToTagMap(compound, name, tab);
|
|
if (file2.exists()) {
|
|
file2.delete();
|
|
}
|
|
file.renameTo(file2);
|
|
} catch (Exception e) {
|
|
LogWriter.except(e);
|
|
}
|
|
}
|
|
|
|
public List<String> getClones(int tab) {
|
|
List<String> list = new ArrayList<String>();
|
|
File dir = new File(getDir(), tab + "");
|
|
if (!dir.exists() || !dir.isDirectory())
|
|
return list;
|
|
for (String file : dir.list()) {
|
|
if (file.endsWith(".json"))
|
|
list.add(file.substring(0, file.length() - 5));
|
|
}
|
|
return list;
|
|
}
|
|
|
|
public List<String> getClonesDate(int tab) {
|
|
List<String> list = new ArrayList<String>();
|
|
File dir = new File(getDir(), tab + "");
|
|
if (!dir.exists() || !dir.isDirectory())
|
|
return list;
|
|
File[] files = dir.listFiles();
|
|
Arrays.sort(files, new Comparator<File>() {
|
|
public int compare(File f1, File f2) {
|
|
return Long.compare(f1.lastModified(), f2.lastModified());
|
|
}
|
|
});
|
|
|
|
for (File file : files) {
|
|
String fileName = file.getName();
|
|
if (fileName.endsWith(".json"))
|
|
list.add(fileName.substring(0, fileName.length() - 5));
|
|
}
|
|
return list;
|
|
}
|
|
|
|
public boolean removeClone(String name, int tab) {
|
|
File file = new File(new File(getDir(), tab + ""), name + ".json");
|
|
if (!file.exists())
|
|
return false;
|
|
file.delete();
|
|
removeFromTagMap(name, tab);
|
|
return true;
|
|
}
|
|
|
|
public String addClone(NBTTagCompound nbttagcompound, String name, int tab) {
|
|
cleanTags(nbttagcompound);
|
|
saveClone(tab, name, nbttagcompound);
|
|
return name;
|
|
}
|
|
|
|
public String addClone(NBTTagCompound nbttagcompound, String name, int tab, NBTTagCompound tempTags) {
|
|
cleanTagList(nbttagcompound, tempTags);
|
|
cleanTags(nbttagcompound);
|
|
saveClone(tab, name, nbttagcompound);
|
|
return name;
|
|
}
|
|
|
|
public NBTTagCompound cleanTagList(NBTTagCompound nbttagcompound, NBTTagCompound tempTags) {
|
|
HashSet<UUID> tagUUIDs = new HashSet<UUID>();
|
|
if (nbttagcompound.hasKey("TagUUIDs")) {
|
|
NBTTagList nbtTagList = nbttagcompound.getTagList("TagUUIDs", 8);
|
|
for (int i = 0; i < nbtTagList.tagCount(); i++) {
|
|
tagUUIDs.add(UUID.fromString(nbtTagList.getStringTagAt(i)));
|
|
}
|
|
|
|
nbttagcompound.removeTag("TagUUIDs");
|
|
}
|
|
if (tempTags.hasKey("TempTagUUIDs")) {
|
|
NBTTagList nbtTagList = tempTags.getTagList("TempTagUUIDs", 8);
|
|
for (int i = 0; i < nbtTagList.tagCount(); i++) {
|
|
tagUUIDs.add(UUID.fromString(nbtTagList.getStringTagAt(i)));
|
|
}
|
|
|
|
tempTags.removeTag("TempTagUUIDs");
|
|
}
|
|
|
|
if (tagUUIDs.size() > 0) {
|
|
NBTTagList nbtTagList = new NBTTagList();
|
|
for (UUID uuid : tagUUIDs) {
|
|
nbtTagList.appendTag(new NBTTagString(uuid.toString()));
|
|
}
|
|
nbttagcompound.setTag("TagUUIDs", nbtTagList);
|
|
}
|
|
|
|
return nbttagcompound;
|
|
}
|
|
|
|
public boolean addToTagMap(NBTTagCompound nbttagcompound, String name, int tab) {
|
|
HashSet<UUID> tagUUIDs = new HashSet<UUID>();
|
|
if (nbttagcompound.hasKey("TagUUIDs")) {
|
|
NBTTagList nbtTagList = nbttagcompound.getTagList("TagUUIDs", 8);
|
|
for (int i = 0; i < nbtTagList.tagCount(); i++) {
|
|
tagUUIDs.add(UUID.fromString(nbtTagList.getStringTagAt(i)));
|
|
}
|
|
}
|
|
|
|
TagMap tagMap = ServerTagMapController.Instance.getTagMap(tab);
|
|
if (!tagUUIDs.isEmpty()) {
|
|
tagMap.putClone(name, tagUUIDs);
|
|
} else {
|
|
tagMap.removeClone(name);
|
|
}
|
|
ServerTagMapController.Instance.saveTagMap(tagMap);
|
|
return true;
|
|
}
|
|
|
|
public boolean removeFromTagMap(String name, int tab) {
|
|
TagMap tagMap = ServerTagMapController.Instance.getTagMap(tab);
|
|
if (tagMap.removeClone(name)) {
|
|
ServerTagMapController.Instance.saveTagMap(tagMap);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public void cleanTags(NBTTagCompound nbttagcompound) {
|
|
if (nbttagcompound.hasKey("ItemGiverId"))
|
|
nbttagcompound.setInteger("ItemGiverId", 0);
|
|
if (nbttagcompound.hasKey("TransporterId"))
|
|
nbttagcompound.setInteger("TransporterId", -1);
|
|
|
|
nbttagcompound.removeTag("StartPosNew");
|
|
nbttagcompound.removeTag("StartPos");
|
|
nbttagcompound.removeTag("MovingPathNew");
|
|
nbttagcompound.removeTag("Pos");
|
|
nbttagcompound.removeTag("Riding");
|
|
|
|
if (!nbttagcompound.hasKey("ModRev"))
|
|
nbttagcompound.setInteger("ModRev", 1);
|
|
|
|
if (nbttagcompound.hasKey("TransformRole")) {
|
|
NBTTagCompound adv = nbttagcompound.getCompoundTag("TransformRole");
|
|
adv.setInteger("TransporterId", -1);
|
|
nbttagcompound.setTag("TransformRole", adv);
|
|
}
|
|
|
|
if (nbttagcompound.hasKey("TransformJob")) {
|
|
NBTTagCompound adv = nbttagcompound.getCompoundTag("TransformJob");
|
|
adv.setInteger("ItemGiverId", 0);
|
|
nbttagcompound.setTag("TransformJob", adv);
|
|
}
|
|
|
|
if (nbttagcompound.hasKey("TransformAI")) {
|
|
NBTTagCompound adv = nbttagcompound.getCompoundTag("TransformAI");
|
|
adv.removeTag("StartPosNew");
|
|
adv.removeTag("StartPos");
|
|
adv.removeTag("MovingPathNew");
|
|
nbttagcompound.setTag("TransformAI", adv);
|
|
}
|
|
}
|
|
|
|
public IEntity spawn(double x, double y, double z, int tab, String name, IWorld world, boolean ignoreProtection) {
|
|
NBTTagCompound compound = this.getCloneData((ICommandSender) null, name, tab);
|
|
if (compound == null) {
|
|
throw new CustomNPCsException("Unknown clone tab:" + tab + " name:" + name, new Object[0]);
|
|
} else {
|
|
Entity entity;
|
|
if (!ignoreProtection) {
|
|
entity = NoppesUtilServer.spawnCloneWithProtection(compound, (int) x, (int) y, (int) z, world.getMCWorld());
|
|
} else {
|
|
entity = NoppesUtilServer.spawnClone(compound, (int) x, (int) y, (int) z, world.getMCWorld());
|
|
}
|
|
return entity == null ? null : NpcAPI.Instance().getIEntity(entity);
|
|
}
|
|
}
|
|
|
|
public IEntity spawn(IPos pos, int tab, String name, IWorld world, boolean ignoreProtection) {
|
|
return this.spawn(pos.getX(), pos.getY(), pos.getZ(), tab, name, world, ignoreProtection);
|
|
}
|
|
|
|
public IEntity spawn(double x, double y, double z, int tab, String name, IWorld world) {
|
|
return spawn(x, y, z, tab, name, world, true);
|
|
}
|
|
|
|
public IEntity spawn(IPos pos, int tab, String name, IWorld world) {
|
|
return this.spawn(pos.getX(), pos.getY(), pos.getZ(), tab, name, world);
|
|
}
|
|
|
|
public IEntity[] getTab(int tab, IWorld world) {
|
|
File dir = new File(getDir(), tab + "");
|
|
if (!dir.exists() || !dir.isDirectory() || dir.listFiles() == null) {
|
|
return new IEntity[]{};
|
|
}
|
|
|
|
ArrayList<IEntity> arrayList = new ArrayList<>();
|
|
|
|
try {
|
|
for (File file : dir.listFiles()) {
|
|
if (file.getName().endsWith(".json")) {
|
|
NBTTagCompound compound = NBTJsonUtil.LoadFile(file);
|
|
Instance.cleanTags(compound);
|
|
Entity entity = EntityList.createEntityFromNBT(compound, world.getMCWorld());
|
|
arrayList.add(entity == null ? null : NpcAPI.Instance().getIEntity(entity));
|
|
}
|
|
}
|
|
} catch (Exception ignored) {
|
|
}
|
|
|
|
return arrayList.toArray(new IEntity[]{});
|
|
}
|
|
|
|
public IEntity get(int tab, String name, IWorld world) {
|
|
NBTTagCompound compound = this.getCloneData((ICommandSender) null, name, tab);
|
|
if (compound == null) {
|
|
throw new CustomNPCsException("Unknown clone tab:" + tab + " name:" + name, new Object[0]);
|
|
} else {
|
|
Instance.cleanTags(compound);
|
|
Entity entity = EntityList.createEntityFromNBT(compound, world.getMCWorld());
|
|
return entity == null ? null : NpcAPI.Instance().getIEntity(entity);
|
|
}
|
|
}
|
|
|
|
public boolean has(int tab, String name) {
|
|
NBTTagCompound compound = this.getCloneData((ICommandSender) null, name, tab);
|
|
return compound != null;
|
|
}
|
|
|
|
|
|
public void set(int tab, String name, IEntity entity) {
|
|
NBTTagCompound compound = new NBTTagCompound();
|
|
if (!entity.getMCEntity().writeMountToNBT(compound))
|
|
throw new CustomNPCsException("Cannot save dead entities", new Object[0]);
|
|
|
|
this.cleanTags(compound);
|
|
this.saveClone(tab, name, compound);
|
|
}
|
|
|
|
public void remove(int tab, String name) {
|
|
this.removeClone(name, tab);
|
|
}
|
|
}
|