001/*
002 * Copyright (c) 2016-2017 Daniel Ennis (Aikar) - MIT License
003 *
004 *  Permission is hereby granted, free of charge, to any person obtaining
005 *  a copy of this software and associated documentation files (the
006 *  "Software"), to deal in the Software without restriction, including
007 *  without limitation the rights to use, copy, modify, merge, publish,
008 *  distribute, sublicense, and/or sell copies of the Software, and to
009 *  permit persons to whom the Software is furnished to do so, subject to
010 *  the following conditions:
011 *
012 *  The above copyright notice and this permission notice shall be
013 *  included in all copies or substantial portions of the Software.
014 *
015 *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
016 *  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
017 *  MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
018 *  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
019 *  LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
020 *  OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
021 *  WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
022 */
023
024package co.aikar.commands;
025
026import com.google.common.collect.HashMultimap;
027import com.google.common.collect.SetMultimap;
028import net.md_5.bungee.api.CommandSender;
029import net.md_5.bungee.api.plugin.Command;
030import net.md_5.bungee.api.plugin.TabExecutor;
031
032import java.util.List;
033import java.util.ArrayList;
034import java.util.Set;
035import java.util.HashSet;
036
037public class BungeeRootCommand extends Command implements RootCommand, TabExecutor {
038
039    private final BungeeCommandManager manager;
040    private final String name;
041    private BaseCommand defCommand;
042    private SetMultimap<String, RegisteredCommand> subCommands = HashMultimap.create();
043    private List<BaseCommand> children = new ArrayList<>();
044    boolean isRegistered = false;
045
046    BungeeRootCommand(BungeeCommandManager manager, String name) {
047        super(name);
048        this.manager = manager;
049        this.name = name;
050    }
051
052    @Override
053    public String getCommandName() {
054        return name;
055    }
056
057    @Override
058    public void addChild(BaseCommand command) {
059        if (this.defCommand == null || !command.subCommands.get("__default").isEmpty()) {
060            this.defCommand = command;
061
062        }
063        this.addChildShared(this.children, this.subCommands, command);
064    }
065
066    @Override
067    public CommandManager getManager() {
068        return manager;
069    }
070
071    @Override
072    public SetMultimap<String, RegisteredCommand> getSubCommands() {
073        return subCommands;
074    }
075
076    @Override
077    public void execute(CommandSender sender, String[] args) {
078        execute(manager.getCommandIssuer(sender), getName(), args);
079    }
080
081    @Override
082    public Iterable<String> onTabComplete(CommandSender commandSender, String[] strings) {
083        return onTabComplete(manager.getCommandIssuer(commandSender), getName(), strings);
084    }
085
086    private List<String> onTabComplete(CommandIssuer sender, String alias, String[] args) throws IllegalArgumentException {
087        Set<String> completions = new HashSet<>();
088        this.children.forEach(child -> completions.addAll(child.tabComplete(sender, alias, args)));
089        return new ArrayList<>(completions);
090    }
091
092    @Override
093    public BaseCommand getDefCommand() {
094        return defCommand;
095    }
096}