Mahdi Sajadieh

# Support both package and standalone imports
try:
    from .architecture import Architecture
except ImportError:
    # Fallback to absolute imports (for multiprocessing spawned processes)
    from redb.extractors.decompiler.bninja.arch.architecture import Architecture


class ArchitectureCreator:
    """Get the correct architecture from the string of architecture retrieved from Binary Ninja."""

    def __init__(self, arch):
        self.arch = arch.lower()

    def get(self) -> Architecture:
        if self.arch == "x86":
            try:
                from .x86 import Arch_x86
            except ImportError:
                # Fallback to absolute imports (for multiprocessing spawned processes)
                from redb.extractors.decompiler.bninja.arch.x86 import Arch_x86

            return Arch_x86()
        else:
            raise ValueError(f"Unsupported architecture: {self.arch}")