BETA10 import: Fix FUN_... collisions (#1105)

* Improve BETA10 name import, refactor

* Run formatter

---------

Co-authored-by: jonschz <jonschz@users.noreply.github.com>

tools/ghidra_scripts/lego_util/ghidra_helper.py

@@ -8,6 +8,7 @@ from lego_util.exceptions import (
     TypeNotFoundInGhidraError,
     MultipleTypesFoundInGhidraError,
 )
+from lego_util.globals import GLOBALS, SupportedModules
 
 # Disable spurious warnings in vscode / pylance
 # pyright: reportMissingModuleSource=false
@@ -105,6 +106,12 @@ def sanitize_name(name: str) -> str:
         .replace("`", "'")
     )
 
+    # Importing function names like `FUN_10001234` into BETA10 can be confusing
+    # because Ghidra's auto-generated functions look exactly the same.
+    # Therefore, such function names are replaced by `LEGO_10001234` in the BETA10 import.
+    if GLOBALS.module == SupportedModules.BETA10:
+        new_name = re.sub(r"FUN_([0-9a-f]{8})", r"LEGO1_\1", new_name)
+
     if "<" in name:
         new_name = "_template_" + new_name
 

tools/ghidra_scripts/lego_util/globals.py

@@ -0,0 +1,42 @@
+import logging
+from enum import Enum
+from dataclasses import dataclass, field
+from lego_util.statistics import Statistics
+
+
+class SupportedModules(Enum):
+    LEGO1 = 1
+    BETA10 = 2
+
+    def orig_filename(self):
+        if self == self.LEGO1:
+            return "LEGO1.DLL"
+        return "BETA10.DLL"
+
+    def recomp_filename_without_extension(self):
+        # in case we want to support more functions
+        return "LEGO1"
+
+    def build_dir_name(self):
+        if self == self.BETA10:
+            return "build_debug"
+        return "build"
+
+
+@dataclass
+class Globals:
+    verbose: bool
+    loglevel: int
+    module: SupportedModules
+    running_from_ghidra: bool = False
+    # statistics
+    statistics: Statistics = field(default_factory=Statistics)
+
+
+# hard-coded settings that we don't want to prompt in Ghidra every time
+GLOBALS = Globals(
+    verbose=False,
+    # loglevel=logging.INFO,
+    loglevel=logging.DEBUG,
+    module=SupportedModules.LEGO1,  # this default value will be used when run outside of Ghidra
+)