mirror of
https://github.com/mat-1/azalea.git
synced 2025-08-02 06:16:04 +00:00
d112856ff6
* add example generated metadata.rs * metadata.rs codegen * add the files * add comment to top of metadata.rs * avoid clone * metadata * defaults * defaults * fix metadata readers and writers * fix bad bitmasks and ignore some clippy warnings in generated code * add set_index function to entity metadatas * applying metadata
87 lines
3.9 KiB
Python
87 lines
3.9 KiB
Python
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):
|
|
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
|