Add entityAtCursor function

lib/plugins/ray_trace.js

@@ -1,4 +1,5 @@
 const { Vec3 } = require('vec3')
+const { RaycastIterator } = require('prismarine-world').iterators
 
 module.exports = (bot) => {
   function getViewDirection (pitch, yaw) {
@@ -23,4 +24,29 @@ module.exports = (bot) => {
 
     return bot.world.raycast(eyePosition, viewDirection, maxDistance, matcher)
   }
+
+  bot.entityAtCursor = (maxDistance = 3.5) => {
+    const entities = Object.values(bot.entities)
+      .filter(entity => entity.type !== 'object' && entity.username !== bot.username && entity.position.distanceSquared(bot.entity.position) <= maxDistance ** 2)
+      .sort((entity1, entity2) => entity1.position.distanceSquared(bot.entity.position) - entity2.position.distanceSquared(bot.entity.position))
+
+    const dir = new Vec3(-Math.sin(bot.entity.yaw) * Math.cos(bot.entity.pitch), Math.sin(bot.entity.pitch), -Math.cos(bot.entity.yaw) * Math.cos(bot.entity.pitch))
+    const iterator = new RaycastIterator(bot.entity.position.offset(0, bot.entity.height, 0), dir.normalize(), maxDistance)
+
+    let pos = iterator.next()
+    while (pos) {
+      for (let i = 0; i < entities.length; i++) {
+        const entity = entities[i]
+        const w = entity.width / 2
+
+        const shapes = [[-w, 0, -w, w, entity.height + (entity.type === 'player' ? 0.18 : 0), w]]
+
+        if (iterator.intersect(shapes, entity.position)) {
+          return entities[i]
+        }
+      }
+      pos = iterator.next()
+    }
+    return null
+  }
 }