55 lines
2.5 KiB
Java
55 lines
2.5 KiB
Java
package net.minecraft.commands.arguments;
|
|
|
|
import com.mojang.brigadier.context.CommandContext;
|
|
import com.mojang.brigadier.exceptions.CommandSyntaxException;
|
|
import com.mojang.brigadier.exceptions.DynamicCommandExceptionType;
|
|
import com.mojang.serialization.DynamicOps;
|
|
import java.util.Arrays;
|
|
import java.util.Collection;
|
|
import net.minecraft.commands.CommandBuildContext;
|
|
import net.minecraft.commands.CommandSourceStack;
|
|
import net.minecraft.core.HolderLookup;
|
|
import net.minecraft.nbt.NbtOps;
|
|
import net.minecraft.nbt.SnbtGrammar;
|
|
import net.minecraft.nbt.Tag;
|
|
import net.minecraft.network.chat.Component;
|
|
import net.minecraft.network.chat.ComponentSerialization;
|
|
import net.minecraft.network.chat.ComponentUtils;
|
|
import net.minecraft.util.parsing.packrat.commands.CommandArgumentParser;
|
|
import net.minecraft.util.parsing.packrat.commands.ParserBasedArgument;
|
|
import net.minecraft.world.entity.Entity;
|
|
import org.jetbrains.annotations.Nullable;
|
|
|
|
public class ComponentArgument extends ParserBasedArgument<Component> {
|
|
private static final Collection<String> EXAMPLES = Arrays.asList("\"hello world\"", "'hello world'", "\"\"", "{text:\"hello world\"}", "[\"\"]");
|
|
public static final DynamicCommandExceptionType ERROR_INVALID_COMPONENT = new DynamicCommandExceptionType(
|
|
object -> Component.translatableEscape("argument.component.invalid", object)
|
|
);
|
|
private static final DynamicOps<Tag> OPS = NbtOps.INSTANCE;
|
|
private static final CommandArgumentParser<Tag> TAG_PARSER = SnbtGrammar.createParser(OPS);
|
|
|
|
private ComponentArgument(HolderLookup.Provider registries) {
|
|
super(TAG_PARSER.withCodec(registries.createSerializationContext(OPS), TAG_PARSER, ComponentSerialization.CODEC, ERROR_INVALID_COMPONENT));
|
|
}
|
|
|
|
public static Component getRawComponent(CommandContext<CommandSourceStack> context, String name) {
|
|
return context.getArgument(name, Component.class);
|
|
}
|
|
|
|
public static Component getResolvedComponent(CommandContext<CommandSourceStack> context, String name, @Nullable Entity entity) throws CommandSyntaxException {
|
|
return ComponentUtils.updateForEntity(context.getSource(), getRawComponent(context, name), entity, 0);
|
|
}
|
|
|
|
public static Component getResolvedComponent(CommandContext<CommandSourceStack> context, String name) throws CommandSyntaxException {
|
|
return getResolvedComponent(context, name, context.getSource().getEntity());
|
|
}
|
|
|
|
public static ComponentArgument textComponent(CommandBuildContext context) {
|
|
return new ComponentArgument(context);
|
|
}
|
|
|
|
@Override
|
|
public Collection<String> getExamples() {
|
|
return EXAMPLES;
|
|
}
|
|
}
|