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 java.util.ArrayList;
027import java.util.List;
028import java.util.concurrent.CompletableFuture;
029
030import com.google.common.collect.HashMultimap;
031import com.google.common.collect.SetMultimap;
032import com.velocitypowered.api.command.SimpleCommand;
033
034public class VelocityRootCommand implements SimpleCommand, RootCommand {
035
036    private final VelocityCommandManager manager;
037    private final String name;
038    private BaseCommand defCommand;
039    private SetMultimap<String, RegisteredCommand> subCommands = HashMultimap.create();
040    private List<BaseCommand> children = new ArrayList<>();
041    boolean isRegistered = false;
042
043    VelocityRootCommand(VelocityCommandManager manager, String name) {
044        this.manager = manager;
045        this.name = name;
046    }
047
048    @Override
049    public String getCommandName() {
050        return name;
051    }
052
053    @Override
054    public void addChild(BaseCommand command) {
055        if (this.defCommand == null || !command.subCommands.get(BaseCommand.DEFAULT).isEmpty()) {
056            this.defCommand = command;
057
058        }
059        this.addChildShared(this.children, this.subCommands, command);
060    }
061
062    @Override
063    public CommandManager getManager() {
064        return manager;
065    }
066
067    @Override
068    public SetMultimap<String, RegisteredCommand> getSubCommands() {
069        return subCommands;
070    }
071
072    @Override
073    public List<BaseCommand> getChildren() {
074        return children;
075    }
076
077    @Override
078    public BaseCommand getDefCommand() {
079        return defCommand;
080    }
081
082    @Override
083    public void execute(Invocation invocation) {
084        execute(manager.getCommandIssuer(invocation.source()), getCommandName(), invocation.arguments());
085    }
086
087
088    @Override
089    public List<String> suggest(Invocation invocation) {
090        return getTabCompletions(manager.getCommandIssuer(invocation.source()), getCommandName(), invocation.arguments());
091    }
092
093    @Override
094    public CompletableFuture<List<String>> suggestAsync(Invocation invocation) {
095        return CompletableFuture.completedFuture(getTabCompletions(manager.getCommandIssuer(invocation.source()), getCommandName(), invocation.arguments()));
096    }
097}