Paper command

Co-authored-by: Zach Brown <1254957+zachbr@users.noreply.github.com>
This commit is contained in:
Jason Penilla
2016-02-29 21:02:09 -06:00
parent a0389538ec
commit bd7282c66d
11 changed files with 586 additions and 11 deletions

View File

@@ -0,0 +1,158 @@
package io.papermc.paper.command.subcommands;
import com.google.common.collect.Maps;
import io.papermc.paper.command.CommandUtil;
import io.papermc.paper.command.PaperSubcommand;
import java.util.Collections;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;
import net.kyori.adventure.text.Component;
import net.kyori.adventure.text.event.ClickEvent;
import net.kyori.adventure.text.event.HoverEvent;
import net.minecraft.core.registries.BuiltInRegistries;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.server.level.ServerChunkCache;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.world.entity.EntityType;
import net.minecraft.world.level.ChunkPos;
import org.apache.commons.lang3.tuple.MutablePair;
import org.apache.commons.lang3.tuple.Pair;
import org.bukkit.Bukkit;
import org.bukkit.HeightMap;
import org.bukkit.World;
import org.bukkit.command.CommandSender;
import org.bukkit.craftbukkit.CraftWorld;
import org.bukkit.entity.Player;
import org.checkerframework.checker.nullness.qual.NonNull;
import org.checkerframework.checker.nullness.qual.Nullable;
import org.checkerframework.framework.qual.DefaultQualifier;
import static net.kyori.adventure.text.Component.text;
import static net.kyori.adventure.text.format.NamedTextColor.GREEN;
import static net.kyori.adventure.text.format.NamedTextColor.RED;
@DefaultQualifier(NonNull.class)
public final class EntityCommand implements PaperSubcommand {
@Override
public boolean execute(final CommandSender sender, final String subCommand, final String[] args) {
this.listEntities(sender, args);
return true;
}
@Override
public List<String> tabComplete(final CommandSender sender, final String subCommand, final String[] args) {
if (args.length == 1) {
return CommandUtil.getListMatchingLast(sender, args, "help", "list");
} else if (args.length == 2) {
return CommandUtil.getListMatchingLast(sender, args, BuiltInRegistries.ENTITY_TYPE.keySet().stream().map(ResourceLocation::toString).sorted().toArray(String[]::new));
}
return Collections.emptyList();
}
/*
* Ported from MinecraftForge - author: LexManos <LexManos@gmail.com> - License: LGPLv2.1
*/
private void listEntities(final CommandSender sender, final String[] args) {
// help
if (args.length < 1 || !args[0].toLowerCase(Locale.ROOT).equals("list")) {
sender.sendMessage(text("Use /paper entity [list] help for more information on a specific command", RED));
return;
}
if ("list".equals(args[0].toLowerCase(Locale.ROOT))) {
String filter = "*";
if (args.length > 1) {
if (args[1].toLowerCase(Locale.ROOT).equals("help")) {
sender.sendMessage(text("Use /paper entity list [filter] [worldName] to get entity info that matches the optional filter.", RED));
return;
}
filter = args[1];
}
final String cleanfilter = filter.replace("?", ".?").replace("*", ".*?");
Set<ResourceLocation> names = BuiltInRegistries.ENTITY_TYPE.keySet().stream()
.filter(n -> n.toString().matches(cleanfilter))
.collect(Collectors.toSet());
if (names.isEmpty()) {
sender.sendMessage(text("Invalid filter, does not match any entities. Use /paper entity list for a proper list", RED));
sender.sendMessage(text("Usage: /paper entity list [filter] [worldName]", RED));
return;
}
String worldName;
if (args.length > 2) {
worldName = args[2];
} else if (sender instanceof Player) {
worldName = ((Player) sender).getWorld().getName();
} else {
sender.sendMessage(text("Please specify the name of a world", RED));
sender.sendMessage(text("To do so without a filter, specify '*' as the filter", RED));
sender.sendMessage(text("Usage: /paper entity list [filter] [worldName]", RED));
return;
}
Map<ResourceLocation, MutablePair<Integer, Map<ChunkPos, Integer>>> list = Maps.newHashMap();
@Nullable World bukkitWorld = Bukkit.getWorld(worldName);
if (bukkitWorld == null) {
sender.sendMessage(text("Could not load world for " + worldName + ". Please select a valid world.", RED));
sender.sendMessage(text("Usage: /paper entity list [filter] [worldName]", RED));
return;
}
ServerLevel world = ((CraftWorld) bukkitWorld).getHandle();
Map<ResourceLocation, Integer> nonEntityTicking = Maps.newHashMap();
ServerChunkCache chunkProviderServer = world.getChunkSource();
world.getAllEntities().forEach(e -> {
ResourceLocation key = EntityType.getKey(e.getType());
MutablePair<Integer, Map<ChunkPos, Integer>> info = list.computeIfAbsent(key, k -> MutablePair.of(0, Maps.newHashMap()));
ChunkPos chunk = e.chunkPosition();
info.left++;
info.right.put(chunk, info.right.getOrDefault(chunk, 0) + 1);
if (!world.isPositionEntityTicking(e.blockPosition())) {
nonEntityTicking.merge(key, 1, Integer::sum);
}
});
if (names.size() == 1) {
ResourceLocation name = names.iterator().next();
Pair<Integer, Map<ChunkPos, Integer>> info = list.get(name);
int nonTicking = nonEntityTicking.getOrDefault(name, 0);
if (info == null) {
sender.sendMessage(text("No entities found.", RED));
return;
}
sender.sendMessage("Entity: " + name + " Total Ticking: " + (info.getLeft() - nonTicking) + ", Total Non-Ticking: " + nonTicking);
info.getRight().entrySet().stream()
.sorted((a, b) -> !a.getValue().equals(b.getValue()) ? b.getValue() - a.getValue() : a.getKey().toString().compareTo(b.getKey().toString()))
.limit(10).forEach(e -> {
final int x = (e.getKey().x << 4) + 8;
final int z = (e.getKey().z << 4) + 8;
final Component message = text(" " + e.getValue() + ": " + e.getKey().x + ", " + e.getKey().z + (chunkProviderServer.isPositionTicking(e.getKey().toLong()) ? " (Ticking)" : " (Non-Ticking)"))
.hoverEvent(HoverEvent.showText(text("Click to teleport to chunk", GREEN)))
.clickEvent(ClickEvent.clickEvent(ClickEvent.Action.RUN_COMMAND, "/minecraft:execute as @s in " + world.getWorld().getKey() + " run tp " + x + " " + (world.getWorld().getHighestBlockYAt(x, z, HeightMap.MOTION_BLOCKING) + 1) + " " + z));
sender.sendMessage(message);
});
} else {
List<Pair<ResourceLocation, Integer>> info = list.entrySet().stream()
.filter(e -> names.contains(e.getKey()))
.map(e -> Pair.of(e.getKey(), e.getValue().left))
.sorted((a, b) -> !a.getRight().equals(b.getRight()) ? b.getRight() - a.getRight() : a.getKey().toString().compareTo(b.getKey().toString()))
.toList();
if (info.isEmpty()) {
sender.sendMessage(text("No entities found.", RED));
return;
}
int count = info.stream().mapToInt(Pair::getRight).sum();
int nonTickingCount = nonEntityTicking.values().stream().mapToInt(Integer::intValue).sum();
sender.sendMessage("Total Ticking: " + (count - nonTickingCount) + ", Total Non-Ticking: " + nonTickingCount);
info.forEach(e -> {
int nonTicking = nonEntityTicking.getOrDefault(e.getKey(), 0);
sender.sendMessage(" " + (e.getValue() - nonTicking) + " (" + nonTicking + ") " + ": " + e.getKey());
});
sender.sendMessage("* First number is ticking entities, second number is non-ticking entities");
}
}
}
}

