azalea-buf

1 files changed, 16 insertions, 0 deletions

diff --git a/azalea-world/src/lib.rs b/azalea-world/src/lib.rs
index 10beb309..6e922bb9 100644
--- a/azalea-world/src/lib.rs
+++ b/azalea-world/src/lib.rs
@@ -72,6 +72,22 @@ impl World {
         Ok(())
     }
 
+    pub fn move_entity_with_delta(&mut self, entity_id: u32, delta: PositionDelta) -> Result<(), String> {
+        let entity = self
+            .entity_storage
+            .get_mut_by_id(entity_id)
+            .ok_or_else(|| "Moving entity that doesn't exist".to_string())?;
+        let old_chunk = ChunkPos::from(entity.pos());
+        let new_chunk = ChunkPos::from(&new_pos);
+        // this is fine because we update the chunk below
+        entity.unsafe_move(new_pos);
+        if old_chunk != new_chunk {
+            self.entity_storage
+                .update_entity_chunk(entity_id, &old_chunk, &new_chunk);
+        }
+        Ok(())
+    }
+
     pub fn add_entity(&mut self, entity: Entity) {
         self.entity_storage.insert(entity);
     }