1
2
Fork 0
mirror of https://github.com/mat-1/azalea.git synced 2025-08-02 14:26:04 +00:00
azalea/codegen/migrate.py
mat 40e4096d24
1.21.2 (#171)
* partially implement 24w35a

* start updating to 24w39a + itemcomponent codegen

* fix codegen and broken packets to finish updating to 24w39a :D

* update to 1.21.2 except for blocks

* update ServerboundPlayerInputPacket impl
2024-10-23 00:08:13 -05:00

152 lines
6.1 KiB
Python
Executable file

from lib.code.packet import fix_state
from lib.utils import PacketIdentifier, group_packets
import lib.code.inventory
import lib.code.language
import lib.code.registry
import lib.code.version
import lib.code.blocks
import lib.code.packet
import lib.code.shapes
import lib.code.entity
import lib.code.utils
import lib.download
import lib.extract
import sys
lib.download.clear_version_cache()
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()
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)
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)
old_packet_list = lib.extract.get_packet_list(old_version_id)
new_packet_list = lib.extract.get_packet_list(new_version_id)
# 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 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
# if packet_name not in old_packets.values():
# 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)
# 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'])
# 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...')
# block_states_burger = lib.extract.get_block_states_burger(new_version_id)
# block_states_report = lib.extract.get_block_states_report(new_version_id)
# # TODO: pixlyzer is currently broken so uhhhh
# shape_datas = lib.extract.get_pixlyzer_data(
# '1.20.3-pre4', 'shapes')
# pixlyzer_block_datas = lib.extract.get_pixlyzer_data(
# '1.20.3-pre4', 'blocks')
# lib.code.blocks.generate_blocks(
# block_states_burger, block_states_report, pixlyzer_block_datas, 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...')
import genregistries
genregistries.generate(new_version_id)
# print('Generating entity metadata...')
# burger_entities_data = new_burger_data[0]['entities']
# lib.code.entity.generate_entity_metadata(burger_entities_data, new_mappings)
print('Finishing touches, setting version in README and formatting code...')
lib.code.version.set_version_id(new_version_id)
lib.code.utils.fmt()
print('Done!')
print('Make sure to `cargo check` and look for the generated `TODO`s to make sure everything is correct!')