View File

@@ -0,0 +1,38 @@
package io.papermc.paper.command.subcommands;
import io.papermc.paper.command.PaperSubcommand;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import org.bukkit.command.Command;
import org.bukkit.command.CommandSender;
import org.bukkit.craftbukkit.CraftServer;
import org.checkerframework.checker.nullness.qual.NonNull;
import org.checkerframework.framework.qual.DefaultQualifier;
import static net.kyori.adventure.text.Component.text;
import static net.kyori.adventure.text.format.NamedTextColor.GREEN;
import static net.kyori.adventure.text.format.NamedTextColor.RED;
import static net.kyori.adventure.text.format.NamedTextColor.YELLOW;
@DefaultQualifier(NonNull.class)
public final class HeapDumpCommand implements PaperSubcommand {
@Override
public boolean execute(final CommandSender sender, final String subCommand, final String[] args) {
this.dumpHeap(sender);
return true;
}
private void dumpHeap(final CommandSender sender) {
java.nio.file.Path dir = java.nio.file.Paths.get("./dumps");
String name = "heap-dump-" + DateTimeFormatter.ofPattern("yyyy-MM-dd_HH.mm.ss").format(LocalDateTime.now());
Command.broadcastCommandMessage(sender, text("Writing JVM heap data...", YELLOW));
java.nio.file.Path file = CraftServer.dumpHeap(dir, name);
if (file != null) {
Command.broadcastCommandMessage(sender, text("Heap dump saved to " + file, GREEN));
} else {
Command.broadcastCommandMessage(sender, text("Failed to write heap dump, see server log for details", RED));
}
}
}

