From 9808f62fe4c004fd865970b4a1cfdf8f2cbadfe7 Mon Sep 17 00:00:00 2001
From: Brady <zeromemesdev@gmail.com>
Date: Sun, 18 Jun 2023 21:25:00 -0500
Subject: [PATCH] Send chunks to nether-pathfinder on `BlockUpdateEvent`

---
 .../java/baritone/behavior/ElytraBehavior.java    | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/src/main/java/baritone/behavior/ElytraBehavior.java b/src/main/java/baritone/behavior/ElytraBehavior.java
index 97dc24102..6260b42bf 100644
--- a/src/main/java/baritone/behavior/ElytraBehavior.java
+++ b/src/main/java/baritone/behavior/ElytraBehavior.java
@@ -18,9 +18,9 @@
 package baritone.behavior;
 
 import baritone.Baritone;
+import baritone.api.event.events.BlockChangeEvent;
 import baritone.api.event.events.ChunkEvent;
 import baritone.api.event.events.TickEvent;
-import baritone.api.event.events.type.EventState;
 import baritone.api.utils.*;
 import baritone.behavior.elytra.NetherPathfinderContext;
 import baritone.behavior.elytra.UnpackedSegment;
@@ -284,12 +284,19 @@ public final class ElytraBehavior extends Behavior implements Helper {
 
     @Override
     public void onChunkEvent(ChunkEvent event) {
-        if (event.getState() == EventState.POST && event.getType().isPopulate()) {
+        if (event.isPostPopulate()) {
             final Chunk chunk = ctx.world().getChunk(event.getX(), event.getZ());
             this.context.queueForPacking(chunk);
         }
     }
 
+    @Override
+    public void onBlockChange(BlockChangeEvent event) {
+        event.getAffectedChunks().stream()
+                .map(pos -> ctx.world().getChunk(pos.x, pos.z))
+                .forEach(this.context::queueForPacking);
+    }
+
     public void path(BlockPos destination) {
         this.pathManager.pathToDestination(destination);
     }
@@ -382,10 +389,6 @@ public final class ElytraBehavior extends Behavior implements Helper {
                             continue;
                         }
                         forceUseFirework = pitch.second();
-                        logDirect("final dy " + dy);
-                        logDirect("i " + i);
-                        logDirect("playerNear " + playerNear);
-                        logDirect("relaxation " + relaxation);
                         goingTo = path.get(i);
                         this.aimPos = path.get(i).add(0, dy, 0);
                         baritone.getLookBehavior().updateTarget(new Rotation(yaw, pitch.first()), false);