mirror of
https://github.com/mat-1/azalea.git
synced 2024-09-20 07:02:31 +00:00
92 lines
3.7 KiB
Python
92 lines
3.7 KiB
Python
from lib.code.packet import fix_state
|
|
from lib.utils import PacketIdentifier, group_packets
|
|
import lib.code.utils
|
|
import lib.code.version
|
|
import lib.code.packet
|
|
import lib.download
|
|
import lib.extract
|
|
import sys
|
|
|
|
lib.download.clear_version_cache()
|
|
|
|
old_version_id = lib.code.version.get_version_id()
|
|
old_mappings = lib.download.get_mappings_for_version(old_version_id)
|
|
old_burger_data = lib.extract.get_burger_data_for_version(old_version_id)
|
|
old_packet_list = list(old_burger_data[0]['packets']['packet'].values())
|
|
|
|
new_version_id = sys.argv[1]
|
|
new_mappings = lib.download.get_mappings_for_version(new_version_id)
|
|
new_burger_data = lib.extract.get_burger_data_for_version(new_version_id)
|
|
new_packet_list = list(new_burger_data[0]['packets']['packet'].values())
|
|
|
|
|
|
old_packets: dict[PacketIdentifier, str] = {}
|
|
old_packets_data: dict[PacketIdentifier, dict] = {}
|
|
new_packets: dict[PacketIdentifier, str] = {}
|
|
new_packets_data: dict[PacketIdentifier, dict] = {}
|
|
|
|
for packet in old_packet_list:
|
|
assert packet['class'].endswith('.class')
|
|
packet_name = old_mappings.get_class(packet['class'][:-6])
|
|
packet_ident = PacketIdentifier(
|
|
packet['id'], packet['direction'].lower(), fix_state(packet['state']))
|
|
old_packets[packet_ident] = packet_name
|
|
old_packets_data[packet_ident] = packet
|
|
for packet in new_packet_list:
|
|
assert packet['class'].endswith('.class')
|
|
packet_name = new_mappings.get_class(packet['class'][:-6])
|
|
packet_ident = PacketIdentifier(
|
|
packet['id'], packet['direction'].lower(), fix_state(packet['state']))
|
|
new_packets[packet_ident] = packet_name
|
|
new_packets_data[packet_ident] = packet
|
|
|
|
# find removed packets
|
|
removed_packets: list[PacketIdentifier] = []
|
|
for packet, packet_name in old_packets.items():
|
|
if packet_name not in new_packets.values():
|
|
removed_packets.append(packet)
|
|
print('Removed packet:', packet, packet_name)
|
|
for (direction, state), packets in group_packets(removed_packets).items():
|
|
lib.code.packet.remove_packet_ids(packets, direction, state)
|
|
|
|
print()
|
|
|
|
# find packets that changed ids
|
|
changed_packets: dict[PacketIdentifier, int] = {}
|
|
for old_packet, old_packet_name in old_packets.items():
|
|
for new_packet, new_packet_name in new_packets.items():
|
|
if old_packet_name == new_packet_name and old_packet.direction == new_packet.direction and old_packet.state == new_packet.state and old_packet.packet_id != new_packet.packet_id:
|
|
changed_packets[old_packet] = new_packet.packet_id
|
|
print('Changed packet id:', old_packet, '->',
|
|
new_packet, f'({new_packet_name})')
|
|
break
|
|
for (direction, state), packets in group_packets(list(changed_packets.keys())).items():
|
|
id_map: dict[int, int] = {}
|
|
for old_packet_id in packets:
|
|
new_packet_id = changed_packets[PacketIdentifier(
|
|
old_packet_id, direction, state)]
|
|
id_map[old_packet_id] = new_packet_id
|
|
lib.code.packet.change_packet_ids(id_map, direction, state)
|
|
|
|
|
|
print()
|
|
|
|
# find added/changed packets
|
|
added_or_changed_packets: list[PacketIdentifier] = []
|
|
for packet, packet_name in new_packets.items():
|
|
if packet_name not in old_packets.values():
|
|
added_or_changed_packets.append(packet)
|
|
print('Added packet:', packet, packet_name)
|
|
if new_packets_data[packet].get('instructions') != old_packets_data[packet].get('instructions'):
|
|
print('hmm')
|
|
for packet in added_or_changed_packets:
|
|
lib.code.packet.generate_packet(
|
|
new_burger_data[0]['packets']['packet'], new_mappings, packet.packet_id, packet.direction, packet.state)
|
|
|
|
lib.code.version.set_protocol_version(
|
|
new_burger_data[0]['version']['protocol'])
|
|
lib.code.version.set_version_id(new_version_id)
|
|
|
|
lib.code.utils.fmt()
|
|
|
|
print('Done!')
|