mat/azalea
1
2
Fork 0
mirror of https://github.com/mat-1/azalea.git synced 2025-08-02 14:26:04 +00:00
Code Issues Projects Releases Packages Activity

Almost

Browse source
This commit is contained in:
mat 2022-05-25 00:35:53 -05:00
parent dc5a9149a5
commit 8953cf43e2
2 changed files with 17 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
codegen/lib/code/packet.py
View file

@ -7,13 +7,17 @@ def make_packet_mod_rs_line(packet_id: int, packet_class_name: str):
return f' {padded_hex(packet_id)}: {to_snake_case(packet_class_name)}::{to_camel_case(packet_class_name)},'
def fix_state(state: str):
return {'PLAY': 'game'}.get(state, state.lower())
def generate_packet(burger_packets, mappings: Mappings, target_packet_id, target_packet_direction, target_packet_state):
for packet in burger_packets.values():
if packet['id'] != target_packet_id:
continue
direction = packet['direction'].lower() # serverbound or clientbound
state = {'PLAY': 'game'}.get(packet['state'], packet['state'].lower())
state = fix_state(packet['state'])
if state != target_packet_state or direction != target_packet_direction:
continue
@ -109,12 +113,12 @@ def generate_packet(burger_packets, mappings: Mappings, target_packet_id, target
f.write('\n'.join(mod_rs))
def set_packets(packet_ids: list, packet_class_names: list, direction: str, state: str):
def set_packets(packet_ids: list[int], packet_class_names: list[str], direction: str, state: str):
assert len(packet_ids) == len(packet_class_names)
# sort the packets by id
packet_ids, packet_class_names = [list(x) for x in zip(
*sorted(zip(packet_ids, packet_class_names), key=lambda pair: pair[0]))]
zipped_packets: list[tuple[int, str]] = [list(x) for x in zip(
*sorted(zip(packet_ids, packet_class_names), key=lambda pair: pair[0]))] # type: ignore
mod_rs_dir = f'../azalea-protocol/src/packets/{state}/mod.rs'
with open(mod_rs_dir, 'r') as f:
@ -127,7 +131,7 @@ def set_packets(packet_ids: list, packet_class_names: list, direction: str, stat
if line.strip() == 'Serverbound => {':
if direction == 'serverbound':
ignore_lines = True
for packet_id, packet_class_name in zip(packet_ids, packet_class_names):
for packet_id, packet_class_name in zipped_packets:
new_mod_rs.append(
make_packet_mod_rs_line(packet_id, packet_class_name)
)
@ -136,7 +140,7 @@ def set_packets(packet_ids: list, packet_class_names: list, direction: str, stat
elif line.strip() == 'Clientbound => {':
if direction == 'serverbound':
ignore_lines = True
for packet_id, packet_class_name in zip(packet_ids, packet_class_names):
for packet_id, packet_class_name in zipped_packets:
new_mod_rs.append(
make_packet_mod_rs_line(packet_id, packet_class_name)
)
@ -208,9 +212,11 @@ def remove_packet_ids(packet_ids: list[int], direction: str, state: str):
direction, state)
new_packet_ids = []
new_packet_class_names = []
for packet_id in existing_packet_ids:
for packet_id, packet_class_name in zip(existing_packet_ids, existing_packet_class_names):
if packet_id not in packet_ids:
new_packet_ids.append(packet_id)
new_packet_class_names.append(packet_class_name)
set_packets(new_packet_ids, existing_packet_class_names, direction, state)
set_packets(new_packet_ids, new_packet_class_names, direction, state)

5
codegen/migrate.py
View file

@ -1,3 +1,4 @@
from lib.code.packet import fix_state
from lib.utils import PacketIdentifier, group_packets
import lib.code.utils
import lib.code.version
@ -24,12 +25,12 @@ for packet in old_packet_list:
assert packet['class'].endswith('.class')
packet_name = old_mappings.get_class(packet['class'][:-6])
old_packets[PacketIdentifier(
packet['id'], packet['direction'].lower(), packet['state'].lower())] = packet_name
packet['id'], packet['direction'].lower(), fix_state(packet['state']))] = packet_name
for packet in new_packet_list:
assert packet['class'].endswith('.class')
packet_name = new_mappings.get_class(packet['class'][:-6])
new_packets[PacketIdentifier(
packet['id'], packet['direction'].lower(), packet['state'].lower())] = packet_name
packet['id'], packet['direction'].lower(), fix_state(packet['state']))] = packet_name
# find removed packets
removed_packets: list[PacketIdentifier] = []