75 lines
2.2 KiB
Java
75 lines
2.2 KiB
Java
package me.TheBukor.SkStuff.expressions;
|
|
|
|
import javax.annotation.Nullable;
|
|
|
|
import org.bukkit.event.Event;
|
|
|
|
import ch.njol.skript.classes.Changer.ChangeMode;
|
|
import ch.njol.skript.lang.Expression;
|
|
import ch.njol.skript.lang.SkriptParser.ParseResult;
|
|
import ch.njol.skript.lang.util.SimpleExpression;
|
|
import ch.njol.util.Kleenean;
|
|
import ch.njol.util.coll.CollectionUtils;
|
|
import me.TheBukor.SkStuff.SkStuff;
|
|
import me.TheBukor.SkStuff.util.ReflectionUtils;
|
|
|
|
public class ExprNBTListContents extends SimpleExpression<Object> {
|
|
private Expression<Object> nbtList;
|
|
|
|
private Class<?> nbtBaseClass = ReflectionUtils.getNMSClass("NBTBase");
|
|
|
|
@Override
|
|
public Class<? extends Object> getReturnType() {
|
|
return Object.class;
|
|
}
|
|
|
|
@Override
|
|
public boolean isSingle() {
|
|
return false;
|
|
}
|
|
|
|
@SuppressWarnings("unchecked")
|
|
@Override
|
|
public boolean init(Expression<?>[] expr, int matchedPattern, Kleenean arg2, ParseResult result) {
|
|
nbtList = (Expression<Object>) expr[0];
|
|
return true;
|
|
}
|
|
|
|
@Override
|
|
public String toString(@Nullable Event e, boolean debug) {
|
|
return "contents from NBT list " + nbtList.toString(e, debug);
|
|
}
|
|
|
|
@Override
|
|
@Nullable
|
|
protected Object[] get(Event e) {
|
|
Object list = nbtList.getSingle(e);
|
|
return SkStuff.getNMSMethods().getContents(list);
|
|
}
|
|
|
|
@Override
|
|
public void change(Event e, @Nullable Object[] delta, ChangeMode mode) {
|
|
Object list = nbtList.getSingle(e);
|
|
if (mode == ChangeMode.ADD) {
|
|
if (nbtBaseClass.isAssignableFrom(delta[0].getClass()))
|
|
SkStuff.getNMSMethods().addToList(list, delta[0]);
|
|
else if (delta[0] instanceof Number)
|
|
SkStuff.getNMSMethods().addToList(list, SkStuff.getNMSMethods().convertToNBT((Number) delta[0]));
|
|
else if (delta[0] instanceof String)
|
|
SkStuff.getNMSMethods().addToList(list, SkStuff.getNMSMethods().convertToNBT((String) delta[0]));
|
|
} else if (mode == ChangeMode.REMOVE || mode == ChangeMode.REMOVE_ALL) {
|
|
// TODO Code to remove a single or all objects of some value in an NBT array.
|
|
}
|
|
}
|
|
|
|
@SuppressWarnings("unchecked")
|
|
@Override
|
|
@Nullable
|
|
public Class<?>[] acceptChange(ChangeMode mode) {
|
|
if (mode == ChangeMode.ADD || mode == ChangeMode.REMOVE || mode == ChangeMode.REMOVE_ALL) {
|
|
return CollectionUtils.array(Object.class);
|
|
}
|
|
return null;
|
|
}
|
|
}
|