f3fcb59fa1
using an Interface full of useful NMS methods. I'm still using reflection, but only when absolutely needed. I'm also working on a new and better implementation for NBT Lists. Made NBT Lists and NBT Compounds serializable (can be saved in variables through restarts), needs testing.
439 lines
13 KiB
Java
439 lines
13 KiB
Java
package me.TheBukor.SkStuff.util;
|
|
|
|
import java.io.EOFException;
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.io.FileNotFoundException;
|
|
import java.io.FileOutputStream;
|
|
import java.io.IOException;
|
|
import java.io.OutputStream;
|
|
import java.util.ArrayList;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
|
|
import javax.annotation.Nullable;
|
|
|
|
import org.bukkit.Material;
|
|
import org.bukkit.block.Block;
|
|
import org.bukkit.craftbukkit.v1_8_R2.CraftWorld;
|
|
import org.bukkit.craftbukkit.v1_8_R2.entity.CraftEntity;
|
|
import org.bukkit.craftbukkit.v1_8_R2.inventory.CraftItemStack;
|
|
import org.bukkit.entity.Entity;
|
|
import org.bukkit.inventory.ItemStack;
|
|
|
|
import ch.njol.skript.Skript;
|
|
import ch.njol.skript.classes.Changer;
|
|
import ch.njol.skript.classes.ClassInfo;
|
|
import ch.njol.skript.classes.Parser;
|
|
import ch.njol.skript.lang.ParseContext;
|
|
import ch.njol.skript.registrations.Classes;
|
|
import ch.njol.util.coll.CollectionUtils;
|
|
import net.minecraft.server.v1_8_R2.BlockPosition;
|
|
import net.minecraft.server.v1_8_R2.EntityInsentient;
|
|
import net.minecraft.server.v1_8_R2.MojangsonParseException;
|
|
import net.minecraft.server.v1_8_R2.MojangsonParser;
|
|
import net.minecraft.server.v1_8_R2.NBTBase;
|
|
import net.minecraft.server.v1_8_R2.NBTCompressedStreamTools;
|
|
import net.minecraft.server.v1_8_R2.NBTTagCompound;
|
|
import net.minecraft.server.v1_8_R2.NBTTagDouble;
|
|
import net.minecraft.server.v1_8_R2.NBTTagEnd;
|
|
import net.minecraft.server.v1_8_R2.NBTTagFloat;
|
|
import net.minecraft.server.v1_8_R2.NBTTagInt;
|
|
import net.minecraft.server.v1_8_R2.NBTTagList;
|
|
import net.minecraft.server.v1_8_R2.NBTTagString;
|
|
import net.minecraft.server.v1_8_R2.PathfinderGoal;
|
|
import net.minecraft.server.v1_8_R2.PathfinderGoalSelector;
|
|
import net.minecraft.server.v1_8_R2.TileEntity;
|
|
import net.minecraft.server.v1_8_R2.World;
|
|
|
|
public class NMS_v1_8_R2 implements NMSInterface {
|
|
|
|
@Override
|
|
public void addToCompound(Object compound, Object toAdd) {
|
|
if (compound instanceof NBTTagCompound && toAdd instanceof NBTTagCompound) {
|
|
((NBTTagCompound) compound).a((NBTTagCompound) toAdd);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void removeFromCompound(Object compound, String ... toRemove) {
|
|
if (compound instanceof NBTTagCompound) {
|
|
((NBTTagCompound) compound).remove(toRemove.toString()); //FIXME
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Object parseRawNBT(String rawNBT) {
|
|
NBTTagCompound parsedNBT = null;
|
|
try {
|
|
parsedNBT = MojangsonParser.parse(rawNBT);
|
|
} catch (MojangsonParseException ex) {
|
|
Skript.warning("Error when parsing NBT - " + ex.getMessage());
|
|
}
|
|
return parsedNBT;
|
|
}
|
|
|
|
@Override
|
|
public int getContentsId(Object nbtList) {
|
|
if (nbtList instanceof NBTTagList) {
|
|
return ((NBTTagList) nbtList).f();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
@Override
|
|
public Object[] getContents(Object nbtList) {
|
|
if (nbtList instanceof NBTTagList) {
|
|
List<Object> contents = new ArrayList<Object>();
|
|
for (int i = 0; i < ((NBTTagList) nbtList).size(); i++) {
|
|
if (getIndex(nbtList, i) != null) {
|
|
contents.add(getIndex(nbtList, i));
|
|
}
|
|
}
|
|
return contents.toArray();
|
|
}
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public void addToList(Object nbtList, Object toAdd) {
|
|
if (nbtList instanceof NBTTagList && toAdd instanceof NBTBase) {
|
|
((NBTTagList) nbtList).add((NBTBase) toAdd);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void removeFromList(Object nbtList, int index) {
|
|
if (nbtList instanceof NBTTagList) {
|
|
((NBTTagList) nbtList).a(index);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void setIndex(Object nbtList, int index, Object toSet) {
|
|
if (nbtList instanceof NBTTagList && toSet instanceof NBTBase) {
|
|
((NBTTagList) nbtList).a(index, (NBTBase) toSet);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Object getIndex(Object nbtList, int index) {
|
|
if (nbtList instanceof NBTTagList) {
|
|
NBTBase value = ((NBTTagList) nbtList).g(index);
|
|
if (value instanceof NBTTagEnd)
|
|
return null;
|
|
else
|
|
return value;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public void removePathfinderGoal(Object entity, Class<?> goalClass, boolean isTargetSelector) {
|
|
if (entity instanceof EntityInsentient) {
|
|
((EntityInsentient) entity).setGoalTarget(null);
|
|
if (isTargetSelector) {
|
|
Iterator<?> goals = ((List<?>) ReflectionUtils.getField("b", PathfinderGoalSelector.class, ((EntityInsentient) entity).targetSelector)).iterator();
|
|
while (goals.hasNext()) {
|
|
Object goal = goals.next();
|
|
if (ReflectionUtils.getField("a", goal.getClass(), goal).getClass() == goalClass) {
|
|
goals.remove();
|
|
}
|
|
}
|
|
} else {
|
|
Iterator<?> goals = ((List<?>) ReflectionUtils.getField("b", PathfinderGoalSelector.class, ((EntityInsentient) entity).goalSelector)).iterator();
|
|
while (goals.hasNext()) {
|
|
Object goal = goals.next();
|
|
if (ReflectionUtils.getField("a", goal.getClass(), goal).getClass() == goalClass) {
|
|
goals.remove();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void addPathfinderGoal(Object entity, int priority, Object goal, boolean isTargetSelector) {
|
|
if (entity instanceof EntityInsentient && goal instanceof PathfinderGoal) {
|
|
if (isTargetSelector)
|
|
((EntityInsentient) entity).targetSelector.a(priority, (PathfinderGoal) goal);
|
|
else
|
|
((EntityInsentient) entity).goalSelector.a(priority, (PathfinderGoal) goal);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void registerCompoundClassInfo() {
|
|
Classes.registerClass(new ClassInfo<NBTTagCompound>(NBTTagCompound.class, "compound").user("((nbt)?( ?tag)?) ?compounds?").name("NBT Compound").changer(new Changer<NBTTagCompound>() {
|
|
|
|
@SuppressWarnings("unchecked")
|
|
@Override
|
|
@Nullable
|
|
public Class<?>[] acceptChange(ChangeMode mode) {
|
|
if (mode == ChangeMode.ADD || mode == ChangeMode.REMOVE || mode == ChangeMode.SET) {
|
|
return CollectionUtils.array(String[].class, NBTTagCompound[].class);
|
|
}
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public void change(NBTTagCompound[] NBT, @Nullable Object[] delta, ChangeMode mode) {
|
|
if (mode == ChangeMode.SET) {
|
|
if (delta[0] instanceof NBTTagCompound) {
|
|
NBT[0] = (NBTTagCompound) delta[0];
|
|
} else {
|
|
NBTTagCompound parsedNBT = (NBTTagCompound) parseRawNBT((String) delta[0]);
|
|
NBT[0] = parsedNBT;
|
|
}
|
|
} else if (mode == ChangeMode.ADD) {
|
|
if (delta[0] instanceof String) {
|
|
NBTTagCompound parsedNBT = (NBTTagCompound) parseRawNBT((String) delta[0]);
|
|
addToCompound(NBT[0], parsedNBT);
|
|
} else {
|
|
addToCompound(NBT[0], delta[0]);
|
|
}
|
|
} else if (mode == ChangeMode.REMOVE) {
|
|
if (delta[0] instanceof NBTTagCompound)
|
|
return;
|
|
for (Object s : delta) {
|
|
NBT[0].remove((String) s);
|
|
}
|
|
}
|
|
}
|
|
}).parser(new Parser<NBTTagCompound>() {
|
|
|
|
@Override
|
|
public String getVariableNamePattern() {
|
|
return ".+";
|
|
}
|
|
|
|
@Override
|
|
@Nullable
|
|
public NBTTagCompound parse(String rawNBT, ParseContext context) {
|
|
if (rawNBT.startsWith("nbt:{") && rawNBT.endsWith("}")) {
|
|
rawNBT.substring(4);
|
|
NBTTagCompound NBT = (NBTTagCompound) parseRawNBT(rawNBT);
|
|
if (NBT.toString().equals("{}")) {
|
|
return null;
|
|
}
|
|
return NBT;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public String toString(NBTTagCompound compound, int arg1) {
|
|
return compound.toString();
|
|
}
|
|
|
|
@Override
|
|
public String toVariableNameString(NBTTagCompound compound) {
|
|
return "nbt:" + compound.toString();
|
|
}
|
|
}));
|
|
|
|
}
|
|
|
|
@Override
|
|
public void registerNBTListClassInfo() {
|
|
Classes.registerClass(new ClassInfo<NBTTagList>(NBTTagList.class, "nbtlist").user("nbt ?list ?(tag)?").name("NBT List").changer(new Changer<NBTTagList>() {
|
|
|
|
@SuppressWarnings("unchecked")
|
|
@Override
|
|
@Nullable
|
|
public Class<?>[] acceptChange(ChangeMode mode) {
|
|
if (mode == ChangeMode.ADD || mode == ChangeMode.SET || mode == ChangeMode.DELETE || mode == ChangeMode.RESET) {
|
|
return CollectionUtils.array(Float[].class, Double[].class, String[].class, NBTTagCompound[].class, Integer[].class, NBTTagList[].class);
|
|
}
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public void change(NBTTagList[] nbtList, @Nullable Object[] delta, ChangeMode mode) {
|
|
int typeId = 0;
|
|
if (delta instanceof Float[]) {
|
|
typeId = 5;
|
|
} else if (delta instanceof Double[]) {
|
|
typeId = 6;
|
|
} else if (delta instanceof String[]) {
|
|
typeId = 8;
|
|
} else if (delta instanceof NBTTagList[]) {
|
|
typeId = 9;
|
|
} else if (delta instanceof NBTTagCompound[]) {
|
|
typeId = 10;
|
|
} else if (delta instanceof Integer[]) {
|
|
typeId = 11;
|
|
} else {
|
|
return;
|
|
}
|
|
if (mode == ChangeMode.SET) {
|
|
if (typeId == 9)
|
|
nbtList[0] = (NBTTagList) delta[0];
|
|
} else if (mode == ChangeMode.ADD) {
|
|
if (getContentsId(nbtList[0]) == typeId) {
|
|
if (typeId == 5) {
|
|
NBTTagFloat floatTag = new NBTTagFloat((float) delta[0]);
|
|
addToList(nbtList[0], floatTag);
|
|
} else if (typeId == 6) {
|
|
NBTTagDouble doubleTag = new NBTTagDouble((double) delta[0]);
|
|
addToList(nbtList[0], doubleTag);
|
|
} else if (typeId == 8) {
|
|
NBTTagString stringTag = new NBTTagString((String) delta [0]);
|
|
addToList(nbtList[0], stringTag);
|
|
} else if (typeId == 10) {
|
|
addToList(nbtList[0], delta[0]);
|
|
} else if (typeId == 11) {
|
|
NBTTagInt intTag = new NBTTagInt((int) delta[0]);
|
|
addToList(nbtList[0], intTag);
|
|
}
|
|
}
|
|
} else if (mode == ChangeMode.DELETE || mode == ChangeMode.RESET) {
|
|
nbtList[0] = new NBTTagList();
|
|
}
|
|
}
|
|
}).parser(new Parser<NBTTagList>() {
|
|
|
|
@Override
|
|
public String getVariableNamePattern() {
|
|
return ".+";
|
|
}
|
|
|
|
@Override
|
|
@Nullable
|
|
public NBTTagList parse(String ignored, ParseContext context) {
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public String toString(NBTTagList nbtList, int arg1) {
|
|
return nbtList.toString();
|
|
}
|
|
|
|
@Override
|
|
public String toVariableNameString(NBTTagList nbtList) {
|
|
return nbtList.toString();
|
|
}
|
|
}));
|
|
}
|
|
|
|
@Override
|
|
public Object getEntityNBT(Entity entity) {
|
|
net.minecraft.server.v1_8_R2.Entity nmsEntity = ((CraftEntity) entity).getHandle();
|
|
NBTTagCompound NBT = new NBTTagCompound();
|
|
nmsEntity.e(NBT);
|
|
return NBT;
|
|
}
|
|
|
|
@Override
|
|
public Object getTileNBT(Block block) {
|
|
NBTTagCompound NBT = new NBTTagCompound();
|
|
World nmsWorld = ((CraftWorld) block.getWorld()).getHandle();
|
|
TileEntity tileEntity = nmsWorld.getTileEntity(new BlockPosition(block.getX(), block.getY(), block.getZ()));
|
|
if (tileEntity == null)
|
|
return null;
|
|
tileEntity.b(NBT);
|
|
return NBT;
|
|
}
|
|
|
|
@Override
|
|
public Object getItemNBT(ItemStack itemStack) {
|
|
if (itemStack.getType() == Material.AIR)
|
|
return null;
|
|
NBTTagCompound NBT = new NBTTagCompound();
|
|
net.minecraft.server.v1_8_R2.ItemStack nmsItem = CraftItemStack.asNMSCopy(itemStack);
|
|
NBT = nmsItem.getTag();
|
|
if (NBT == null || NBT.toString().equals("{}")) //Null or empty.
|
|
return null;
|
|
return new Object[] { NBT };
|
|
}
|
|
|
|
@Override
|
|
public void setEntityNBT(Entity entity, Object newCompound) {
|
|
if (newCompound instanceof NBTTagCompound) {
|
|
net.minecraft.server.v1_8_R2.Entity nmsEntity = ((CraftEntity) entity).getHandle();
|
|
nmsEntity.f((NBTTagCompound) newCompound);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void setTileNBT(Block block, Object newCompound) {
|
|
if (newCompound instanceof NBTTagCompound) {
|
|
World nmsWorld = ((CraftWorld) block.getWorld()).getHandle();
|
|
TileEntity tileEntity = nmsWorld.getTileEntity(new BlockPosition(block.getX(), block.getY(), block.getZ()));
|
|
if (tileEntity == null)
|
|
return;
|
|
tileEntity.a((NBTTagCompound) newCompound);
|
|
tileEntity.update();
|
|
nmsWorld.notify(tileEntity.getPosition());
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public ItemStack getItemWithNBT(ItemStack itemStack, Object compound) {
|
|
if (compound instanceof NBTTagCompound) {
|
|
if (itemStack.getType() == Material.AIR || itemStack == null)
|
|
return null;
|
|
if (compound == null || compound.toString().equals("{}"))
|
|
return itemStack;
|
|
net.minecraft.server.v1_8_R2.ItemStack nmsItem = CraftItemStack.asNMSCopy(itemStack);
|
|
nmsItem.setTag((NBTTagCompound) compound);
|
|
ItemStack newItem = CraftItemStack.asCraftMirror(nmsItem);
|
|
return newItem;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public Object getFileNBT(File file) {
|
|
FileInputStream fis = null;
|
|
try {
|
|
fis = new FileInputStream(file);
|
|
} catch (FileNotFoundException ex) {
|
|
return null; //File doesn't exist.
|
|
}
|
|
NBTTagCompound fileNBT = null;
|
|
try {
|
|
fileNBT = NBTCompressedStreamTools.a(fis);
|
|
fis.close();
|
|
} catch (IOException ex) {
|
|
ex.printStackTrace();
|
|
} finally {
|
|
try {
|
|
fis.close();
|
|
} catch (IOException ex) {
|
|
ex.printStackTrace();
|
|
}
|
|
}
|
|
return fileNBT;
|
|
}
|
|
|
|
@Override
|
|
public void setFileNBT(File file, Object newCompound) {
|
|
OutputStream os = null;
|
|
try {
|
|
os = new FileOutputStream(file);
|
|
} catch (FileNotFoundException ex) {
|
|
ex.printStackTrace();
|
|
}
|
|
try {
|
|
NBTCompressedStreamTools.a((NBTTagCompound) newCompound, os);
|
|
os.close();
|
|
} catch (Exception ex) {
|
|
if (ex instanceof EOFException) {
|
|
; //Ignore, just end of the file
|
|
} else {
|
|
ex.printStackTrace();
|
|
}
|
|
} finally {
|
|
try {
|
|
os.close();
|
|
} catch (Exception ex) {
|
|
if (ex instanceof EOFException) {
|
|
; //Ignore.
|
|
} else {
|
|
ex.printStackTrace();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} |