Added perms support, fixed maven deps, fixed typos
This commit is contained in:
TopchetoEU
parent
36e00b43ec
commit
a0ab822476
341
src/main/java/me/topchetoeu/bedwars/commands/Command.java
Normal file
341
src/main/java/me/topchetoeu/bedwars/commands/Command.java
Normal file
@ -0,0 +1,341 @@
|
|||||||
|
package me.topchetoeu.bedwars.commands;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Collection;
|
||||||
|
import java.util.Collections;
|
||||||
|
import java.util.HashSet;
|
||||||
|
import java.util.Hashtable;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Set;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
|
import org.bukkit.command.CommandSender;
|
||||||
|
import org.bukkit.plugin.java.JavaPlugin;
|
||||||
|
|
||||||
|
import me.topchetoeu.bedwars.commands.args.ArgParser;
|
||||||
|
import me.topchetoeu.bedwars.commands.args.ArgParserRes;
|
||||||
|
import me.topchetoeu.bedwars.commands.args.CollectionArgParser;
|
||||||
|
import me.topchetoeu.bedwars.commands.args.CollectionProvider;
|
||||||
|
import me.topchetoeu.bedwars.commands.args.EnumArgParser;
|
||||||
|
import me.topchetoeu.bedwars.commands.args.IntArgParser;
|
||||||
|
import me.topchetoeu.bedwars.commands.args.LiteralArgParser;
|
||||||
|
import me.topchetoeu.bedwars.commands.args.LocationArgParser;
|
||||||
|
import me.topchetoeu.bedwars.commands.args.PlayerArgParser;
|
||||||
|
import me.topchetoeu.bedwars.commands.args.StringArgParser;
|
||||||
|
import me.topchetoeu.bedwars.commands.args.Suggestions;
|
||||||
|
import me.topchetoeu.bedwars.permissions.Permissions;
|
||||||
|
import net.md_5.bungee.api.ChatColor;
|
||||||
|
import net.md_5.bungee.api.chat.BaseComponent;
|
||||||
|
import net.md_5.bungee.api.chat.ComponentBuilder;
|
||||||
|
|
||||||
|
public class Command {
|
||||||
|
private String name;
|
||||||
|
private String helpMessage;
|
||||||
|
private CommandExecutor executor;
|
||||||
|
private HashSet<Command> children = new HashSet<>();
|
||||||
|
private ArgParser parser;
|
||||||
|
private boolean recursive = false;
|
||||||
|
private String requiredPermission = null;
|
||||||
|
|
||||||
|
private Set<JavaPlugin> parents = new HashSet<>();
|
||||||
|
|
||||||
|
public String getName() {
|
||||||
|
return name;
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean attachedToAnyPlugin() {
|
||||||
|
return parents.size() > 0;
|
||||||
|
}
|
||||||
|
public Set<JavaPlugin> getParentPlugins() {
|
||||||
|
return Collections.unmodifiableSet(parents);
|
||||||
|
}
|
||||||
|
|
||||||
|
public ArgParser getParser() {
|
||||||
|
return parser;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getHelpMessage() {
|
||||||
|
return helpMessage;
|
||||||
|
}
|
||||||
|
public Command setHelpMessage(String val) {
|
||||||
|
helpMessage = val;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getRequiredPermission() {
|
||||||
|
return requiredPermission;
|
||||||
|
}
|
||||||
|
public Command permission(String wildcard) {
|
||||||
|
requiredPermission = wildcard;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public CommandExecutor getExecutor() {
|
||||||
|
return executor;
|
||||||
|
}
|
||||||
|
public Command setExecutor(CommandExecutor val) {
|
||||||
|
executor = val;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Command addChild(Command cmd) {
|
||||||
|
if (cmd == null) throw new RuntimeException("cmd is null");
|
||||||
|
children.add(cmd);
|
||||||
|
return cmd;
|
||||||
|
}
|
||||||
|
public Command removeChild(Command cmd) {
|
||||||
|
if (cmd == null) throw new RuntimeException("cmd is null");
|
||||||
|
children.remove(cmd);
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
public boolean hasChild(Command cmd) {
|
||||||
|
if (cmd == null) return false;
|
||||||
|
return children.contains(cmd);
|
||||||
|
}
|
||||||
|
public Set<Command> getChildren() {
|
||||||
|
return Collections.unmodifiableSet(children);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Command setRecursive(boolean val) {
|
||||||
|
recursive = val;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
public boolean isRecursive() {
|
||||||
|
return recursive;
|
||||||
|
}
|
||||||
|
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
public void execute(CommandSender sender, String[] _args) {
|
||||||
|
Command toExecute = this;
|
||||||
|
|
||||||
|
Hashtable<String, Object> newArgs = new Hashtable<>();
|
||||||
|
|
||||||
|
List<String> args = new ArrayList<>();
|
||||||
|
Collections.addAll(args, _args);
|
||||||
|
|
||||||
|
String err = null;
|
||||||
|
|
||||||
|
while (args.size() > 0) {
|
||||||
|
Command newCmd = null;
|
||||||
|
Set<Command> children = toExecute.getChildren();
|
||||||
|
if (toExecute.isRecursive()) children = Collections.singleton(toExecute);
|
||||||
|
for (Command cmd : children) {
|
||||||
|
ArgParser parser = cmd.getParser();
|
||||||
|
ArgParserRes res = parser.parse(sender, args);
|
||||||
|
|
||||||
|
if (res.hasError()) err = res.getError();
|
||||||
|
else if (res.hasSucceeded()) {
|
||||||
|
for (int i = 0; i < res.getTakenCount(); i++) {
|
||||||
|
if (args.size() == 0) break;
|
||||||
|
args.remove(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (res.hasResult()) {
|
||||||
|
if (cmd.recursive) {
|
||||||
|
if (!newArgs.containsKey(cmd.name)) newArgs.put(cmd.name, new ArrayList<>());
|
||||||
|
((List<Object>)newArgs.get(cmd.name)).add(res.getResult());
|
||||||
|
}
|
||||||
|
else newArgs.put(cmd.name, res.getResult());
|
||||||
|
}
|
||||||
|
newCmd = cmd;
|
||||||
|
if (cmd.requiredPermission != null) {
|
||||||
|
if (!Permissions.hasPermission(sender, cmd.requiredPermission))
|
||||||
|
err = "You don't have enough permissions to do that.";
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
if (newCmd == null) {
|
||||||
|
toExecute = null;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
toExecute = newCmd;
|
||||||
|
if (err != null) break;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (err == null) {
|
||||||
|
if (toExecute == null) err = "Invalid command syntax.";
|
||||||
|
else if (toExecute.getExecutor() == null) err = "Incomplete command.";
|
||||||
|
}
|
||||||
|
|
||||||
|
if (err != null) sender.spigot().sendMessage(new ComponentBuilder()
|
||||||
|
.append("Error: ")
|
||||||
|
.color(ChatColor.DARK_RED)
|
||||||
|
.append(err)
|
||||||
|
.color(ChatColor.RED)
|
||||||
|
.create()
|
||||||
|
);
|
||||||
|
else {
|
||||||
|
if (toExecute.isRecursive() && !newArgs.containsKey(toExecute.name)) newArgs.put(toExecute.name, new ArrayList<>());
|
||||||
|
BaseComponent[] _err = toExecute.getExecutor().execute(sender, this, newArgs);
|
||||||
|
if (_err != null && _err.length > 0) sender.spigot().sendMessage(new ComponentBuilder()
|
||||||
|
.append("Error: ")
|
||||||
|
.color(ChatColor.DARK_RED)
|
||||||
|
.append(err)
|
||||||
|
.reset()
|
||||||
|
.create()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<String> tabComplete(CommandSender sender, String[] _args) {
|
||||||
|
Command toComplete = this;
|
||||||
|
|
||||||
|
List<String> args = new ArrayList<>();
|
||||||
|
Collections.addAll(args, _args);
|
||||||
|
|
||||||
|
int index = 1;
|
||||||
|
|
||||||
|
while (args.size() > 0) {
|
||||||
|
boolean found = false;
|
||||||
|
index++;
|
||||||
|
for (Command cmd : toComplete.children) {
|
||||||
|
ArgParser parser = cmd.getParser();
|
||||||
|
ArgParserRes res = parser.parse(sender, args);
|
||||||
|
if (res.hasSucceeded()) {
|
||||||
|
for (int i = 0; i < res.getTakenCount(); i++) {
|
||||||
|
if (args.size() == 0) break;
|
||||||
|
args.remove(0);
|
||||||
|
}
|
||||||
|
toComplete = cmd;
|
||||||
|
found = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (!found) break;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (args.size() == 0) return null;
|
||||||
|
|
||||||
|
Suggestions suggestions = new Suggestions();
|
||||||
|
|
||||||
|
for (Command cmd : toComplete.children) {
|
||||||
|
cmd.getParser().addCompleteSuggestions(sender, args, suggestions);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (suggestions.hasError() && !suggestions.getSuggestions().contains(args.get(0))) {
|
||||||
|
sender.spigot().sendMessage(new ComponentBuilder()
|
||||||
|
.append("Error (argument %d): ".formatted(index + 1))
|
||||||
|
.color(ChatColor.DARK_RED)
|
||||||
|
.append(suggestions.getError())
|
||||||
|
.color(ChatColor.RED)
|
||||||
|
.create()
|
||||||
|
);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
List<String> _suggestions = new ArrayList<>(suggestions.getSuggestions());
|
||||||
|
return _suggestions;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Command register(JavaPlugin pl) {
|
||||||
|
if (!(parser instanceof LiteralArgParser))
|
||||||
|
throw new IllegalArgumentException("Only a command with a literal parser may be registered.");
|
||||||
|
if (parents.contains(pl)) throw new IllegalArgumentException("The command is already attached to the given plugin");
|
||||||
|
if (pl == null) throw new RuntimeException("pl is null");
|
||||||
|
parents.add(pl);
|
||||||
|
|
||||||
|
LiteralArgParser parser = (LiteralArgParser)this.parser;
|
||||||
|
String name = parser.getLiteral();
|
||||||
|
|
||||||
|
pl.getCommand(name).setAliases(parser.getAliases());
|
||||||
|
pl.getCommand(name).setExecutor((sender, cmd, alias, args) -> {
|
||||||
|
execute(sender, args);
|
||||||
|
return true;
|
||||||
|
});
|
||||||
|
pl.getCommand(name).setTabCompleter((sender, cmd, alias, args) -> {
|
||||||
|
return tabComplete(sender, args);
|
||||||
|
});
|
||||||
|
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Command(String name, ArgParser parser) {
|
||||||
|
this.name = name;
|
||||||
|
this.parser = parser;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Command literal(String name) {
|
||||||
|
Command cmd = createLiteral(name);
|
||||||
|
addChild(cmd);
|
||||||
|
return cmd;
|
||||||
|
}
|
||||||
|
public Command literal(String name, String ...aliases) {
|
||||||
|
Command cmd = createLiteral(name, aliases);
|
||||||
|
addChild(cmd);
|
||||||
|
return cmd;
|
||||||
|
}
|
||||||
|
public Command location(String name) {
|
||||||
|
Command cmd = createLocation(name);
|
||||||
|
addChild(cmd);
|
||||||
|
return cmd;
|
||||||
|
}
|
||||||
|
public Command _enum(String name, Class<? extends Enum<?>> enumType, boolean caseInsensitive) {
|
||||||
|
Command cmd = createEnum(name, enumType, caseInsensitive);
|
||||||
|
addChild(cmd);
|
||||||
|
return cmd;
|
||||||
|
}
|
||||||
|
public Command player(String name, boolean caseInsensitive) {
|
||||||
|
Command cmd = createPlayer(name, caseInsensitive);
|
||||||
|
addChild(cmd);
|
||||||
|
return cmd;
|
||||||
|
}
|
||||||
|
public Command string(String name, boolean greedy) {
|
||||||
|
Command cmd = createString(name, greedy);
|
||||||
|
addChild(cmd);
|
||||||
|
return cmd;
|
||||||
|
}
|
||||||
|
public Command _int(String name) {
|
||||||
|
Command cmd = createInt(name);
|
||||||
|
addChild(cmd);
|
||||||
|
return cmd;
|
||||||
|
}
|
||||||
|
public Command collection(String name, SimpleCollectionQuery query, boolean caseInsensitive) {
|
||||||
|
Command cmd = createCollection(name, query, caseInsensitive);
|
||||||
|
addChild(cmd);
|
||||||
|
return cmd;
|
||||||
|
}
|
||||||
|
public Command collection(String name, CollectionProvider provider, boolean caseInsensitive) {
|
||||||
|
Command cmd = createCollection(name, provider, caseInsensitive);
|
||||||
|
addChild(cmd);
|
||||||
|
return cmd;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static Command createCollection(String name, SimpleCollectionQuery query, boolean caseInsensitive) {
|
||||||
|
return new Command(name, new CollectionArgParser(
|
||||||
|
() -> query.get().stream().collect(Collectors.toMap(v->v, v->v)),
|
||||||
|
caseInsensitive)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
public static Command createCollection(String name, CollectionProvider provider, boolean caseInsensitive) {
|
||||||
|
return new Command(name, new CollectionArgParser(provider, caseInsensitive));
|
||||||
|
}
|
||||||
|
public static Command createPlayer(String name, boolean caseInsensitive) {
|
||||||
|
return new Command(name, new PlayerArgParser(caseInsensitive));
|
||||||
|
}
|
||||||
|
public static Command createEnum(String name, Class<? extends Enum<?>> enumType, boolean caseInsensitive) {
|
||||||
|
return new Command(name, new EnumArgParser(enumType, caseInsensitive));
|
||||||
|
}
|
||||||
|
public static Command createString(String name, boolean greedy) {
|
||||||
|
return new Command(name, new StringArgParser(greedy));
|
||||||
|
}
|
||||||
|
public static Command createInt(String name) {
|
||||||
|
return new Command(name, new IntArgParser());
|
||||||
|
}
|
||||||
|
public static Command createLocation(String name) {
|
||||||
|
return new Command(name, new LocationArgParser());
|
||||||
|
}
|
||||||
|
public static Command createLiteral(String lit) {
|
||||||
|
return new Command(lit, new LiteralArgParser(lit));
|
||||||
|
}
|
||||||
|
public static Command createLiteral(String lit, String ...aliases) {
|
||||||
|
return new Command(lit, new LiteralArgParser(lit, aliases));
|
||||||
|
}
|
||||||
|
|
||||||
|
public interface SimpleCollectionQuery {
|
||||||
|
Collection<String> get();
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user