package dev.padrewin.hudactionbar.commands; import me.clip.placeholderapi.PlaceholderAPI; import java.util.ArrayList; import java.util.Arrays; import java.util.Collections; import java.util.Comparator; import java.util.HashMap; import java.util.HashSet; import java.util.List; import java.util.Map; import java.util.Set; import java.util.stream.Collectors; import dev.padrewin.hudactionbar.HudActionbar; import dev.padrewin.hudactionbar.manager.CommandManager; import dev.padrewin.hudactionbar.manager.LocaleManager; import org.bukkit.Bukkit; import org.bukkit.ChatColor; import org.bukkit.command.BlockCommandSender; import org.bukkit.command.Command; import org.bukkit.command.CommandSender; import org.bukkit.command.ConsoleCommandSender; import org.bukkit.command.TabExecutor; import org.bukkit.entity.Entity; import org.bukkit.entity.Player; import org.bukkit.entity.minecart.CommandMinecart; import org.bukkit.permissions.Permissible; import org.bukkit.util.StringUtil; public abstract class CommandHandler implements TabExecutor, NamedExecutor { protected final Map registeredCommands = new HashMap<>(); protected final Map registeredHandlers = new HashMap<>(); protected HudActionbar plugin; protected String cmd; protected CommandManager.CommandAliases aliases; public CommandHandler(HudActionbar plugin, String cmd, CommandManager.CommandAliases aliases) { this.plugin = plugin; this.cmd = cmd; this.aliases = aliases; } public void registerCommand(BaseCommand command) { for (String alias : command.getAliases()) { if (registeredCommands.containsKey(alias)) plugin.getLogger().warning("Conflicting command aliases for '" + alias + "', overwriting."); registeredCommands.put(alias, command); } } public void registerHandler(CommandHandler handler) { for (String alias : handler.getAliases()) { if (registeredCommands.containsKey(alias)) plugin.getLogger().warning("Conflicting handler aliases for '" + alias + "', overwriting."); registeredHandlers.put(alias, handler); } } @Override public boolean onCommand(CommandSender sender, Command command, String label, String[] args) { if (args.length == 0) { noArgs(sender); return true; } String subcmd = args[0].toLowerCase(); // Resolve PAPI placeholders for console arguments if (sender instanceof ConsoleCommandSender) for (int i = 1; i < args.length; i++) if (args[i].startsWith("%")) args[i] = PlaceholderAPI.setPlaceholders(null, args[i]); // Resolve entity selectors in command blocks if (sender instanceof BlockCommandSender || sender instanceof CommandMinecart) { for (int i = 1; i < args.length; i++) { String selector = args[i]; if (!selector.startsWith("@")) continue; try { List selected = Bukkit.selectEntities(sender, selector); if (selected.isEmpty()) { sender.sendMessage("Error: No entities found for selector '" + selector + "'"); return true; } if (selected.size() > 1) { sender.sendMessage("Error: More than one entity found for selector '" + selector + "'"); return true; } if (!(selected.get(0) instanceof Player)) { sender.sendMessage("Error: Entity '" + selected.get(0).getName() + "' is not a player"); return true; } args[i] = selected.get(0).getName(); } catch (IllegalArgumentException e) { sender.sendMessage("Error: Invalid player selector '" + selector + "'"); return true; } } } CommandHandler handler = registeredHandlers.get(subcmd); if (handler != null) { if (!handler.hasPermission(sender)) { plugin.getLocaleManager().sendMessage(sender, "no-permission"); return true; } handler.onCommand(sender, command, label, shortenArgs(args)); return true; } BaseCommand subCommand = registeredCommands.get(subcmd); if (subCommand == null) { unknownCommand(sender, args); return true; } if (!subCommand.hasPermission(sender)) { plugin.getLocaleManager().sendMessage(sender, "no-permission"); return true; } try { subCommand.execute(plugin, sender, shortenArgs(args)); } catch (ArrayIndexOutOfBoundsException e) { sender.sendMessage(ChatColor.RED + "A HudActionbar error occurred while executing that command. Did you enter an invalid parameter?"); } return true; } @Override public List onTabComplete(CommandSender sender, Command command, String alias, String[] args) { if (args.length == 0) return Collections.emptyList(); String subcmd = args[0].toLowerCase(); if (args.length == 1) { List names = new ArrayList<>(); names.addAll(registeredHandlers.entrySet().stream().filter(x -> x.getValue().hasPermission(sender)).map(Map.Entry::getKey).collect(Collectors.toList())); names.addAll(registeredCommands.entrySet().stream().filter(x -> x.getValue().hasPermission(sender)).map(Map.Entry::getKey).collect(Collectors.toList())); List completions = new ArrayList<>(); StringUtil.copyPartialMatches(subcmd, names, completions); return completions; } CommandHandler h = registeredHandlers.get(subcmd); if (h != null && h.hasPermission(sender)) return h.onTabComplete(sender, command, alias, shortenArgs(args)); BaseCommand sc = registeredCommands.get(subcmd); if (sc != null && sc.hasPermission(sender)) return sc.tabComplete(plugin, sender, shortenArgs(args)); return Collections.emptyList(); } public abstract void noArgs(CommandSender sender); public abstract void unknownCommand(CommandSender sender, String[] args); public List getExecutables() { Set executors = new HashSet<>(); executors.addAll(registeredHandlers.values()); executors.addAll(registeredCommands.values()); List sorted = new ArrayList<>(executors); sorted.sort(Comparator.comparing(NamedExecutor::getName)); return sorted; } @Override public String getName() { return cmd; } @Override public List getAliases() { List a = aliases.get(); return a.isEmpty() ? Collections.singletonList(cmd) : a; } @Override public boolean hasPermission(Permissible p) { return true; } protected String[] shortenArgs(String[] args) { if (args.length == 0) return args; return new ArrayList<>(Arrays.asList(args).subList(1, args.length)).toArray(new String[0]); } }