mirror of
https://github.com/PaperMC/Paper.git
synced 2024-11-09 20:30:28 +01:00
222 lines
11 KiB
Diff
222 lines
11 KiB
Diff
--- a/net/minecraft/server/CommandDispatcher.java
|
|
+++ b/net/minecraft/server/CommandDispatcher.java
|
|
@@ -26,12 +26,21 @@
|
|
import org.apache.logging.log4j.LogManager;
|
|
import org.apache.logging.log4j.Logger;
|
|
|
|
+// CraftBukkit start
|
|
+import com.google.common.base.Joiner;
|
|
+import java.util.LinkedHashSet;
|
|
+import org.bukkit.craftbukkit.command.VanillaCommandWrapper;
|
|
+import org.bukkit.event.player.PlayerCommandSendEvent;
|
|
+import org.bukkit.event.server.ServerCommandEvent;
|
|
+// CraftBukkit end
|
|
+
|
|
public class CommandDispatcher {
|
|
|
|
private static final Logger a = LogManager.getLogger();
|
|
private final com.mojang.brigadier.CommandDispatcher<CommandListenerWrapper> b = new com.mojang.brigadier.CommandDispatcher();
|
|
|
|
- public CommandDispatcher(boolean flag) {
|
|
+ // CraftBukkit start
|
|
+ public final CommandDispatcher init(boolean flag) {
|
|
CommandAdvancement.a(this.b);
|
|
CommandExecute.a(this.b);
|
|
CommmandBossBar.a(this.b);
|
|
@@ -100,6 +109,11 @@
|
|
this.b.findAmbiguities((commandnode, commandnode1, commandnode2, collection) -> {
|
|
CommandDispatcher.a.warn("Ambiguity between arguments {} and {} with inputs: {}", this.b.getPath(commandnode1), this.b.getPath(commandnode2), collection);
|
|
});
|
|
+ return this;
|
|
+ }
|
|
+
|
|
+ public CommandDispatcher() {
|
|
+ // CraftBukkit end
|
|
this.b.setConsumer((commandcontext, flag, i) -> {
|
|
((CommandListenerWrapper) commandcontext.getSource()).a(commandcontext, flag, i);
|
|
});
|
|
@@ -114,8 +128,49 @@
|
|
|
|
}
|
|
|
|
+ // CraftBukkit start
|
|
+ public int dispatchServerCommand(CommandListenerWrapper sender, String command) {
|
|
+ Joiner joiner = Joiner.on(" ");
|
|
+ if (command.startsWith("/")) {
|
|
+ command = command.substring(1);
|
|
+ }
|
|
+
|
|
+ ServerCommandEvent event = new ServerCommandEvent(sender.getBukkitSender(), command);
|
|
+ org.bukkit.Bukkit.getPluginManager().callEvent(event);
|
|
+ if (event.isCancelled()) {
|
|
+ return 0;
|
|
+ }
|
|
+ command = event.getCommand();
|
|
+
|
|
+ String[] args = command.split(" ");
|
|
+
|
|
+ String cmd = args[0];
|
|
+ if (cmd.startsWith("minecraft:")) cmd = cmd.substring("minecraft:".length());
|
|
+ if (cmd.startsWith("bukkit:")) cmd = cmd.substring("bukkit:".length());
|
|
+
|
|
+ // Block disallowed commands
|
|
+ if (cmd.equalsIgnoreCase("stop") || cmd.equalsIgnoreCase("kick") || cmd.equalsIgnoreCase("op")
|
|
+ || cmd.equalsIgnoreCase("deop") || cmd.equalsIgnoreCase("ban") || cmd.equalsIgnoreCase("ban-ip")
|
|
+ || cmd.equalsIgnoreCase("pardon") || cmd.equalsIgnoreCase("pardon-ip") || cmd.equalsIgnoreCase("reload")) {
|
|
+ return 0;
|
|
+ }
|
|
+
|
|
+ // Handle vanilla commands;
|
|
+ if (sender.getWorld().getServer().getCommandBlockOverride(args[0])) {
|
|
+ args[0] = "minecraft:" + args[0];
|
|
+ }
|
|
+
|
|
+ return this.a(sender, joiner.join(args));
|
|
+ }
|
|
+
|
|
public int a(CommandListenerWrapper commandlistenerwrapper, String s) {
|
|
- String s1 = s;
|
|
+ return this.a(commandlistenerwrapper, s, s);
|
|
+ }
|
|
+
|
|
+ public int a(CommandListenerWrapper commandlistenerwrapper, String s, String label) {
|
|
+ VanillaCommandWrapper.WorldRescueContext rescue = new VanillaCommandWrapper.WorldRescueContext().start(commandlistenerwrapper.getWorld());
|
|
+ String s1 = label;
|
|
+ // CraftBukkit end
|
|
|
|
if (s.startsWith("/")) {
|
|
s = s.substring(1);
|
|
@@ -126,7 +181,6 @@
|
|
byte b0;
|
|
|
|
try {
|
|
- byte b1;
|
|
ChatComponentText chatcomponenttext;
|
|
|
|
try {
|
|
@@ -135,65 +189,95 @@
|
|
return i;
|
|
} catch (CommandException commandexception) {
|
|
commandlistenerwrapper.sendFailureMessage(commandexception.a());
|
|
- b1 = 0;
|
|
- return b1;
|
|
+ b0 = 0;
|
|
+ return b0;
|
|
} catch (CommandSyntaxException commandsyntaxexception) {
|
|
commandlistenerwrapper.sendFailureMessage(ChatComponentUtils.a(commandsyntaxexception.getRawMessage()));
|
|
if (commandsyntaxexception.getInput() != null && commandsyntaxexception.getCursor() >= 0) {
|
|
int j = Math.min(commandsyntaxexception.getInput().length(), commandsyntaxexception.getCursor());
|
|
- IChatBaseComponent ichatbasecomponent = (new ChatComponentText("")).a(EnumChatFormat.GRAY).a((chatmodifier) -> {
|
|
- chatmodifier.setChatClickable(new ChatClickable(ChatClickable.EnumClickAction.SUGGEST_COMMAND, s));
|
|
- });
|
|
|
|
+ chatcomponenttext = new ChatComponentText("");
|
|
if (j > 10) {
|
|
- ichatbasecomponent.a("...");
|
|
+ chatcomponenttext.a("...");
|
|
}
|
|
|
|
- ichatbasecomponent.a(commandsyntaxexception.getInput().substring(Math.max(0, j - 10), j));
|
|
+ chatcomponenttext.a(commandsyntaxexception.getInput().substring(Math.max(0, j - 10), j));
|
|
if (j < commandsyntaxexception.getInput().length()) {
|
|
- IChatBaseComponent ichatbasecomponent1 = (new ChatComponentText(commandsyntaxexception.getInput().substring(j))).a(new EnumChatFormat[] { EnumChatFormat.RED, EnumChatFormat.UNDERLINE});
|
|
+ ChatComponentText chatcomponenttext1 = new ChatComponentText(commandsyntaxexception.getInput().substring(j));
|
|
|
|
- ichatbasecomponent.addSibling(ichatbasecomponent1);
|
|
+ chatcomponenttext1.getChatModifier().setColor(EnumChatFormat.RED);
|
|
+ chatcomponenttext1.getChatModifier().setUnderline(Boolean.valueOf(true));
|
|
+ chatcomponenttext.addSibling(chatcomponenttext1);
|
|
}
|
|
|
|
- ichatbasecomponent.addSibling((new ChatMessage("command.context.here", new Object[0])).a(new EnumChatFormat[] { EnumChatFormat.RED, EnumChatFormat.ITALIC}));
|
|
- commandlistenerwrapper.sendFailureMessage(ichatbasecomponent);
|
|
+ ChatMessage chatmessage = new ChatMessage("command.context.here", new Object[0]);
|
|
+
|
|
+ chatmessage.getChatModifier().setItalic(Boolean.valueOf(true));
|
|
+ chatmessage.getChatModifier().setColor(EnumChatFormat.RED);
|
|
+ chatcomponenttext.addSibling(chatmessage);
|
|
+ chatcomponenttext.getChatModifier().setColor(EnumChatFormat.GRAY);
|
|
+ chatcomponenttext.getChatModifier().setChatClickable(new ChatClickable(ChatClickable.EnumClickAction.SUGGEST_COMMAND, s1));
|
|
+ commandlistenerwrapper.sendFailureMessage(chatcomponenttext);
|
|
}
|
|
|
|
- b1 = 0;
|
|
- return b1;
|
|
+ b0 = 0;
|
|
} catch (Exception exception) {
|
|
- chatcomponenttext = new ChatComponentText;
|
|
- }
|
|
-
|
|
- chatcomponenttext.<init>(exception.getMessage() == null ? exception.getClass().getName() : exception.getMessage());
|
|
- ChatComponentText chatcomponenttext1 = chatcomponenttext;
|
|
+ ChatMessage chatmessage1 = new ChatMessage("command.failed", new Object[0]);
|
|
|
|
- if (CommandDispatcher.a.isDebugEnabled()) {
|
|
- StackTraceElement[] astacktraceelement = exception.getStackTrace();
|
|
+ chatcomponenttext = new ChatComponentText(exception.getMessage() == null ? exception.getClass().getName() : exception.getMessage());
|
|
+ if (CommandDispatcher.a.isDebugEnabled()) {
|
|
+ StackTraceElement[] astacktraceelement = exception.getStackTrace();
|
|
|
|
- for (int k = 0; k < Math.min(astacktraceelement.length, 3); ++k) {
|
|
- chatcomponenttext1.a("\n\n").a(astacktraceelement[k].getMethodName()).a("\n ").a(astacktraceelement[k].getFileName()).a(":").a(String.valueOf(astacktraceelement[k].getLineNumber()));
|
|
+ for (int k = 0; k < Math.min(astacktraceelement.length, 3); ++k) {
|
|
+ chatcomponenttext.a("\n\n" + astacktraceelement[k].getMethodName() + "\n " + astacktraceelement[k].getFileName() + ":" + astacktraceelement[k].getLineNumber());
|
|
+ }
|
|
}
|
|
- }
|
|
|
|
- commandlistenerwrapper.sendFailureMessage((new ChatMessage("command.failed", new Object[0])).a((chatmodifier) -> {
|
|
- chatmodifier.setChatHoverable(new ChatHoverable(ChatHoverable.EnumHoverAction.SHOW_TEXT, ichatbasecomponent));
|
|
- }));
|
|
- b0 = 0;
|
|
+ chatmessage1.getChatModifier().setChatHoverable(new ChatHoverable(ChatHoverable.EnumHoverAction.SHOW_TEXT, chatcomponenttext));
|
|
+ commandlistenerwrapper.sendFailureMessage(chatmessage1);
|
|
+ byte b1 = 0;
|
|
+
|
|
+ return b1;
|
|
+ }
|
|
} finally {
|
|
commandlistenerwrapper.getServer().methodProfiler.e();
|
|
+ rescue.end(); // CraftBukkit
|
|
}
|
|
|
|
return b0;
|
|
}
|
|
|
|
public void a(EntityPlayer entityplayer) {
|
|
- HashMap hashmap = Maps.newHashMap();
|
|
- RootCommandNode rootcommandnode = new RootCommandNode();
|
|
+ // CraftBukkit start
|
|
+ // Register Vanilla commands into builtRoot as before
|
|
+ Map hashmap = Maps.newIdentityHashMap(); // Use identity to prevent aliasing issues
|
|
+ RootCommandNode vanillaRoot = new RootCommandNode();
|
|
+
|
|
+ RootCommandNode<CommandListenerWrapper> vanilla = entityplayer.server.vanillaCommandDispatcher.a().getRoot();
|
|
+ hashmap.put(vanilla, vanillaRoot);
|
|
+ this.a(vanilla, vanillaRoot, entityplayer.getCommandListener(), (Map) hashmap);
|
|
+
|
|
+ // Now build the global commands in a second pass
|
|
+ RootCommandNode<ICompletionProvider> rootcommandnode = new RootCommandNode();
|
|
|
|
hashmap.put(this.b.getRoot(), rootcommandnode);
|
|
this.a(this.b.getRoot(), rootcommandnode, entityplayer.getCommandListener(), (Map) hashmap);
|
|
+
|
|
+ Collection<String> bukkit = new LinkedHashSet<>();
|
|
+ for (CommandNode node : rootcommandnode.getChildren()) {
|
|
+ bukkit.add(node.getName());
|
|
+ }
|
|
+
|
|
+ PlayerCommandSendEvent event = new PlayerCommandSendEvent(entityplayer.getBukkitEntity(), new LinkedHashSet<>(bukkit));
|
|
+ event.getPlayer().getServer().getPluginManager().callEvent(event);
|
|
+
|
|
+ // Remove labels that were removed during the event
|
|
+ for (String orig : bukkit) {
|
|
+ if (!event.getCommands().contains(orig)) {
|
|
+ rootcommandnode.removeCommand(orig);
|
|
+ }
|
|
+ }
|
|
+ // CraftBukkit end
|
|
entityplayer.playerConnection.sendPacket(new PacketPlayOutCommands(rootcommandnode));
|
|
}
|
|
|