Added CommandTreeBase (#3193)

src/main/java/net/minecraftforge/server/command/CommandTreeBase.java

@@ -0,0 +1,138 @@
+/*
+ * Minecraft Forge
+ * Copyright (c) 2016.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation version 2.1
+ * of the License.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+ */
+
+package net.minecraftforge.server.command;
+
+import net.minecraft.command.CommandBase;
+import net.minecraft.command.CommandException;
+import net.minecraft.command.ICommand;
+import net.minecraft.command.ICommandSender;
+import net.minecraft.server.MinecraftServer;
+import net.minecraft.util.math.BlockPos;
+import net.minecraft.util.text.TextComponentString;
+
+import javax.annotation.Nullable;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.List;
+
+/**
+ * Base class for commands that has subcommands.
+ * <p>
+ * E.g. /team settings set [value]
+ * settings is subcommand of team and set is subcommand of settings
+ */
+public abstract class CommandTreeBase extends CommandBase
+{
+    private final HashMap<String, ICommand> commandMap = new HashMap<String, ICommand>();
+
+    public void addSubcommand(ICommand c)
+    {
+        commandMap.put(c.getCommandName(), c);
+    }
+
+    public Collection<ICommand> getSubCommands()
+    {
+        return Collections.unmodifiableCollection(commandMap.values());
+    }
+
+    private static String[] shiftArgs(String[] s)
+    {
+        if(s == null || s.length == 0)
+        {
+            return new String[0];
+        }
+
+        String[] s1 = new String[s.length - 1];
+        System.arraycopy(s, 1, s1, 0, s1.length);
+        return s1;
+    }
+
+    @Override
+    public List<String> getTabCompletionOptions(MinecraftServer server, ICommandSender sender, String[] args, @Nullable BlockPos pos)
+    {
+        if(args.length == 1)
+        {
+            List<String> keys = new ArrayList<String>();
+
+            for(ICommand c : commandMap.values())
+            {
+                if(c.checkPermission(server, sender))
+                {
+                    keys.add(c.getCommandName());
+                }
+            }
+
+            Collections.sort(keys, null);
+            return getListOfStringsMatchingLastWord(args, keys);
+        }
+
+        ICommand cmd = commandMap.get(args[0]);
+
+        if(cmd != null)
+        {
+            return cmd.getTabCompletionOptions(server, sender, shiftArgs(args), pos);
+        }
+
+        return super.getTabCompletionOptions(server, sender, args, pos);
+    }
+
+    @Override
+    public boolean isUsernameIndex(String[] args, int index)
+    {
+        if(index > 0 && args.length > 1)
+        {
+            ICommand cmd = commandMap.get(args[0]);
+            if(cmd != null)
+            {
+                return cmd.isUsernameIndex(shiftArgs(args), index - 1);
+            }
+        }
+
+        return false;
+    }
+
+    @Override
+    public void execute(MinecraftServer server, ICommandSender sender, String[] args) throws CommandException
+    {
+        if(args.length < 1)
+        {
+            sender.addChatMessage(new TextComponentString(CommandBase.joinNiceStringFromCollection(commandMap.keySet())));
+        }
+        else
+        {
+            ICommand cmd = commandMap.get(args[0]);
+
+            if(cmd == null)
+            {
+                throw new CommandException("commands.tree_base.invalid_cmd", args[0]);
+            }
+            else if(!cmd.checkPermission(server, sender))
+            {
+                throw new CommandException("commands.generic.permission");
+            }
+            else
+            {
+                cmd.execute(server, sender, shiftArgs(args));
+            }
+        }
+    }
+}