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 org.jetbrains.annotations.NotNull; 029import org.spongepowered.api.command.CommandCallable; 030import org.spongepowered.api.command.CommandException; 031import org.spongepowered.api.command.CommandResult; 032import org.spongepowered.api.command.CommandSource; 033import org.spongepowered.api.text.Text; 034import org.spongepowered.api.world.Location; 035import org.spongepowered.api.world.World; 036 037import javax.annotation.Nullable; 038import java.util.ArrayList; 039import java.util.List; 040import java.util.Optional; 041 042public class SpongeRootCommand implements CommandCallable, RootCommand { 043 044 private final SpongeCommandManager manager; 045 private final String name; 046 private BaseCommand defCommand; 047 private SetMultimap<String, RegisteredCommand> subCommands = HashMultimap.create(); 048 private List<BaseCommand> children = new ArrayList<>(); 049 boolean isRegistered = false; 050 051 SpongeRootCommand(SpongeCommandManager manager, String name) { 052 this.manager = manager; 053 this.name = name; 054 } 055 056 @Override 057 public String getCommandName() { 058 return name; 059 } 060 061 @Override 062 public CommandResult process(@NotNull CommandSource source, @NotNull String arguments) throws CommandException { 063 String[] args = arguments.isEmpty() ? new String[0] : arguments.split(" "); 064 return this.executeSponge(manager.getCommandIssuer(source), this.name, args); 065 } 066 067 @Override 068 public List<String> getSuggestions(@NotNull CommandSource source, @NotNull String arguments, @Nullable Location<World> location) throws CommandException { 069 String[] args = arguments.isEmpty() ? new String[]{""} : arguments.split(" "); 070 return getTabCompletions(manager.getCommandIssuer(source), this.name, args); 071 } 072 073 @Override 074 public boolean testPermission(@NotNull CommandSource source) { 075 return this.hasAnyPermission(manager.getCommandIssuer(source)); 076 } 077 078 @Override 079 public Optional<Text> getShortDescription(@NotNull CommandSource source) { 080 String description = getDescription(); 081 return description != null ? Optional.of(Text.of(description)) : Optional.empty(); 082 } 083 084 @Override 085 public Optional<Text> getHelp(@NotNull CommandSource source) { 086 return Optional.empty(); 087 } 088 089 @Override 090 public Text getUsage(@NotNull CommandSource source) { 091 String usage = getUsage(); 092 return usage != null ? Text.of(usage) : Text.of(); 093 } 094 095 private CommandResult executeSponge(CommandIssuer sender, String commandLabel, String[] args) { 096 BaseCommand cmd = execute(sender, commandLabel, args); 097 SpongeCommandOperationContext lastContext = (SpongeCommandOperationContext) cmd.getLastCommandOperationContext(); 098 return lastContext != null ? lastContext.getResult() : CommandResult.success(); 099 } 100 101 public void addChild(BaseCommand command) { 102 if (this.defCommand == null || !command.subCommands.get(BaseCommand.DEFAULT).isEmpty()) { 103 this.defCommand = command; 104 } 105 addChildShared(this.children, this.subCommands, command); 106 } 107 108 @Override 109 public BaseCommand getDefCommand() { 110 return defCommand; 111 } 112 113 @Override 114 public CommandManager getManager() { 115 return manager; 116 } 117 118 @Override 119 public SetMultimap<String, RegisteredCommand> getSubCommands() { 120 return subCommands; 121 } 122 123 @Override 124 public List<BaseCommand> getChildren() { 125 return children; 126 } 127}