1 files changed, 39 insertions, 0 deletions
diff --git a/src/main/java/at/hannibal2/skyhanni/features/misc/HideFarEntities.kt b/src/main/java/at/hannibal2/skyhanni/features/misc/HideFarEntities.kt
new file mode 100644
index 000000000..cde4acdf4
--- /dev/null
+++ b/src/main/java/at/hannibal2/skyhanni/features/misc/HideFarEntities.kt
@@ -0,0 +1,39 @@
+package at.hannibal2.skyhanni.features.misc
+
+import at.hannibal2.skyhanni.SkyHanniMod
+import at.hannibal2.skyhanni.events.CheckRenderEntityEvent
+import at.hannibal2.skyhanni.events.LorenzTickEvent
+import at.hannibal2.skyhanni.features.garden.GardenAPI
+import at.hannibal2.skyhanni.utils.EntityUtils
+import at.hannibal2.skyhanni.utils.LocationUtils.distanceToPlayer
+import at.hannibal2.skyhanni.utils.LorenzUtils
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
+
+class HideFarEntities {
+    private val config get() = SkyHanniMod.feature.misc.hideFarEntities
+
+    private var ignored = emptySet<Int>()
+
+    @SubscribeEvent
+    fun onTick(event: LorenzTickEvent) {
+        if (!isEnabled()) return
+
+        val maxAmount = config.maxAmount.coerceAtLeast(1)
+        val minDistance = config.minDistance.coerceAtLeast(3)
+
+        ignored = EntityUtils.getAllEntities()
+            .map { it.entityId to it.distanceToPlayer() }
+            .filter { it.second > minDistance }
+            .sortedBy { it.second }.drop(maxAmount)
+            .map { it.first }.toSet()
+    }
+
+    @SubscribeEvent
+    fun onCheckRender(event: CheckRenderEntityEvent<*>) {
+        if (isEnabled() && event.entity.entityId in ignored) {
+            event.cancel()
+        }
+    }
+
+    fun isEnabled() = LorenzUtils.inSkyBlock && config.enabled && !(GardenAPI.inGarden() && config.excludeGarden)
+}