mirror of
https://github.com/kaboomserver/extras.git
synced 2024-05-16 09:51:18 +00:00
db50f0b7ae
* Use more Java 17 features in classes * created more new lines * fixed long lines
221 lines
8.9 KiB
Java
221 lines
8.9 KiB
Java
package pw.kaboom.extras.modules.server;
|
|
|
|
import org.bukkit.block.CommandBlock;
|
|
import org.bukkit.command.BlockCommandSender;
|
|
import org.bukkit.command.CommandSender;
|
|
import org.bukkit.entity.minecart.CommandMinecart;
|
|
import org.bukkit.event.EventHandler;
|
|
import org.bukkit.event.Listener;
|
|
import org.bukkit.event.server.ServerCommandEvent;
|
|
import org.bukkit.plugin.java.JavaPlugin;
|
|
import pw.kaboom.extras.Main;
|
|
|
|
import java.util.Arrays;
|
|
import java.util.Objects;
|
|
import java.util.logging.Level;
|
|
import java.util.logging.Logger;
|
|
import java.util.regex.Matcher;
|
|
import java.util.regex.Pattern;
|
|
|
|
public final class ServerCommand implements Listener {
|
|
private static final Pattern AS_AT_PATTERN = Pattern.compile(
|
|
"\\b(as|at|facing entity) @[ae]\\b");
|
|
private static final Pattern SELECTOR_PATTERN = Pattern.compile("(?>\\s)*@[aepsr](?>\\s)*");
|
|
private static final Logger LOGGER = JavaPlugin.getPlugin(Main.class).getLogger();
|
|
|
|
private static final String[] COMMANDS = { "execute", "clone", "fill", "forceload", "kick",
|
|
"locate", "locatebiome", "me", "msg", "reload", "save-all", "say", "spreadplayers",
|
|
"stop", "summon", "teammsg", "teleport", "tell", "tellraw", "tm", "tp", "w", "place",
|
|
"fillbiome", "ride" };
|
|
|
|
public static boolean checkExecuteCommand(final String cmd) {
|
|
for (String command : COMMANDS) {
|
|
if (command.equalsIgnoreCase(cmd)) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private static String checkSelectors(final String[] arr) {
|
|
final String[] args = Arrays.copyOfRange(arr, 1, arr.length);
|
|
final String str = String.join(" ", args);
|
|
final Matcher matcher = SELECTOR_PATTERN.matcher(str);
|
|
final long count = matcher.results().count();
|
|
|
|
if (count > 3) {
|
|
return "cancel";
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public static String checkCommand(final CommandSender sender, final String command,
|
|
final boolean isConsoleCommand) {
|
|
return checkCommand(sender, command, isConsoleCommand, 1);
|
|
}
|
|
|
|
public static String checkCommand(final CommandSender sender, final String command,
|
|
final boolean isConsoleCommand, int depth) {
|
|
if (depth > 50) {
|
|
return "cancel";
|
|
}
|
|
|
|
final String[] arr = command.split(" ");
|
|
String commandName = arr[0].toLowerCase();
|
|
|
|
if (isConsoleCommand) {
|
|
commandName = "/" + commandName;
|
|
} else if (arr.length >= 2 && commandName.equals("/")) {
|
|
// Command could contain spaces after the slash, e.g. "/ spawn"
|
|
commandName = "/" + arr[1].toLowerCase();
|
|
}
|
|
|
|
try {
|
|
switch (commandName) {
|
|
case "/minecraft:execute", "/execute" -> {
|
|
if (arr.length >= 2) {
|
|
int asAtCount = 0;
|
|
Matcher asAtMatcher = AS_AT_PATTERN.matcher(command.toLowerCase());
|
|
while (asAtMatcher.find()) {
|
|
asAtCount++;
|
|
}
|
|
if (asAtCount >= 2) {
|
|
return "cancel";
|
|
}
|
|
|
|
for (int i = 1; i < arr.length; i++) {
|
|
if ("summon".equalsIgnoreCase(arr[i])) {
|
|
return "cancel";
|
|
}
|
|
if (!"run".equalsIgnoreCase(arr[i])) {
|
|
continue;
|
|
}
|
|
if (i + 1 == arr.length) {
|
|
break;
|
|
}
|
|
if (checkExecuteCommand(arr[i + 1])) {
|
|
return "cancel";
|
|
}
|
|
final String[] executeCommand = Arrays.copyOfRange(
|
|
arr, i + 1, arr.length);
|
|
final String result = checkCommand(sender,
|
|
String.join(" ", executeCommand), true, depth + 1);
|
|
if (result == null) {
|
|
continue;
|
|
} else if (result.equals("cancel")) {
|
|
return "cancel";
|
|
}
|
|
final String pureExecute = String.join(
|
|
" ", Arrays.copyOfRange(arr, 0, i + 1));
|
|
final String finalResult = checkCommand(sender,
|
|
pureExecute + " " + result, isConsoleCommand, depth + 1);
|
|
return Objects.requireNonNullElseGet(finalResult,
|
|
() -> pureExecute + " " + result);
|
|
}
|
|
}
|
|
}
|
|
case "/minecraft:fill", "/fill" -> {
|
|
if (command.contains("auto")) {
|
|
return command.replace("auto", "[auto]");
|
|
}
|
|
}
|
|
case "/minecraft:give", "/give" -> {
|
|
if (Double.parseDouble(arr[arr.length - 1]) > 64) {
|
|
// Limit item count
|
|
arr[arr.length - 1] = "64";
|
|
return String.join(" ", arr);
|
|
}
|
|
}
|
|
case "/minecraft:particle", "/particle" -> {
|
|
int[] numArgs = {14, 10};
|
|
for (int i : numArgs) {
|
|
if (arr.length < i || arr.length > i + 2) {
|
|
continue;
|
|
}
|
|
if (Double.parseDouble(arr[i - 1]) > 10) {
|
|
// Limit particle count
|
|
arr[i - 1] = "10";
|
|
return String.join(" ", arr);
|
|
}
|
|
}
|
|
}
|
|
case "/minecraft:ban", "/ban", "/minecraft:kick", "/kick",
|
|
"/minecraft:tell", "/tell", "/minecraft:msg", "/msg",
|
|
"/minecraft:w", "/w", "/minecraft:say", "/say" -> {
|
|
return checkSelectors(arr);
|
|
}
|
|
case "/minecraft:spreadplayers", "/spreadplayers" -> {
|
|
if (arr.length == 7 && (arr[6].contains("@e") || arr[6].contains("@a"))) {
|
|
return "cancel";
|
|
} else if (arr.length >= 5) {
|
|
if (Double.parseDouble(arr[3]) > 0) {
|
|
arr[3] = "0";
|
|
}
|
|
if (Double.parseDouble(arr[4]) < 8) {
|
|
arr[4] = "8";
|
|
}
|
|
if (Double.parseDouble(arr[4]) > 50) {
|
|
arr[4] = "50";
|
|
}
|
|
return String.join(" ", arr);
|
|
}
|
|
}
|
|
case "/viaversion:viaver", "/viaversion:viaversion", "/viaversion:vvbukkit",
|
|
"/viaver", "/viaversion", "/vvbukkit" -> {
|
|
if (arr.length >= 2
|
|
&& "debug".equalsIgnoreCase(arr[1])) {
|
|
return "cancel";
|
|
}
|
|
}
|
|
case "/scissors:scissors", "/scissors" -> {
|
|
if (arr.length >= 2
|
|
&& "reload".equalsIgnoreCase(arr[1])) {
|
|
return "cancel";
|
|
}
|
|
}
|
|
case "/geyser-spigot:geyser", "/geyser" -> {
|
|
if (arr.length >= 2
|
|
&& "dump".equalsIgnoreCase(arr[1])) {
|
|
return "cancel";
|
|
}
|
|
}
|
|
default -> {
|
|
}
|
|
}
|
|
} catch (NumberFormatException exception) {
|
|
// Do nothing
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
@EventHandler
|
|
void onServerCommand(final ServerCommandEvent event) {
|
|
final CommandSender sender = event.getSender();
|
|
|
|
if (sender instanceof BlockCommandSender blockCommandSender) {
|
|
final var commandBlock = (CommandBlock) blockCommandSender.getBlock().getState();
|
|
|
|
commandBlock.setCommand("");
|
|
commandBlock.update();
|
|
} else if (sender instanceof CommandMinecart commandMinecart) {
|
|
commandMinecart.setCommand("");
|
|
}
|
|
|
|
final String command = event.getCommand();
|
|
final boolean isConsoleCommand = true;
|
|
final String checkedCommand = checkCommand(sender, command, isConsoleCommand);
|
|
|
|
if (checkedCommand != null) {
|
|
if ("cancel".equals(checkedCommand)) {
|
|
event.setCancelled(true);
|
|
} else {
|
|
event.setCommand(checkedCommand);
|
|
}
|
|
}
|
|
|
|
LOGGER.log(Level.INFO, "Console command: " + command);
|
|
}
|
|
}
|