Files
Tplus/src/main/java/net/nuggetmc/ai/commands/CommandHandler.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());
}
}