stbridge/setup.py at master · dragonlock2/stbridge · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import multiprocessing
import platform
import subprocess
import sys
from pathlib import Path
from setuptools import setup
from pybind11.setup_helpers import Pybind11Extension, build_ext

def msvc_platform():
    bits = 64 if sys.maxsize > 2**32 else 32
    if platform.machine() in ["x86", "AMD64"]:
        plat = "Win32" if bits == 32 else "x64"
    elif platform.machine() in ["ARM", "ARM64"]:
        plat = "ARM" if bits == 32 else "ARM64"
    else:
        plat = ""
    return plat

class build_jabi(build_ext):
    def run(self):
        # build libusb from source
        msvc = sys.platform == "win32" and "GCC" not in sys.version
        plat = msvc_platform()
        if msvc and not list(Path(f"src/libusb/build").glob(f"**/{plat}")):
            msbuild = subprocess.check_output([
                "C:/Program Files (x86)/Microsoft Visual Studio/Installer/vswhere.exe",
                "-requires", "Microsoft.Component.MSBuild", "-find", "MSBuild/**/Bin/MSBuild.exe",
                "-latest", # TODO pick correct version
            ]).decode('utf-8').strip()
            if not msbuild:
                raise Exception("pls install msvc")
            cmds = [
                [msbuild, "/p:configuration=release", f"/p:platform={plat}", "/target:libusb_static", "msvc/libusb.sln"]
            ]
        elif sys.platform != "win32" and not Path("src/libusb/libusb/.libs").exists():
            cmds = [
                ["./bootstrap.sh"],
                ["./autogen.sh", "--disable-udev"],
                ["./configure", "--enable-static", "--disable-shared", "--disable-udev", "--with-pic"],
                ["make", f"-j{multiprocessing.cpu_count()}"],
            ]
        else:
            # MinGW subprocess doesn't run correctly, can manually run above
            cmds = []
        for c in cmds:
            subprocess.run(c, cwd="src/libusb").check_returncode()

        # link libusb
        if msvc:
            self.library_dirs.append(str(list(Path("src/libusb/build").glob(f"**/{plat}/**/libusb-1.0.lib"))[0].parent))
            self.libraries.append("libusb-1.0")
        else:
            self.library_dirs.append("src/libusb/libusb/.libs")
            self.libraries.append("usb-1.0")

        # continue build
        build_ext.run(self)

setup(ext_modules=[
    Pybind11Extension(
        name = "stbridge",
        sources = [
            "src/bindings.cpp",
            "src/stbridge.cpp",
            "src/STLINK-V3-BRIDGE_libusb/src/bridge/bridge.cpp",
            "src/STLINK-V3-BRIDGE_libusb/src/common/stlink_device.cpp",
            "src/STLINK-V3-BRIDGE_libusb/src/common/stlink_interface.cpp",
            "src/STLINK-V3-BRIDGE_libusb/src/error/ErrLog.cpp",
        ],
        include_dirs = [
            "src/libusb/libusb",
            "src/STLINK-V3-BRIDGE_libusb/src/bridge",
            "src/STLINK-V3-BRIDGE_libusb/src/common",
            "src/STLINK-V3-BRIDGE_libusb/src/error",
        ],
    )],
    cmdclass={"build_ext": build_jabi},
)

# python -m build
# python -m twine upload dist/**/*