Minestom/demo/src/main/java/net/minestom/demo/commands/HorseCommand.java

71 lines
3.3 KiB
Java

package net.minestom.demo.commands;
import net.kyori.adventure.text.Component;
import net.minestom.server.command.CommandSender;
import net.minestom.server.command.builder.Command;
import net.minestom.server.command.builder.CommandContext;
import net.minestom.server.command.builder.arguments.ArgumentType;
import net.minestom.server.command.builder.condition.Conditions;
import net.minestom.server.command.builder.exception.ArgumentSyntaxException;
import net.minestom.server.entity.EntityCreature;
import net.minestom.server.entity.EntityType;
import net.minestom.server.entity.Player;
import net.minestom.server.entity.metadata.animal.HorseMeta;
import java.util.Locale;
import java.util.stream.Collectors;
import java.util.stream.Stream;
public class HorseCommand extends Command {
public HorseCommand() {
super("horse");
setCondition(Conditions::playerOnly);
setDefaultExecutor(this::defaultExecutor);
var babyArg = ArgumentType.Boolean("baby");
var markingArg = ArgumentType.Enum("marking", HorseMeta.Marking.class);
var colorArg = ArgumentType.Enum("color", HorseMeta.Color.class);
setArgumentCallback(this::onBabyError, babyArg);
setArgumentCallback(this::onMarkingError, markingArg);
setArgumentCallback(this::onColorError, colorArg);
addSyntax(this::onHorseCommand, babyArg, markingArg, colorArg);
}
private void defaultExecutor(CommandSender sender, CommandContext context) {
sender.sendMessage(Component.text("Correct usage: /horse <baby> <marking> <color>"));
}
private void onBabyError(CommandSender sender, ArgumentSyntaxException exception) {
sender.sendMessage(Component.text("SYNTAX ERROR: '" + exception.getInput() + "' should be replaced by 'true' or 'false'"));
}
private void onMarkingError(CommandSender sender, ArgumentSyntaxException exception) {
String values = Stream.of(HorseMeta.Marking.values())
.map(value -> "'" + value.name().toLowerCase(Locale.ROOT) + "'")
.collect(Collectors.joining(", "));
sender.sendMessage(Component.text("SYNTAX ERROR: '" + exception.getInput() + "' should be replaced by " + values + "."));
}
private void onColorError(CommandSender sender, ArgumentSyntaxException exception) {
String values = Stream.of(HorseMeta.Color.values())
.map(value -> "'" + value.name().toLowerCase(Locale.ROOT) + "'")
.collect(Collectors.joining(", "));
sender.sendMessage(Component.text("SYNTAX ERROR: '" + exception.getInput() + "' should be replaced by " + values + "."));
}
private void onHorseCommand(CommandSender sender, CommandContext context) {
var player = (Player) sender;
boolean baby = context.get("baby");
HorseMeta.Marking marking = context.get("marking");
HorseMeta.Color color = context.get("color");
var horse = new EntityCreature(EntityType.HORSE);
var meta = (HorseMeta) horse.getEntityMeta();
meta.setBaby(baby);
meta.setVariant(new HorseMeta.Variant(marking, color));
//noinspection ConstantConditions - It should be impossible to execute a command without being in an instance
horse.setInstance(player.getInstance(), player.getPosition());
}
}