setup.py

from os import path
from platform import system
from sysconfig import get_config_var

from setuptools import Extension, find_packages, setup
from setuptools.command.bdist_wheel import bdist_wheel
from setuptools.command.build import build

sources = [
    "bindings/python/tree_sitter_bsl/binding.c",
    "grammars/bsl/src/parser.c",
    "grammars/sdbl/src/parser.c",
]
if path.exists("grammars/bsl/src/scanner.c"):
    sources.append("grammars/bsl/src/scanner.c")

macros: list[tuple[str, str | None]] = [
    ("PY_SSIZE_T_CLEAN", None),
    ("TREE_SITTER_HIDE_SYMBOLS", None),
]
if limited_api := not get_config_var("Py_GIL_DISABLED"):
    macros.append(("Py_LIMITED_API", "0x030A0000"))

if system() != "Windows":
    cflags = ["-std=c11", "-fvisibility=hidden"]
else:
    cflags = ["/std:c11", "/utf-8"]


class Build(build):
    def run(self):
        query_dir = "grammars/bsl/queries"
        if path.isdir(query_dir):
            dest = path.join(self.build_lib, "tree_sitter_bsl", "queries")
            self.copy_tree(query_dir, dest)
        super().run()


class BdistWheel(bdist_wheel):
    def get_tag(self):
        python, abi, platform = super().get_tag()
        if python.startswith("cp"):
            python, abi = "cp310", "abi3"
        return python, abi, platform


setup(
    packages=find_packages("bindings/python"),
    package_dir={"": "bindings/python"},
    package_data={
        "tree_sitter_bsl": ["*.pyi", "py.typed"],
        "tree_sitter_bsl.queries": ["*.scm"],
    },
    ext_package="tree_sitter_bsl",
    ext_modules=[
        Extension(
            name="_binding",
            sources=sources,
            extra_compile_args=cflags,
            define_macros=macros,
            include_dirs=["grammars/bsl/src", "grammars/sdbl/src"],
            py_limited_api=limited_api,
        )
    ],
    cmdclass={
        "build": Build,
        "bdist_wheel": BdistWheel,
    },
    zip_safe=False
)