Begin port to 26.3: movement

mosfet/protocol/entities_26_3.py

@@ -0,0 +1,102 @@
+"""Entity and player-list updates for the 26.3 play protocol."""
+from munch import Munch
+
+from minecraft.networking.modern import (
+    AddEntityPacket, EntityPositionSyncPacket, MoveEntityPositionPacket,
+    MoveEntityPositionRotationPacket, MoveEntityRotationPacket,
+    PlayerInfoRemovePacket, PlayerInfoUpdatePacket, RemoveEntitiesPacket,
+)
+from mosfet.info import mobs
+
+
+class ModernEntityManager:
+    def __init__(self, global_state):
+        self.g = global_state
+        self.g.mobs = self.g.get('mobs') or {}
+        self.g.players = self.g.get('players') or {}
+        self.g.objects = self.g.get('objects') or {}
+        self.g.player_names = self.g.get('player_names') or {}
+        register = self.g.connection.register_packet_listener
+        for packet_type, handler in (
+            (AddEntityPacket, self.handle_add_entity),
+            (RemoveEntitiesPacket, self.handle_remove_entities),
+            (MoveEntityPositionPacket, self.handle_move),
+            (MoveEntityPositionRotationPacket, self.handle_move),
+            (MoveEntityRotationPacket, self.handle_rotation),
+            (EntityPositionSyncPacket, self.handle_position_sync),
+            (PlayerInfoUpdatePacket, self.handle_player_info),
+            (PlayerInfoRemovePacket, self.handle_player_info_remove),
+        ):
+            register(handler, packet_type)
+
+    def handle_add_entity(self, packet):
+        name = mobs.MOB_NAMES.get(packet.entity_type_id, 'unknown')
+        common = dict(entity_id=packet.entity_id, x=packet.x,
+                      y=packet.y, z=packet.z)
+        if name == 'player':
+            player_uuid = str(packet.entity_uuid)
+            self.g.players[packet.entity_id] = Munch(
+                **common, player_uuid=player_uuid,
+                yaw=packet.yaw, pitch=packet.pitch)
+        elif name == 'item':
+            self.g.objects[packet.entity_id] = Munch(
+                **common, entity_uuid=str(packet.entity_uuid),
+                velocity_x=packet.velocity_x * 8000,
+                velocity_y=packet.velocity_y * 8000,
+                velocity_z=packet.velocity_z * 8000)
+        else:
+            self.g.mobs[packet.entity_id] = Munch(
+                **common, entity_uuid=str(packet.entity_uuid),
+                type=packet.entity_type_id, yaw=packet.yaw,
+                pitch=packet.pitch)
+
+    def _entity(self, entity_id):
+        for collection in (self.g.players, self.g.mobs, self.g.objects):
+            entity = collection.get(entity_id)
+            if entity is not None:
+                return entity
+        return None
+
+    def handle_move(self, packet):
+        entity = self._entity(packet.entity_id)
+        if entity is None:
+            return
+        entity.x += packet.delta_x
+        entity.y += packet.delta_y
+        entity.z += packet.delta_z
+        if hasattr(packet, 'yaw'):
+            entity.yaw = packet.yaw
+            entity.pitch = packet.pitch
+
+    def handle_rotation(self, packet):
+        entity = self._entity(packet.entity_id)
+        if entity is not None:
+            entity.yaw = packet.yaw
+            entity.pitch = packet.pitch
+
+    def handle_position_sync(self, packet):
+        entity = self._entity(packet.entity_id)
+        if entity is not None:
+            entity.x, entity.y, entity.z = packet.x, packet.y, packet.z
+            entity.yaw = packet.yaw
+            entity.pitch = packet.pitch
+
+    def handle_remove_entities(self, packet):
+        for entity_id in packet.entity_ids:
+            self.g.players.pop(entity_id, None)
+            self.g.mobs.pop(entity_id, None)
+            self.g.objects.pop(entity_id, None)
+
+    def handle_player_info(self, packet):
+        for profile_uuid, name in packet.entries:
+            if name:
+                uuid_text = str(profile_uuid)
+                self.g.player_names[uuid_text] = name
+                self.g.player_names[name] = profile_uuid
+                self.g.player_names[name.lower()] = profile_uuid
+
+    def handle_player_info_remove(self, packet):
+        removed = {str(uuid) for uuid in packet.uuids}
+        for key, value in list(self.g.player_names.items()):
+            if str(value) in removed or str(key) in removed:
+                del self.g.player_names[key]