Unicopia/src/main/java/com/minelittlepony/unicopia/command/EmoteCommand.java

package com.minelittlepony.unicopia.command;

import com.minelittlepony.unicopia.client.render.PlayerPoser.Animation;
import com.minelittlepony.unicopia.entity.player.Pony;
import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.arguments.IntegerArgumentType;
import com.mojang.brigadier.exceptions.CommandSyntaxException;
import net.minecraft.server.command.CommandManager;
import net.minecraft.server.command.ServerCommandSource;

public class EmoteCommand {
    static void register(CommandDispatcher<ServerCommandSource> dispatcher) {
        dispatcher.register(CommandManager
                .literal("emote")
                .then(CommandManager.argument("animation", EnumArgumentType.of(Animation.class)).executes(source -> apply(
                        source.getSource(),
                        source.getArgument("animation", Animation.class)
                    )
                ).then(CommandManager.argument("duration", IntegerArgumentType.integer(1, 99)).executes(source -> apply(
                        source.getSource(),
                        source.getArgument("animation", Animation.class),
                        source.getArgument("duration", Integer.class)
                    )
                )
        )));
    }

    static int apply(ServerCommandSource source, Animation animation) throws CommandSyntaxException {
        return apply(source, animation, animation.getDuration());
    }

    static int apply(ServerCommandSource source, Animation animation, int duration) throws CommandSyntaxException {
        Pony.of(source.getPlayer()).setAnimation(animation, duration);
        return 0;
    }
}