Monkey-patch in custom packets

custom/networking/packets/clientbound/play/chunk_data.py

@@ -9,10 +9,7 @@ from minecraft.networking.types import (
 from ....types import nbt
 
 class ChunkDataPacket(Packet):
-    @staticmethod
-    def get_id(context):
-        return 0x20 # FIXME
-
+    id = 0x20
     packet_name = 'chunk data'
     fields = 'x', 'bit_mask_y', 'z', 'full_chunk'
 

start.py

@@ -5,13 +5,26 @@ import sys
 import re
 from optparse import OptionParser
 
+from custom.managers import DataManager, ChunksManager
+from custom.networking.packets.clientbound.play import chunk_data
+
+import minecraft.networking.packets
+
+def get_packets(old_get_packets):
+    def wrapper(func, context):
+        print('Monkey-patched.')
+        packets = func(context)
+        packets.add(chunk_data.ChunkDataPacket)
+        return packets
+    return lambda x: wrapper(old_get_packets, x)
+
+minecraft.networking.packets.clientbound.play.get_packets = get_packets(minecraft.networking.packets.clientbound.play.get_packets)
+
 from minecraft import authentication
 from minecraft.exceptions import YggdrasilError
 from minecraft.networking.connection import Connection
 from minecraft.networking.packets import Packet, clientbound, serverbound
 
-from custom.managers import DataManager, ChunksManager
-
 
 def get_options():
     parser = OptionParser()
@@ -111,12 +124,18 @@ def main():
         print("Message (%s): %s" % (
             chat_packet.field_string('position'), chat_packet.json_data))
 
-    chunks = ChunksManager(mcdata)
-    chunks.register(connection)
+    #chunks = ChunksManager(mcdata)
+    #chunks.register(connection)
 
     connection.register_packet_listener(
         print_chat, clientbound.play.ChatMessagePacket)
 
+    def handle_chunk(chunk_packet):
+        print(chunk_packet)
+
+    def register(self, connection):
+        connection.register_packet_listener(handle_chunk, chunk_data.ChunkDataPacket)
+
     connection.connect()
 
     while True: