Feature: Horse utils

Author: emyfops

common/src/main/kotlin/com/lambda/module/modules/movement/HorseUtils.kt

@@ -0,0 +1,49 @@
+package com.lambda.module.modules.movement
+
+import com.lambda.event.events.PacketEvent
+import com.lambda.event.events.TickEvent
+import com.lambda.event.listener.SafeListener.Companion.listener
+import com.lambda.module.Module
+import com.lambda.module.tag.ModuleTag
+import com.lambda.util.world.EntityUtils.getEntities
+import net.minecraft.entity.passive.AbstractHorseEntity
+import net.minecraft.network.packet.c2s.play.PlayerInteractEntityC2SPacket
+
+object HorseUtils : Module(
+    name = "HorseUtils",
+    description = "Various utilities for horses.",
+    defaultTags = setOf(ModuleTag.MOVEMENT, ModuleTag.BYPASS)
+) {
+    private val page by setting("Page", Page.General)
+
+    /* General */
+    private val forceMount by setting("Force Mount", true, description = "Attempts to force mount chested entities.", visibility = { page == Page.General })
+    private val tameHorses by setting("Tame Horses", true, description = "Automatically tames horses (client-side only).", visibility = { page == Page.General })
+
+    /* Rendering */
+    private val showInfo by setting("Show Info", true, description = "Renders information about entities.", visibility = { page == Page.Rendering })
+
+    private val theHonses = mutableListOf<AbstractHorseEntity>() // Petah, the honse is here
+    private val tame: (AbstractHorseEntity) -> Unit = { horse -> if (tameHorses) horse.setHorseFlag(4, true) }
+
+    private enum class Page {
+        General, Rendering
+    }
+
+    init {
+        listener<TickEvent.Pre> {
+            getEntities(theHonses, iterator = tame)
+        }
+
+        listener<PacketEvent.Send.Pre> { event ->
+            if (!forceMount) return@listener
+            if (event.packet !is PlayerInteractEntityC2SPacket) return@listener
+            if (event.packet.type !is PlayerInteractEntityC2SPacket.InteractAtHandler) return@listener
+
+            val entity = world.getEntityById(event.packet.entityId) ?: return@listener
+            if (entity !is AbstractHorseEntity) return@listener
+
+            event.cancel()
+        }
+    }
+}

common/src/main/kotlin/com/lambda/util/world/EntityUtils.kt

@@ -7,6 +7,7 @@ import net.minecraft.util.math.ChunkSectionPos
 import net.minecraft.util.math.Vec3d
 import kotlin.math.ceil
 
+val nullptr = null
 
 object EntityUtils {
     /**
@@ -34,9 +35,8 @@ object EntityUtils {
         }
 
         // Speculative execution trolling
-        val entities = ArrayList<T>()
-            if (range > 64) getEntities(entities, predicate, iterator)
-            else getFastEntities(pos, range, entities, predicate, iterator)
+        if (range > 64) getEntities(nullptr, predicate, iterator)
+        else getFastEntities(pos, range, nullptr, predicate, iterator)
 
         return closest
     }
@@ -65,14 +65,16 @@ object EntityUtils {
      *
      * @param pos The position to search from.
      * @param distance The maximum distance to search for entities.
+     * @param pointer The mutable list to store the entities in.
      * @param predicate Optional predicate to filter entities. It allows custom filtering based on entity properties.
+     * @param iterator Optional iterator to perform operations on each entity.
      * @return A list of entities of type [T] within the specified distance from the position, excluding the player.
      *
      */
     inline fun <reified T : Entity> SafeContext.getFastEntities(
         pos: Vec3d,
         distance: Double,
-        pointer: MutableList<T>? = null,
+        pointer: MutableList<T>? = nullptr,
         noinline predicate: (T) -> Boolean = { true },
         noinline iterator: (T) -> Unit = { },
     ) {
@@ -103,11 +105,12 @@ object EntityUtils {
      * This function retrieves entities of type [T] within a specified distance from a given position. Unlike
      * [getFastEntities], it traverses all entities in the world to find matches, while also excluding the player entity.
      *
-     * @param predicate Optional predicate to filter entities.
-     * @return A list of entities of type [T] within the specified distance from the position without the player.
+     * @param pointer The mutable list to store the entities in.
+     * @param predicate Optional predicate to filter entities. It allows custom filtering based on entity properties.
+     * @param iterator Optional iterator to perform operations on each entity.
      */
     inline fun <reified T : Entity> SafeContext.getEntities(
-        pointer: MutableList<T>? = null,
+        pointer: MutableList<T>? = nullptr,
         noinline predicate: (T) -> Boolean = { true },
         noinline iterator: (T) -> Unit = { },
     ) {

common/src/main/resources/lambda.accesswidener

@@ -11,6 +11,7 @@ accessible field net/minecraft/client/world/ClientWorld entityManager Lnet/minec
 # Entity
 accessible field net/minecraft/entity/projectile/FireworkRocketEntity shooter Lnet/minecraft/entity/LivingEntity;
 accessible method net/minecraft/entity/Entity movementInputToVelocity (Lnet/minecraft/util/math/Vec3d;FF)Lnet/minecraft/util/math/Vec3d;
+accessible method net/minecraft/entity/passive/AbstractHorseEntity setHorseFlag (IZ)V
 
 # Camera
 accessible method net/minecraft/client/render/Camera setPos (DDD)V
@@ -34,6 +35,12 @@ accessible field net/minecraft/text/Style insertion Ljava/lang/String;
 accessible field net/minecraft/text/Style font Lnet/minecraft/util/Identifier;
 accessible method net/minecraft/text/Style <init> (Lnet/minecraft/text/TextColor;Ljava/lang/Boolean;Ljava/lang/Boolean;Ljava/lang/Boolean;Ljava/lang/Boolean;Ljava/lang/Boolean;Lnet/minecraft/text/ClickEvent;Lnet/minecraft/text/HoverEvent;Ljava/lang/String;Lnet/minecraft/util/Identifier;)V
 
+# Packet
+accessible field net/minecraft/network/packet/c2s/play/PlayerInteractEntityC2SPacket entityId I
+accessible field net/minecraft/network/packet/c2s/play/PlayerInteractEntityC2SPacket type Lnet/minecraft/network/packet/c2s/play/PlayerInteractEntityC2SPacket$InteractTypeHandler;
+accessible class net/minecraft/network/packet/c2s/play/PlayerInteractEntityC2SPacket$InteractTypeHandler
+accessible class net/minecraft/network/packet/c2s/play/PlayerInteractEntityC2SPacket$InteractAtHandler
+
 # Other
 accessible field net/minecraft/client/world/ClientEntityManager cache Lnet/minecraft/world/entity/SectionedEntityCache;
 accessible field net/minecraft/world/entity/EntityTrackingSection collection Lnet/minecraft/util/collection/TypeFilterableList;