from conan import ConanFile
from conan.tools.cmake import CMake, CMakeToolchain, cmake_layout
from conan.tools.files import copy, get, rmdir
from conan.tools.microsoft import is_msvc, is_msvc_static_runtime
from conan.tools.scm import Version
import os

required_conan_version = ">=2.4"


class CapstoneConan(ConanFile):
    name = "capstone"
    description = (
        "Capstone disassembly/disassembler framework: Core (Arm, Arm64, BPF, "
        "EVM, M68K, M680X, MOS65xx, Mips, PPC, RISCV, Sparc, SystemZ, "
        "TMS320C64x, Web Assembly, X86, X86_64, XCore) + bindings."
    )
    license = "BSD-3-Clause"
    url = "https://github.com/conan-io/conan-center-index"
    homepage = "http://www.capstone-engine.org"
    topics = (
        "reverse-engineering", "disassembler", "security", "framework", "arm", "arm64",
        "x86", "sparc", "powerpc", "mips", "x86-64", "ethereum", "systemz",
        "webassembly", "m68k", "m0s65xx", "m680x", "tms320c64x", "bpf", "riscv",
    )
    package_type = "library"
    settings = "os", "arch", "compiler", "build_type"
    options = {
        "shared": [True, False],
        "fPIC": [True, False],
        "use_default_alloc": [True, False],
    }
    default_options = {
        "shared": False,
        "fPIC": True,
        "use_default_alloc": True,
    }

    implements = ["auto_shared_fpic"]
    languages = "C"

    _archs = ["arm", "m68k", "mips", "ppc", "sparc", "sysz", "xcore", "x86", "tms320c64x", "m680x", "evm", "tricore", "riscv"]
    options.update({a: [True, False] for a in _archs})
    default_options.update({a: False for a in _archs})
    default_options.update({"tricore": True, "riscv": True})

    def layout(self):
        cmake_layout(self, src_folder="src")

    def source(self):
        get(self, **self.conan_data["sources"][self.version], strip_root=True)

    def generate(self):
        tc = CMakeToolchain(self)
        tc.cache_variables["BUILD_STATIC_LIBS"] = not self.options.shared
        tc.cache_variables["BUILD_SHARED_LIBS"] = self.options.shared
        tc.cache_variables["BUILD_STATIC_RUNTIME"] = is_msvc_static_runtime(self)

        tc.cache_variables["CAPSTONE_BUILD_TESTS"] = False
        tc.cache_variables["CAPSTONE_BUILD_CSTOOL"] = False
        tc.cache_variables["CAPSTONE_ARCHITECTURE_DEFAULT"] = False
        tc.cache_variables["CAPSTONE_USE_DEFAULT_ALLOC"] = self.options.use_default_alloc
        for a in self._archs:
            tc.cache_variables[f"CAPSTONE_{a.upper()}_SUPPORT"] = self.options.get_safe(a)
        
        tc.generate()

    def build(self):
        cmake = CMake(self)
        cmake.configure()
        cmake.build()

    def package(self):
        copy(self, "LICENSE*.txt", src=self.source_folder, dst=os.path.join(self.package_folder, "licenses"))
        cmake = CMake(self)
        cmake.install()
        rmdir(self, os.path.join(self.package_folder, "lib", "cmake"))
        rmdir(self, os.path.join(self.package_folder, "lib", "pkgconfig"))

    def package_info(self):
        self.cpp_info.libs = ["capstone"]
        if self.options.shared:
            self.cpp_info.defines.append("CAPSTONE_SHARED")