43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
import json
|
|
|
|
from minecraft.networking.packets import clientbound, serverbound
|
|
|
|
class ChatManager:
|
|
def __init__(self, global_state):
|
|
self.g = global_state
|
|
self.handler = None
|
|
|
|
self.g.connection.register_packet_listener(self.print_chat, clientbound.play.ChatMessagePacket)
|
|
|
|
def translate_chat(self, data):
|
|
if isinstance(data, str):
|
|
return data
|
|
elif 'extra' in data:
|
|
return ''.join([self.translate_chat(x) for x in data['extra']])
|
|
elif 'text' in data:
|
|
return data['text']
|
|
else:
|
|
return '?'
|
|
|
|
def print_chat(self, chat_packet):
|
|
# TODO: Replace with handler
|
|
try:
|
|
source = chat_packet.field_string('position')
|
|
text = self.translate_chat(json.loads(chat_packet.json_data))
|
|
print('[%s] %s'%(source, text))
|
|
if self.handler:
|
|
self.handler((source, text))
|
|
except Exception as ex:
|
|
print('Exception %r on message (%s): %s' % (ex, chat_packet.field_string('position'), chat_packet.json_data))
|
|
|
|
def set_handler(self, func):
|
|
self.handler = func
|
|
|
|
def send(self, text):
|
|
if not text:
|
|
# Prevents connection bug when sending empty chat message
|
|
return
|
|
packet = serverbound.play.ChatPacket()
|
|
packet.message = text
|
|
self.g.connection.write_packet(packet)
|