mirror of
https://github.com/mat-1/azalea.git
synced 2024-09-20 07:02:31 +00:00
12a9c8ce65
* 23w03a * 23w04a * 23w05a * 23w06a * fix * 23w07a mojang broke their json data generator so some stuff is missing * didn't mean to commit that file here * 1.19.4-pre2 * fix * 1.19.4-pre3 * fix * how did these packets get here * 1.19.4-pre4 * 1.19.4-rc1 * 1.19.4-rc2 * 1.19.4-rc3 * merge main * remove debugging code * 1.19.4
94 lines
4.2 KiB
Python
Executable file
94 lines
4.2 KiB
Python
Executable file
from typing import Optional
|
|
|
|
|
|
class Mappings:
|
|
__slots__ = ('classes', 'fields', 'methods', 'field_types', 'method_types')
|
|
|
|
def __init__(self, classes, fields, methods, field_types, method_types):
|
|
self.classes = classes
|
|
self.fields = fields
|
|
self.methods = methods
|
|
self.field_types = field_types
|
|
self.method_types = method_types
|
|
|
|
@staticmethod
|
|
def parse(mappings_txt):
|
|
classes = {}
|
|
fields = {}
|
|
methods = {}
|
|
field_types = {}
|
|
method_types = {}
|
|
|
|
current_obfuscated_class_name = None
|
|
|
|
for line in mappings_txt.splitlines():
|
|
if line.startswith('#') or line == '':
|
|
continue
|
|
|
|
if line.startswith(' '):
|
|
# if a line starts with 4 spaces, that means it's a method or a field
|
|
if '(' in line:
|
|
# if it has an opening parenthesis, it's a method
|
|
real_name_with_parameters_and_line, obfuscated_name = line.strip().split(' -> ')
|
|
real_name_with_parameters = real_name_with_parameters_and_line.split(
|
|
':')[-1]
|
|
|
|
real_type, real_name = real_name_with_parameters.split('(')[
|
|
0].split(' ')
|
|
parameters = real_name_with_parameters.split('(')[1].split(')')[
|
|
0]
|
|
|
|
if current_obfuscated_class_name not in methods:
|
|
methods[current_obfuscated_class_name] = {}
|
|
method_types[current_obfuscated_class_name] = {}
|
|
methods[current_obfuscated_class_name][
|
|
f'{obfuscated_name}({parameters})'] = real_name
|
|
method_types[current_obfuscated_class_name][
|
|
f'{obfuscated_name}({parameters})'] = real_type
|
|
else:
|
|
# otherwise, it's a field
|
|
real_name_with_type, obfuscated_name = line.strip().split(' -> ')
|
|
real_type, real_name = real_name_with_type.split(' ')
|
|
|
|
if current_obfuscated_class_name not in fields:
|
|
fields[current_obfuscated_class_name] = {}
|
|
field_types[current_obfuscated_class_name] = {}
|
|
fields[current_obfuscated_class_name][obfuscated_name] = real_name
|
|
field_types[current_obfuscated_class_name][obfuscated_name] = real_type
|
|
else:
|
|
# otherwise it's a class
|
|
real_name, obfuscated_name = line.strip(':').split(' -> ')
|
|
current_obfuscated_class_name = obfuscated_name
|
|
|
|
classes[obfuscated_name] = real_name
|
|
|
|
return Mappings(classes, fields, methods, field_types, method_types)
|
|
|
|
def get_field(self, obfuscated_class_name, obfuscated_field_name):
|
|
return self.fields.get(obfuscated_class_name, {}).get(obfuscated_field_name)
|
|
|
|
def get_class(self, obfuscated_class_name):
|
|
if '<' in obfuscated_class_name:
|
|
first_part, args = obfuscated_class_name.split('<')
|
|
args = args.rstrip('>').strip(';').split(';')
|
|
print(args)
|
|
assert len(args) == 1
|
|
arg = self.get_class(args[0][1:])
|
|
return f'{first_part}<{arg}>'
|
|
return self.classes[obfuscated_class_name]
|
|
|
|
def get_method(self, obfuscated_class_name, obfuscated_method_name, obfuscated_signature):
|
|
# print(obfuscated_class_name, self.methods[obfuscated_class_name])
|
|
return self.methods[obfuscated_class_name][f'{obfuscated_method_name}({obfuscated_signature})']
|
|
|
|
def get_field_type(self, obfuscated_class_name, obfuscated_field_name) -> str:
|
|
return self.field_types[obfuscated_class_name][obfuscated_field_name]
|
|
|
|
def get_method_type(self, obfuscated_class_name, obfuscated_method_name, obfuscated_signature) -> str:
|
|
return self.method_types[obfuscated_class_name][f'{obfuscated_method_name}({obfuscated_signature})']
|
|
|
|
def get_class_from_deobfuscated_name(self, deobfuscated_name) -> Optional[str]:
|
|
for obfuscated_name, real_name in self.classes.items():
|
|
if real_name == deobfuscated_name:
|
|
return obfuscated_name
|
|
return None
|