46 lines
1.8 KiB
Java
46 lines
1.8 KiB
Java
|
package net.nuggetmc.ai.commands;
|
||
|
|
|
||
|
|
import com.jonahseguin.drink.CommandService;
|
||
|
|
import com.jonahseguin.drink.Drink;
|
||
|
|
import com.jonahseguin.drink.annotation.Command;
|
||
|
|
import com.jonahseguin.drink.command.DrinkCommandContainer;
|
||
|
|
import com.jonahseguin.drink.command.DrinkCommandService;
|
||
|
|
import net.nuggetmc.ai.PlayerAI;
|
||
|
|
import net.nuggetmc.ai.commands.commands.PlayerAICommand;
|
||
|
|
import org.bukkit.ChatColor;
|
||
|
|
|
||
|
|
import java.lang.reflect.Method;
|
||
|
|
import java.util.Arrays;
|
||
|
|
import java.util.List;
|
||
|
|
import java.util.Map;
|
||
|
|
import java.util.stream.Collectors;
|
||
|
|
|
||
|
|
public class CommandHandler {
|
||
|
|
|
||
|
|
private final DrinkCommandService drink;
|
||
|
|
|
||
|
|
public CommandHandler() {
|
||
|
|
drink = (DrinkCommandService) Drink.get(PlayerAI.getInstance());
|
||
|
|
drink.register(new PlayerAICommand(this), "playerai", "pai");
|
||
|
|
drink.registerCommands();
|
||
|
|
}
|
||
|
|
|
||
|
|
public List<String> getUsage(Class<? extends CommandInstance> clazz) {
|
||
|
|
return getSubCommands(clazz).stream().map(c -> {
|
||
|
|
Command command = c.getAnnotation(Command.class);
|
||
|
|
return ChatColor.GRAY + " ▪ " + ChatColor.YELLOW + "/" + getRootName(clazz) + (command.name().isEmpty() ? "" : " " + command.name()) + ChatColor.GRAY + " ▪ "
|
||
|
|
+ ChatColor.RESET + command.desc();
|
||
|
|
}).collect(Collectors.toList());
|
||
|
|
}
|
||
|
|
|
||
|
|
private String getRootName(Class<? extends CommandInstance> clazz) {
|
||
|
|
return drink.getCommands().entrySet().stream()
|
||
|
|
.filter(c -> c.getValue().getObject().getClass().isAssignableFrom(clazz)).map(Map.Entry::getKey).findFirst().orElse(null);
|
||
|
|
}
|
||
|
|
|
||
|
|
private List<Method> getSubCommands(Class<? extends CommandInstance> clazz) {
|
||
|
|
return Arrays.stream(clazz.getDeclaredMethods()).filter(m -> m.isAnnotationPresent(Command.class)).collect(Collectors.toList());
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|