Update

Author: mos9527

.vscode/launch.json

@@ -10,13 +10,11 @@
             "type": "debugpy",
             "request": "launch",
             "module": "UnityPyTypetreeCodegen",
-            "args": [
-                "--files",
-                "D:\\Reverse\\proseka_reverse\\game_binary\\v510\\exported\\shim",
-                // "--json",
-                // "c:\\Users\\mos9527\\TypetreeTests\\Test\\TypeTree.json",
-                "--filter",
-                "(Live2D|Cubism)+.*"
+            "args": [                
+                "--il2cpp",
+                "D:\\Reverse\\proseka_reverse\\UntitledCharts\\scripts\\.temp\\typetree\\libil2cpp.so",
+                "--metadata",
+                "D:\\Reverse\\proseka_reverse\\UntitledCharts\\scripts\\.temp\\typetree\\global-metadata.dat"
             ]
         }
     ]

UnityPyTypetreeCodegen/__main__.py

@@ -303,11 +303,11 @@ def emit_line(*lines: str):
             f.write("\n")
 
     logger.info(
-        f"Subpass 1: Generating class dependency graph for {namespace or "<default namespace>"}"
+        f"Subpass 1: Generating class dependency graph for {namespace or '<default namespace>'}"
     )
     emit_line("# fmt: off")
     emit_line("# Auto-generated by https://github.com/mos9527/UnityPyTypetreeCodegen")
-    emit_line(f"# Python definition for {namespace or "<default namespace>"}", "")
+    emit_line(f"# Python definition for {namespace or '<default namespace>'}", "")
     if import_root:
         emit_line(f"from {import_root} import *")
     for clazz, parent in import_defs.items():
@@ -452,29 +452,44 @@ def __open(fname: str):
 
 
 import re, logging
-from UnityPy.helpers.TypeTreeGenerator import TypeTreeGenerator
-
+from TypeTreeGeneratorAPI import TypeTreeGenerator
 
 def __main__():
     parser = argparse.ArgumentParser()
-    parser.add_argument(
-        "--files",
-        help="Load typetree dump from game assembly folder containing the DLLs",
-    )
-    parser.add_argument(
-        "--json",
-        help="Load tree dump in json format {str[fullname]: List[TypeTreeNode]},...",
-    )
     parser.add_argument(
         "--unity-version",
         help="Unity version to use for typetree generation",
         default="2022.3.21f1",
     )
+    parser.add_argument(
+        "--backend",
+        help="Backend to use for code generation",
+        default="AssetRipper",
+    )
     parser.add_argument(
         "--filter",
         help="Filter classnames by regex",
         default=".*",
     )
+    parser.add_argument(
+        "--json",
+        help="[JSON] Load tree dump in json format {str[fullname]: List[TypeTreeNode]},...",
+    )
+    parser.add_argument(
+        "--asm",
+        help="[Asm] Load typetree dump from game assembly DLL",
+        type=str
+    )
+    parser.add_argument(
+        "--il2cpp",
+        help="[IL2CPP] Load typetree dump from IL2CPP binaries",
+        type=str
+    )
+    parser.add_argument(
+        "--metadata",
+        help="[IL2CPP] Load typetree dump from metadata files",
+        type=str
+    )
     parser.add_argument(
         "--log-level",
         choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
@@ -490,20 +505,26 @@ def __main__():
     shutil.rmtree(args.outdir, ignore_errors=True)
     os.makedirs(args.outdir, exist_ok=True)
     typetree = dict()
-    if args.files:
-        generator = TypeTreeGenerator(args.unity_version)
-        generator.load_local_dll_folder(args.files)
-        defines = generator.get_monobehavior_definitions()
-        typetree = dict()
-        for module, fullname in defines:
-            try:
-                nodes = generator.get_nodes(module, fullname)
-                typetree[fullname] = nodes
-            except Exception as e:
-                logger.error(f"Failed to generate typetree for {module}.{fullname}")
-    if args.json:
+    gen = TypeTreeGenerator(args.unity_version, args.backend)
+    def populate_gen():
+        # https://github.com/UnityPy-Org/TypeTreeGeneratorAPI/pull/1
+        for asm,clz in gen.get_monobehavior_definitions():
+            node = gen.get_nodes_as_json(asm, clz)
+            pass
+    if args.asm:
+        print("Loading .NET Assembly", args.asm)
+        gen.load_dll(args.asm)
+        populate_gen()
+    elif args.il2cpp and args.metadata:
+        print("Loading IL2CPP", args.il2cpp, args.metadata)
+        gen.load_il2cpp(open(args.il2cpp, "rb").read(), open(args.metadata, "rb").read())
+        populate_gen()
+    elif args.json:
+        print("Loading JSON", args.json)
         with open(args.json, "r") as f:
             typetree = json.load(f)
+    else:
+        raise ValueError("No valid input source specified.")
     if typetree:
         regex = re.compile(args.filter)
         typetree = {k: v for k, v in typetree.items() if regex.match(k)}