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());
    }

}