From 1d8d473e260f3243d85276237bfb24c2f805ca3d Mon Sep 17 00:00:00 2001
From: Tanner Collin <git@tannercollin.com>
Date: Tue, 23 Feb 2021 21:37:14 +0000
Subject: [PATCH] Display when the bot is loading chunks

---
 bot.py               | 14 +++++++++-----
 protocol/managers.py |  3 +++
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/bot.py b/bot.py
index 22abac4..c26cf33 100644
--- a/bot.py
+++ b/bot.py
@@ -55,10 +55,18 @@ def tick(global_state):
     target = None
 
     # make sure current chunks are loaded for physics
-    if not g.chunks.check_loaded(p, 9):
+    if not g.chunks.check_loaded(p, 288):
+        if not g.chunks.loading:
+            print('Loading chunks', end='', flush=True)
+            g.chunks.loading = True
         packet = serverbound.play.PositionAndLookPacket(x=p.x, feet_y=p.y, z=p.z, pitch=0, yaw=0, on_ground=True)
         g.connection.write_packet(packet, force=True)
         return
+    else:
+        if g.chunks.loading:
+            print()
+            print('Chunks loaded.')
+        g.chunks.loading = False
 
     g.chunks.unload_chunks(p)
 
@@ -246,10 +254,6 @@ def bot(global_state):
             time.sleep(utils.TICK)
         print('Player loaded.')
 
-        while not g.chunks.check_loaded(g.pos, 288):
-            time.sleep(utils.TICK)
-        print('Chunks loaded.')
-
         init(g)
         g.game.close_window()
         print('Initialized.')
diff --git a/protocol/managers.py b/protocol/managers.py
index 8e3d21e..24d487f 100644
--- a/protocol/managers.py
+++ b/protocol/managers.py
@@ -44,6 +44,7 @@ class ChunksManager:
         self.chunks = {}
         self.biomes = {}
         self.index = {}
+        self.loading = False
 
     def handle_block(self, block_packet):
         self.set_block_at(block_packet.location.x, block_packet.location.y, block_packet.location.z, block_packet.block_state_id)
@@ -74,6 +75,8 @@ class ChunksManager:
                         self.index[item_id].append(coords)
 
         self.biomes[(chunk_packet.x, None, chunk_packet.z)] = chunk_packet.biomes # FIXME
+        if self.loading:
+            print('.', end='', flush=True)
 
     def register(self, connection):
         connection.register_packet_listener(self.handle_block, clientbound.play.BlockChangePacket)