View File

@@ -0,0 +1,33 @@
package io.papermc.paper.command.subcommands;
import io.papermc.paper.command.PaperSubcommand;
import net.minecraft.server.MinecraftServer;
import org.bukkit.command.Command;
import org.bukkit.command.CommandSender;
import org.bukkit.craftbukkit.CraftServer;
import org.checkerframework.checker.nullness.qual.NonNull;
import org.checkerframework.framework.qual.DefaultQualifier;
import static net.kyori.adventure.text.Component.text;
import static net.kyori.adventure.text.format.NamedTextColor.GREEN;
import static net.kyori.adventure.text.format.NamedTextColor.RED;
@DefaultQualifier(NonNull.class)
public final class ReloadCommand implements PaperSubcommand {
@Override
public boolean execute(final CommandSender sender, final String subCommand, final String[] args) {
this.doReload(sender);
return true;
}
private void doReload(final CommandSender sender) {
Command.broadcastCommandMessage(sender, text("Please note that this command is not supported and may cause issues.", RED));
Command.broadcastCommandMessage(sender, text("If you encounter any issues please use the /stop command to restart your server.", RED));
MinecraftServer server = ((CraftServer) sender.getServer()).getServer();
server.paperConfigurations.reloadConfigs(server);
server.server.reloadCount++;
Command.broadcastCommandMessage(sender, text("Paper config reload complete.", GREEN));
}
}

View File

@@ -0,0 +1,21 @@
package io.papermc.paper.command.subcommands;
import io.papermc.paper.command.PaperSubcommand;
import net.minecraft.server.MinecraftServer;
import org.bukkit.command.Command;
import org.bukkit.command.CommandSender;
import org.checkerframework.checker.nullness.qual.NonNull;
import org.checkerframework.checker.nullness.qual.Nullable;
import org.checkerframework.framework.qual.DefaultQualifier;
@DefaultQualifier(NonNull.class)
public final class VersionCommand implements PaperSubcommand {
@Override
public boolean execute(final CommandSender sender, final String subCommand, final String[] args) {
final @Nullable Command ver = MinecraftServer.getServer().server.getCommandMap().getCommand("version");
if (ver != null) {
ver.execute(sender, "paper", new String[0]);
}
return true;
}
}