package com.serveroptimizer.tasks import com.serveroptimizer.ServerOptimizerPlugin import org.bukkit.entity.* import org.bukkit.scheduler.BukkitTask class EntityCleanupTask(private val plugin: ServerOptimizerPlugin) { private var task: BukkitTask? = null fun start() { task = plugin.server.scheduler.runTaskTimer(plugin, Runnable { if (!plugin.configManager.isEntityCleanupEnabled()) return@Runnable cleanupAllWorlds() }, 200L, plugin.configManager.getCleanupInterval()) } fun stop() { task?.cancel() task = null } fun cleanupWorld(world: org.bukkit.World): Int { if (!plugin.worldSettingsManager.isEntityCleanupEnabledForWorld(world)) return 0 val wName = world.name var removed = 0 val maxMobs = plugin.configManager.getMaxMobsPerChunk(wName) val maxAnimals = plugin.configManager.getMaxAnimalsPerChunk(wName) val maxItems = plugin.configManager.getMaxItemsOnGround(wName) val maxItemAge = plugin.configManager.getMaxItemAgeTicks(wName) val maxProjectiles = plugin.configManager.getMaxProjectiles(wName) val removeOrbs = plugin.configManager.isRemoveEmptyOrbs(wName) val arrowTicks = plugin.configManager.getRemoveArrowsTicks(wName) val chunkMobs = mutableMapOf>() val chunkAnimals = mutableMapOf>() val groundItems = mutableListOf() val projectiles = mutableListOf() val orbs = mutableListOf() val arrows = mutableListOf() for (entity in world.entities) { when (entity) { is Animals -> { val key = chunkKey(entity.location.blockX shr 4, entity.location.blockZ shr 4) chunkAnimals.getOrPut(key) { mutableListOf() }.add(entity) } is Mob -> { val key = chunkKey(entity.location.blockX shr 4, entity.location.blockZ shr 4) chunkMobs.getOrPut(key) { mutableListOf() }.add(entity) } is Item -> groundItems.add(entity) is AbstractArrow -> arrows.add(entity) is Projectile -> projectiles.add(entity) is ExperienceOrb -> if (removeOrbs && entity.experience == 0) orbs.add(entity) } } for ((_, mobs) in chunkMobs) { if (mobs.size > maxMobs) { val toRemove = mobs.sortedBy { it.health }.take(mobs.size - maxMobs) toRemove.forEach { if (!it.isDead) { it.remove(); removed++ } } } } for ((_, animals) in chunkAnimals) { if (animals.size > maxAnimals) { val toRemove = animals.take(animals.size - maxAnimals) toRemove.forEach { if (!it.isDead) { it.remove(); removed++ } } } } val oldItems = groundItems.filter { it.ticksLived >= maxItemAge } oldItems.forEach { it.remove(); removed++ } if (groundItems.size > maxItems) { val sorted = groundItems.sortedByDescending { it.ticksLived } val excess = sorted.take(groundItems.size - maxItems) excess.forEach { if (!it.isDead) { it.remove(); removed++ } } } if (projectiles.size > maxProjectiles) { val excess = projectiles.take(projectiles.size - maxProjectiles) excess.forEach { if (!it.isDead) { it.remove(); removed++ } } } arrows.filter { it.ticksLived >= arrowTicks }.forEach { it.remove(); removed++ } orbs.forEach { it.remove(); removed++ } if (plugin.configManager.isMergeNearbyItems()) { mergeNearbyItems(world, plugin.configManager.getItemMergeRadius()) } return removed } private fun mergeNearbyItems(world: org.bukkit.World, radius: Double) { val items = world.entities.filterIsInstance().filter { !it.isDead } val processed = mutableSetOf() for (item in items) { if (item.entityId in processed || item.isDead) continue val nearby = item.getNearbyEntities(radius, radius, radius) .filterIsInstance() .filter { it.entityId != item.entityId && it.entityId !in processed && !it.isDead } .filter { it.itemStack.type == item.itemStack.type && it.itemStack.itemMeta == item.itemStack.itemMeta } for (other in nearby) { val newAmount = item.itemStack.amount + other.itemStack.amount val maxStack = item.itemStack.maxStackSize if (newAmount <= maxStack) { item.itemStack = item.itemStack.clone().also { it.amount = newAmount } other.remove() processed.add(other.entityId) } } processed.add(item.entityId) } } private fun cleanupAllWorlds() { for (world in plugin.server.worlds) { val count = cleanupWorld(world) plugin.alertManager.sendCleanupResult(world.name, count) } } private fun chunkKey(cx: Int, cz: Int): Long = cx.toLong() shl 32 or (cz.toLong() and 0xFFFFFFFFL) }