2022-05-25 05:35:53 +00:00
|
|
|
from lib.code.packet import fix_state
|
2022-05-25 05:30:47 +00:00
|
|
|
from lib.utils import PacketIdentifier, group_packets
|
2023-05-04 01:57:27 +00:00
|
|
|
import lib.code.inventory
|
2022-12-08 03:09:58 +00:00
|
|
|
import lib.code.language
|
|
|
|
import lib.code.registry
|
2022-05-25 04:05:44 +00:00
|
|
|
import lib.code.version
|
2022-08-28 01:31:21 +00:00
|
|
|
import lib.code.blocks
|
2022-05-25 05:21:05 +00:00
|
|
|
import lib.code.packet
|
2022-12-08 03:09:58 +00:00
|
|
|
import lib.code.shapes
|
2023-03-14 21:33:03 +00:00
|
|
|
import lib.code.entity
|
2022-08-28 01:31:21 +00:00
|
|
|
import lib.code.utils
|
2022-05-25 04:05:44 +00:00
|
|
|
import lib.download
|
2022-05-27 06:37:52 +00:00
|
|
|
import lib.extract
|
2022-05-25 04:05:44 +00:00
|
|
|
import sys
|
2022-06-22 00:05:44 +00:00
|
|
|
|
|
|
|
lib.download.clear_version_cache()
|
2022-05-25 04:05:44 +00:00
|
|
|
|
2022-06-22 00:57:18 +00:00
|
|
|
if len(sys.argv) == 1:
|
|
|
|
print('\033[91mYou must provide a version to migrate to.\033[m')
|
|
|
|
version_manifest = lib.download.get_version_manifest()
|
|
|
|
newest_version = version_manifest['latest']['snapshot']
|
|
|
|
print(f'Hint: newest version is \033[1m{newest_version}\033[m')
|
|
|
|
exit()
|
|
|
|
|
|
|
|
|
2022-05-25 04:05:44 +00:00
|
|
|
old_version_id = lib.code.version.get_version_id()
|
|
|
|
old_mappings = lib.download.get_mappings_for_version(old_version_id)
|
2022-05-27 06:37:52 +00:00
|
|
|
old_burger_data = lib.extract.get_burger_data_for_version(old_version_id)
|
2022-05-25 04:05:44 +00:00
|
|
|
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)
|
2022-05-27 06:37:52 +00:00
|
|
|
new_burger_data = lib.extract.get_burger_data_for_version(new_version_id)
|
2022-05-25 04:05:44 +00:00
|
|
|
new_packet_list = list(new_burger_data[0]['packets']['packet'].values())
|
|
|
|
|
2022-05-25 05:21:05 +00:00
|
|
|
|
|
|
|
old_packets: dict[PacketIdentifier, str] = {}
|
2022-06-17 01:59:19 +00:00
|
|
|
old_packets_data: dict[PacketIdentifier, dict] = {}
|
2022-05-25 05:21:05 +00:00
|
|
|
new_packets: dict[PacketIdentifier, str] = {}
|
2022-06-17 01:59:19 +00:00
|
|
|
new_packets_data: dict[PacketIdentifier, dict] = {}
|
2022-05-25 04:05:44 +00:00
|
|
|
|
|
|
|
for packet in old_packet_list:
|
|
|
|
assert packet['class'].endswith('.class')
|
|
|
|
packet_name = old_mappings.get_class(packet['class'][:-6])
|
2022-06-17 01:59:19 +00:00
|
|
|
packet_ident = PacketIdentifier(
|
|
|
|
packet['id'], packet['direction'].lower(), fix_state(packet['state']))
|
|
|
|
old_packets[packet_ident] = packet_name
|
|
|
|
old_packets_data[packet_ident] = packet
|
2022-05-25 04:05:44 +00:00
|
|
|
for packet in new_packet_list:
|
|
|
|
assert packet['class'].endswith('.class')
|
|
|
|
packet_name = new_mappings.get_class(packet['class'][:-6])
|
2022-06-17 01:59:19 +00:00
|
|
|
packet_ident = PacketIdentifier(
|
|
|
|
packet['id'], packet['direction'].lower(), fix_state(packet['state']))
|
|
|
|
new_packets[packet_ident] = packet_name
|
|
|
|
new_packets_data[packet_ident] = packet
|
2022-05-25 05:21:05 +00:00
|
|
|
|
|
|
|
# find removed packets
|
|
|
|
removed_packets: list[PacketIdentifier] = []
|
2022-05-25 05:38:27 +00:00
|
|
|
for packet, packet_name in old_packets.items():
|
2022-05-26 01:18:03 +00:00
|
|
|
if packet_name not in new_packets.values():
|
2022-05-25 05:21:05 +00:00
|
|
|
removed_packets.append(packet)
|
2022-05-26 01:18:03 +00:00
|
|
|
print('Removed packet:', packet, packet_name)
|
2022-05-26 01:21:50 +00:00
|
|
|
for (direction, state), packets in group_packets(removed_packets).items():
|
|
|
|
lib.code.packet.remove_packet_ids(packets, direction, state)
|
2022-05-25 04:05:44 +00:00
|
|
|
|
|
|
|
print()
|
|
|
|
|
2022-05-25 05:21:05 +00:00
|
|
|
# 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():
|
2022-05-26 01:09:48 +00:00
|
|
|
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:
|
2022-05-25 05:21:05 +00:00
|
|
|
changed_packets[old_packet] = new_packet.packet_id
|
2022-05-25 05:30:47 +00:00
|
|
|
print('Changed packet id:', old_packet, '->',
|
|
|
|
new_packet, f'({new_packet_name})')
|
2022-05-25 05:39:33 +00:00
|
|
|
break
|
2022-05-26 01:21:50 +00:00
|
|
|
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)
|
2022-05-25 05:21:05 +00:00
|
|
|
|
2022-05-25 04:05:44 +00:00
|
|
|
|
|
|
|
print()
|
|
|
|
|
2022-06-17 01:59:19 +00:00
|
|
|
# find added/changed packets
|
|
|
|
added_or_changed_packets: list[PacketIdentifier] = []
|
2022-07-06 23:52:24 +00:00
|
|
|
for new_packet, packet_name in new_packets.items():
|
|
|
|
old_packet = None
|
|
|
|
for old_packet_tmp, old_packet_name in old_packets.items():
|
|
|
|
if old_packet_name == packet_name:
|
|
|
|
old_packet = old_packet_tmp
|
|
|
|
break
|
|
|
|
|
2022-05-25 05:38:27 +00:00
|
|
|
if packet_name not in old_packets.values():
|
2022-07-06 23:52:24 +00:00
|
|
|
added_or_changed_packets.append(new_packet)
|
|
|
|
print('Added packet:', new_packet, packet_name)
|
|
|
|
elif old_packet and not lib.code.packet.are_packet_instructions_identical(new_packets_data[new_packet].get('instructions'), old_packets_data[old_packet].get('instructions')):
|
|
|
|
added_or_changed_packets.append(new_packet)
|
|
|
|
print('Changed packet:', new_packet, packet_name)
|
2022-06-17 01:59:19 +00:00
|
|
|
for packet in added_or_changed_packets:
|
2022-05-26 01:21:50 +00:00
|
|
|
lib.code.packet.generate_packet(
|
|
|
|
new_burger_data[0]['packets']['packet'], new_mappings, packet.packet_id, packet.direction, packet.state)
|
2022-05-26 18:45:48 +00:00
|
|
|
|
|
|
|
lib.code.version.set_protocol_version(
|
|
|
|
new_burger_data[0]['version']['protocol'])
|
|
|
|
|
2022-08-28 01:31:21 +00:00
|
|
|
print('Updated protocol!')
|
|
|
|
|
|
|
|
|
|
|
|
old_ordered_blocks = lib.extract.get_ordered_blocks_burger(old_version_id)
|
|
|
|
new_ordered_blocks = lib.extract.get_ordered_blocks_burger(new_version_id)
|
|
|
|
if old_ordered_blocks != new_ordered_blocks:
|
|
|
|
print('Blocks changed, updating...')
|
2022-12-08 03:09:58 +00:00
|
|
|
|
2022-08-28 01:31:21 +00:00
|
|
|
block_states_burger = lib.extract.get_block_states_burger(new_version_id)
|
|
|
|
block_states_report = lib.extract.get_block_states_report(new_version_id)
|
|
|
|
|
2022-12-08 03:09:58 +00:00
|
|
|
shape_datas = lib.extract.get_pixlyzer_data(
|
|
|
|
new_version_id, 'shapes')
|
|
|
|
pixlyzer_block_datas = lib.extract.get_pixlyzer_data(
|
|
|
|
new_version_id, 'blocks')
|
|
|
|
|
2022-08-28 01:31:21 +00:00
|
|
|
lib.code.blocks.generate_blocks(
|
2022-12-08 03:09:58 +00:00
|
|
|
block_states_burger, block_states_report, new_ordered_blocks, new_mappings)
|
|
|
|
lib.code.shapes.generate_block_shapes(
|
|
|
|
pixlyzer_block_datas, shape_datas['shapes'], shape_datas['aabbs'], block_states_report, block_states_burger, new_mappings)
|
|
|
|
|
|
|
|
print('Getting en_us.json...')
|
|
|
|
language = lib.extract.get_en_us_lang(new_version_id)
|
|
|
|
lib.code.language.write_language(language)
|
|
|
|
|
|
|
|
print('Generating registries...')
|
2023-07-15 03:20:40 +00:00
|
|
|
import genregistries
|
|
|
|
genregistries.generate(new_version_id)
|
2022-12-08 03:09:58 +00:00
|
|
|
|
2023-03-14 21:33:03 +00:00
|
|
|
print('Generating entity metadata...')
|
|
|
|
burger_entities_data = new_burger_data[0]['entities']
|
|
|
|
lib.code.entity.generate_entity_metadata(burger_entities_data, new_mappings)
|
|
|
|
|
2022-12-08 03:09:58 +00:00
|
|
|
print('Finishing touches, setting version in README and formatting code...')
|
|
|
|
lib.code.version.set_version_id(new_version_id)
|
2022-08-28 01:31:21 +00:00
|
|
|
|
|
|
|
|
2022-05-25 04:05:44 +00:00
|
|
|
lib.code.utils.fmt()
|
|
|
|
|
|
|
|
print('Done!')
|
2022-12-08 03:09:58 +00:00
|
|
|
print('Make sure to `cargo check` and look for the generated `TODO`s to make sure everything is correct!')
|