aegis-python/setup.py at main · LeoVasanko/aegis-python · 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
"""Setup script for aeg - builds CFFI extension with libaegis C library."""

import sys
import sysconfig
from pathlib import Path

from cffi import FFI
from setuptools import setup

libaegis_static = Path("libaegis/zig-out/lib") / (
    "aegis.lib" if sys.platform == "win32" else "libaegis.a"
)

ffibuilder = FFI()
ffibuilder.cdef((Path(__file__).parent / "src/aeg/aegis_cdef.h").read_text())

# Free-threaded Python does not support Limited API (abi3)
is_free_threaded = sysconfig.get_config_var("Py_GIL_DISABLED")

ffibuilder.set_source(
    "aeg._aegis",
    """
    #include "aegis.h"
    #include "aegis128l.h"
    #include "aegis128x2.h"
    #include "aegis128x4.h"
    #include "aegis256.h"
    #include "aegis256x2.h"
    #include "aegis256x4.h"
    """,
    include_dirs=["libaegis/src/include"],
    extra_objects=[str(libaegis_static.resolve())],
    py_limited_api=not is_free_threaded,
)

if __name__ == "__main__":
    setup(
        cffi_modules=["setup.py:ffibuilder"],
        options=(
            {"bdist_wheel": {"py_limited_api": "cp310"}} if not is_free_threaded else {}
        ),
    )