1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
import re
import os
# utilities that could be used for things other than codegen
def to_snake_case(name: str):
s = re.sub("([A-Z])", r"_\1", name).replace(".", "_").replace("/", "_")
return s.lower().strip("_")
def to_camel_case(name: str):
s = re.sub(
r"[_ ](\w)",
lambda m: m.group(1).upper(),
name.replace(".", "_").replace("/", "_"),
)
s = upper_first_letter(s)
# if the first character is a number, we need to add an underscore
# maybe we could convert it to the number name (like 2 would become "two")?
if s[0].isdigit():
s = f"_{s}"
return s
def upper_first_letter(name: str):
return name[0].upper() + name[1:]
def padded_hex(n: int):
return f"0x{n:02X}"
class PacketIdentifier:
def __init__(self, packet_id: int, direction: str, state: str):
self.packet_id = packet_id
self.direction = direction
self.state = state
def __eq__(self, other):
return (
self.packet_id == other.packet_id
and self.direction == other.direction
and self.state == other.state
)
def __hash__(self):
return hash((self.packet_id, self.direction, self.state))
def __str__(self):
return f"{self.packet_id} {self.direction} {self.state}"
def __repr__(self):
return f"PacketIdentifier({self.packet_id}, {self.direction}, {self.state})"
def group_packets(packets: list[PacketIdentifier]):
packet_groups: dict[tuple[str, str], list[int]] = {}
for packet in packets:
key = (packet.direction, packet.state)
if key not in packet_groups:
packet_groups[key] = []
packet_groups[key].append(packet.packet_id)
return packet_groups
def get_dir_location(name: str):
return os.path.join(os.path.dirname(os.path.dirname(__file__)), name)
def identifier_to_namespace(ident: str):
return ident.split(":")[0]
def identifier_to_path(ident: str):
if ":" not in ident:
return ident
namespace, path = ident.lstrip("#").split(":")
if namespace in {"minecraft", "brigadier"}:
return path
# support for mods
return f"{namespace}_{path}"
|
