setup.py

import re
import os
import sys
import shutil
import platform
import subprocess
from setuptools import setup, find_packages
from setuptools.command.bdist_wheel import bdist_wheel
from glob import glob


data_path = "bin/EndstoneRuntime"
native_path = data_path + "/Native"
library_path = data_path + "/Lib"


class TagProvider(bdist_wheel):
    def get_tag(self):
        python_version = f"cp{sys.version_info[0]}{sys.version_info[1]}"

        if platform.system() != "Windows":
            raise Exception("unsupported system")
        if platform.architecture() != ("64bit", "WindowsPE"):
            raise Exception("unsupported architecture")
        return (python_version, python_version, "win_amd64")


def build_native():
    python_version = f"{sys.version_info[0]}.{sys.version_info[1]}.x"
    env = os.environ.copy()
    env["PYTHON_VERSION"] = python_version
    proj_dir = os.path.dirname(__file__)
    subprocess.run(
        ["xmake", "f", "-p", "windows", "-a", "x64", "-m", "release", "-y"],
        env=env,
        cwd=proj_dir,
        check=True,
    )
    subprocess.run(
        ["xmake", "-v", "-y"],
        env=env,
        cwd=proj_dir,
        check=True,
    )

    os.makedirs(native_path, exist_ok=True)
    ext = [".dll", ".pdb", ".json"]
    for e in ext:
        for f in glob(f"bin/EndstoneRuntime/*{e}"):
            shutil.copy(f, native_path)


if not os.path.exists(data_path):
    build_native()

setup(
    packages=find_packages(library_path) + ["."],
    package_dir={"": library_path, ".": native_path},
    package_data={"": ["*"]},
    setup_requires=["setuptools_scm"],
    use_scm_version={
        "root": "endstone",
        "relative_to": __file__,
        "version_scheme": "guess-next-dev",
        "local_scheme": "no-local-version",
    },
    cmdclass={"bdist_wheel": TagProvider},
)