mirror of https://github.com/plexusorg/Plex.git
100 lines
3.7 KiB
Java
100 lines
3.7 KiB
Java
package dev.plex.command.impl;
|
|
|
|
import dev.plex.command.PlexCommand;
|
|
import dev.plex.command.annotation.CommandParameters;
|
|
import dev.plex.command.annotation.CommandPermissions;
|
|
import dev.plex.command.source.RequiredCommandSource;
|
|
import dev.plex.rank.enums.Rank;
|
|
import dev.plex.util.PlexUtils;
|
|
import net.kyori.adventure.text.Component;
|
|
import org.bukkit.Bukkit;
|
|
import org.bukkit.World;
|
|
import org.bukkit.command.CommandSender;
|
|
import org.bukkit.entity.Entity;
|
|
import org.bukkit.entity.EntityType;
|
|
import org.bukkit.entity.Player;
|
|
import org.jetbrains.annotations.NotNull;
|
|
import org.jetbrains.annotations.Nullable;
|
|
|
|
import java.util.*;
|
|
|
|
@CommandPermissions(level = Rank.ADMIN, permission = "plex.entitywipe", source = RequiredCommandSource.ANY)
|
|
@CommandParameters(name = "entitywipe", description = "Remove various server entities that may cause lag, such as dropped items, minecarts, and boats.", usage = "/<command> [name]", aliases = "ew,rd")
|
|
public class EntityWipeCMD extends PlexCommand
|
|
{
|
|
@Override
|
|
protected Component execute(@NotNull CommandSender sender, @Nullable Player playerSender, @NotNull String[] args)
|
|
{
|
|
List<String> entityBlacklist = plugin.config.getStringList("entitywipe_list");
|
|
|
|
List<String> entityWhitelist = new LinkedList<>(Arrays.asList(args));
|
|
|
|
EntityType[] entityTypes = EntityType.values();
|
|
entityWhitelist.removeIf(name ->
|
|
{
|
|
boolean res = Arrays.stream(entityTypes).noneMatch(entityType -> name.equalsIgnoreCase(entityType.name()));
|
|
if (res)
|
|
{
|
|
sender.sendMessage(messageComponent("invalidEntityType", name));
|
|
}
|
|
return res;
|
|
});
|
|
|
|
boolean useBlacklist = args.length == 0;
|
|
|
|
HashMap<String, Integer> entityCounts = new HashMap<>();
|
|
|
|
for (World world : Bukkit.getWorlds())
|
|
{
|
|
for (Entity entity : world.getEntities())
|
|
{
|
|
if (entity.getType() != EntityType.PLAYER)
|
|
{
|
|
String type = entity.getType().name();
|
|
|
|
if (useBlacklist ? entityBlacklist.stream().noneMatch(entityName -> entityName.equalsIgnoreCase(type)) : entityWhitelist.stream().anyMatch(entityName -> entityName.equalsIgnoreCase(type)))
|
|
{
|
|
entity.remove();
|
|
|
|
entityCounts.put(type, entityCounts.getOrDefault(type, 0) + 1);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
int entityCount = entityCounts.values().stream().mapToInt(a -> a).sum();
|
|
|
|
if (useBlacklist)
|
|
{
|
|
PlexUtils.broadcast(messageComponent("removedEntities", sender.getName(), entityCount));
|
|
}
|
|
else
|
|
{
|
|
if (entityCount == 0)
|
|
{
|
|
sender.sendMessage(messageComponent("noRemovedEntities"));
|
|
return null;
|
|
}
|
|
String list = String.join(", ", entityCounts.keySet());
|
|
list = list.replaceAll("(, )(?!.*\1)", (list.indexOf(", ") == list.lastIndexOf(", ") ? "" : ",") + " and ");
|
|
PlexUtils.broadcast(messageComponent("removedEntitiesOfTypes", sender.getName(), entityCount, list));
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public @NotNull List<String> tabComplete(@NotNull CommandSender sender, @NotNull String alias, @NotNull String[] args) throws IllegalArgumentException
|
|
{
|
|
List<String> entities = new ArrayList<>();
|
|
for (World world : Bukkit.getWorlds())
|
|
{
|
|
for (Entity entity : world.getEntities())
|
|
{
|
|
if (entity.getType() != EntityType.PLAYER)
|
|
{
|
|
entities.add(entity.getType().name());
|
|
}
|
|
}
|
|
}
|
|
return entities.stream().toList();
|
|
}
|
|
} |