Add /testfor command support to command block. Adds BUKKIT-3813

1 files changed, 23 insertions, 2 deletions

diff --git a/src/main/java/net/minecraft/server/TileEntityCommand.java b/src/main/java/net/minecraft/server/TileEntityCommand.java
index f05573d1..9967524d 100644
--- a/src/main/java/net/minecraft/server/TileEntityCommand.java
+++ b/src/main/java/net/minecraft/server/TileEntityCommand.java
@@ -31,7 +31,7 @@ public class TileEntityCommand extends TileEntity implements ICommandListener {
             MinecraftServer minecraftserver = MinecraftServer.getServer();
 
             if (minecraftserver != null && minecraftserver.getEnableCommandBlock()) {
-                // CraftBukkit start - handle command block as console
+                // CraftBukkit start - handle command block commands using Bukkit dispatcher
                 org.bukkit.command.SimpleCommandMap commandMap = minecraftserver.server.getCommandMap();
                 Joiner joiner = Joiner.on(" ");
                 String command = this.b;
@@ -58,6 +58,26 @@ public class TileEntityCommand extends TileEntity implements ICommandListener {
                     return 0;
                 }
 
+                // testfor command requires special handling
+                if (args[0].equalsIgnoreCase("testfor")) {
+                    if (args.length < 2) {
+                        return 0;
+                    }
+
+                    EntityPlayer[] players = PlayerSelector.getPlayers(this, args[1]);
+
+                    if (players != null && players.length > 0) {
+                        return players.length;
+                    } else {
+                        EntityPlayer player = MinecraftServer.getServer().getPlayerList().f(args[1]); // Should be getPlayer
+                        if (player == null) {
+                            return 0;
+                        } else {
+                            return 1;
+                        }
+                    }
+                }
+
                 commands.add(args);
 
                 // find positions of command block syntax, if any
@@ -86,8 +106,9 @@ public class TileEntityCommand extends TileEntity implements ICommandListener {
                         minecraftserver.getLogger().warning(String.format("CommandBlock at (%d,%d,%d) failed to handle command", this.x, this.y, this.z), exception);
                     }
                 }
-                // CraftBukkit end
+
                 return completed;
+                // CraftBukkit end
             } else {
                 return 0;
             }