101 lines
4.1 KiB
Java
101 lines
4.1 KiB
Java
/*
|
|
* Decompiled with CFR 0.152.
|
|
*
|
|
* Could not load the following classes:
|
|
* com.mojang.brigadier.Message
|
|
* com.mojang.brigadier.builder.ArgumentBuilder
|
|
* com.mojang.brigadier.context.CommandContext
|
|
* com.mojang.brigadier.exceptions.CommandSyntaxException
|
|
* com.mojang.brigadier.exceptions.SimpleCommandExceptionType
|
|
* com.mojang.logging.LogUtils
|
|
* org.slf4j.Logger
|
|
*/
|
|
package net.minecraft.server.commands.data;
|
|
|
|
import com.mojang.brigadier.Message;
|
|
import com.mojang.brigadier.builder.ArgumentBuilder;
|
|
import com.mojang.brigadier.context.CommandContext;
|
|
import com.mojang.brigadier.exceptions.CommandSyntaxException;
|
|
import com.mojang.brigadier.exceptions.SimpleCommandExceptionType;
|
|
import com.mojang.logging.LogUtils;
|
|
import java.util.Locale;
|
|
import java.util.UUID;
|
|
import java.util.function.Function;
|
|
import net.minecraft.advancements.criterion.NbtPredicate;
|
|
import net.minecraft.commands.CommandSourceStack;
|
|
import net.minecraft.commands.Commands;
|
|
import net.minecraft.commands.arguments.EntityArgument;
|
|
import net.minecraft.commands.arguments.NbtPathArgument;
|
|
import net.minecraft.core.HolderLookup;
|
|
import net.minecraft.nbt.CompoundTag;
|
|
import net.minecraft.nbt.NbtUtils;
|
|
import net.minecraft.nbt.Tag;
|
|
import net.minecraft.network.chat.Component;
|
|
import net.minecraft.server.commands.data.DataAccessor;
|
|
import net.minecraft.server.commands.data.DataCommands;
|
|
import net.minecraft.util.ProblemReporter;
|
|
import net.minecraft.world.entity.Entity;
|
|
import net.minecraft.world.entity.player.Player;
|
|
import net.minecraft.world.level.storage.TagValueInput;
|
|
import org.slf4j.Logger;
|
|
|
|
public class EntityDataAccessor
|
|
implements DataAccessor {
|
|
private static final Logger LOGGER = LogUtils.getLogger();
|
|
private static final SimpleCommandExceptionType ERROR_NO_PLAYERS = new SimpleCommandExceptionType((Message)Component.translatable("commands.data.entity.invalid"));
|
|
public static final Function<String, DataCommands.DataProvider> PROVIDER = arg -> new DataCommands.DataProvider((String)arg){
|
|
final /* synthetic */ String val$arg;
|
|
{
|
|
this.val$arg = string;
|
|
}
|
|
|
|
@Override
|
|
public DataAccessor access(CommandContext<CommandSourceStack> context) throws CommandSyntaxException {
|
|
return new EntityDataAccessor(EntityArgument.getEntity(context, this.val$arg));
|
|
}
|
|
|
|
@Override
|
|
public ArgumentBuilder<CommandSourceStack, ?> wrap(ArgumentBuilder<CommandSourceStack, ?> parent, Function<ArgumentBuilder<CommandSourceStack, ?>, ArgumentBuilder<CommandSourceStack, ?>> function) {
|
|
return parent.then(Commands.literal("entity").then(function.apply((ArgumentBuilder<CommandSourceStack, ?>)Commands.argument(this.val$arg, EntityArgument.entity()))));
|
|
}
|
|
};
|
|
private final Entity entity;
|
|
|
|
public EntityDataAccessor(Entity entity) {
|
|
this.entity = entity;
|
|
}
|
|
|
|
@Override
|
|
public void setData(CompoundTag tag) throws CommandSyntaxException {
|
|
if (this.entity instanceof Player) {
|
|
throw ERROR_NO_PLAYERS.create();
|
|
}
|
|
UUID uuid = this.entity.getUUID();
|
|
try (ProblemReporter.ScopedCollector reporter = new ProblemReporter.ScopedCollector(this.entity.problemPath(), LOGGER);){
|
|
this.entity.load(TagValueInput.create((ProblemReporter)reporter, (HolderLookup.Provider)this.entity.registryAccess(), tag));
|
|
this.entity.setUUID(uuid);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public CompoundTag getData() {
|
|
return NbtPredicate.getEntityTagToCompare(this.entity);
|
|
}
|
|
|
|
@Override
|
|
public Component getModifiedSuccess() {
|
|
return Component.translatable("commands.data.entity.modified", this.entity.getDisplayName());
|
|
}
|
|
|
|
@Override
|
|
public Component getPrintSuccess(Tag data) {
|
|
return Component.translatable("commands.data.entity.query", this.entity.getDisplayName(), NbtUtils.toPrettyComponent(data));
|
|
}
|
|
|
|
@Override
|
|
public Component getPrintSuccess(NbtPathArgument.NbtPath path, double scale, int value) {
|
|
return Component.translatable("commands.data.entity.get", path.asString(), this.entity.getDisplayName(), String.format(Locale.ROOT, "%.2f", scale), value);
|
|
}
|
|
}
|
|
|