2019-04-26 02:36:22 +00:00
|
|
|
/*
|
|
|
|
* WorldEdit, a Minecraft world manipulation toolkit
|
|
|
|
* Copyright (C) sk89q <http://www.sk89q.com>
|
|
|
|
* Copyright (C) WorldEdit team and contributors
|
|
|
|
*
|
|
|
|
* This program 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, either version 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This program 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 program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
|
2019-04-30 22:03:18 +00:00
|
|
|
package com.sk89q.worldedit.internal.command;
|
2019-04-26 02:36:22 +00:00
|
|
|
|
2019-05-15 00:57:05 +00:00
|
|
|
import com.google.common.collect.ImmutableList;
|
2019-05-06 05:06:20 +00:00
|
|
|
import com.google.common.collect.Iterables;
|
2019-04-26 08:18:02 +00:00
|
|
|
import com.sk89q.worldedit.extension.platform.PlatformCommandManager;
|
2019-05-06 05:06:20 +00:00
|
|
|
import com.sk89q.worldedit.internal.util.Substring;
|
2019-05-15 00:57:05 +00:00
|
|
|
import com.sk89q.worldedit.util.formatting.text.Component;
|
|
|
|
import com.sk89q.worldedit.util.formatting.text.TextComponent;
|
2019-04-26 02:36:22 +00:00
|
|
|
import org.enginehub.piston.Command;
|
2019-05-15 00:57:05 +00:00
|
|
|
import org.enginehub.piston.exception.CommandException;
|
2019-04-26 02:36:22 +00:00
|
|
|
import org.enginehub.piston.part.SubCommandPart;
|
|
|
|
|
|
|
|
import java.util.Comparator;
|
2019-05-06 05:06:20 +00:00
|
|
|
import java.util.List;
|
2019-04-26 02:36:22 +00:00
|
|
|
import java.util.Map;
|
2019-05-06 05:06:20 +00:00
|
|
|
import java.util.Optional;
|
2019-04-26 02:36:22 +00:00
|
|
|
import java.util.function.Function;
|
|
|
|
import java.util.stream.Collectors;
|
|
|
|
|
2019-05-06 05:06:20 +00:00
|
|
|
import static com.google.common.base.Preconditions.checkState;
|
|
|
|
import static java.util.stream.Collectors.toList;
|
|
|
|
|
2019-04-26 02:36:22 +00:00
|
|
|
public class CommandUtil {
|
|
|
|
|
|
|
|
public static Map<String, Command> getSubCommands(Command currentCommand) {
|
|
|
|
return currentCommand.getParts().stream()
|
|
|
|
.filter(p -> p instanceof SubCommandPart)
|
|
|
|
.flatMap(p -> ((SubCommandPart) p).getCommands().stream())
|
|
|
|
.collect(Collectors.toMap(Command::getName, Function.identity()));
|
|
|
|
}
|
|
|
|
|
|
|
|
private static String clean(String input) {
|
2019-04-26 08:18:02 +00:00
|
|
|
return PlatformCommandManager.COMMAND_CLEAN_PATTERN.matcher(input).replaceAll("");
|
2019-04-26 02:36:22 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
private static final Comparator<Command> BY_CLEAN_NAME =
|
|
|
|
Comparator.comparing(c -> clean(c.getName()));
|
|
|
|
|
|
|
|
public static Comparator<Command> byCleanName() {
|
|
|
|
return BY_CLEAN_NAME;
|
|
|
|
}
|
|
|
|
|
2019-05-06 05:06:20 +00:00
|
|
|
/**
|
|
|
|
* Fix {@code suggestions} to replace the last space-separated word in {@code arguments}.
|
|
|
|
*/
|
|
|
|
public static List<String> fixSuggestions(String arguments, List<Substring> suggestions) {
|
|
|
|
Substring lastArg = Iterables.getLast(CommandArgParser.spaceSplit(arguments));
|
|
|
|
return suggestions.stream()
|
|
|
|
.map(suggestion -> CommandUtil.suggestLast(lastArg, suggestion))
|
|
|
|
.filter(Optional::isPresent)
|
|
|
|
.map(Optional::get)
|
|
|
|
.collect(toList());
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Given the last word of a command, mutate the suggestion to replace the last word, if
|
|
|
|
* possible.
|
|
|
|
*/
|
|
|
|
private static Optional<String> suggestLast(Substring last, Substring suggestion) {
|
|
|
|
if (suggestion.getStart() == last.getEnd()) {
|
|
|
|
// this suggestion is for the next argument.
|
|
|
|
if (last.getSubstring().isEmpty()) {
|
|
|
|
return Optional.of(suggestion.getSubstring());
|
|
|
|
}
|
|
|
|
return Optional.of(last.getSubstring() + " " + suggestion.getSubstring());
|
|
|
|
}
|
|
|
|
StringBuilder builder = new StringBuilder(last.getSubstring());
|
|
|
|
int start = suggestion.getStart() - last.getStart();
|
|
|
|
int end = suggestion.getEnd() - last.getStart();
|
|
|
|
if (start < 0) {
|
|
|
|
// Quoted suggestion, can't complete it here.
|
|
|
|
return Optional.empty();
|
|
|
|
}
|
|
|
|
checkState(end <= builder.length(),
|
|
|
|
"Suggestion ends too late, last=%s, suggestion=", last, suggestion);
|
|
|
|
builder.replace(start, end, suggestion.getSubstring());
|
|
|
|
return Optional.of(builder.toString());
|
|
|
|
}
|
|
|
|
|
2019-05-15 00:57:05 +00:00
|
|
|
/**
|
|
|
|
* Require {@code condition} to be {@code true}, otherwise throw a {@link CommandException}
|
|
|
|
* with the given message.
|
|
|
|
*
|
|
|
|
* @param condition the condition to check
|
|
|
|
* @param message the message for failure
|
|
|
|
*/
|
|
|
|
public static void checkCommandArgument(boolean condition, String message) {
|
|
|
|
checkCommandArgument(condition, TextComponent.of(message));
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Require {@code condition} to be {@code true}, otherwise throw a {@link CommandException}
|
|
|
|
* with the given message.
|
|
|
|
*
|
|
|
|
* @param condition the condition to check
|
|
|
|
* @param message the message for failure
|
|
|
|
*/
|
|
|
|
public static void checkCommandArgument(boolean condition, Component message) {
|
|
|
|
if (!condition) {
|
|
|
|
throw new CommandException(message, ImmutableList.of());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-04-26 02:36:22 +00:00
|
|
|
private CommandUtil() {
|
|
|
|
}
|
|
|
|
}
|