tintype/setup.py at main · facebookincubator/tintype · 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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
# Copyright (c) Meta Platforms, Inc. and affiliates.
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.

# @nolint
import os
import sys
import sysconfig

from pybind11.setup_helpers import build_ext, Pybind11Extension
from setuptools import setup

# The repo root is the tintype package itself.
# After ShipIt exports fbcode/tintype/ → repo root, the directory layout is:
#
#   ./                    ← repo root = tintype package
#   ├── __init__.py
#   ├── _snapshot.cpp     ← pybind11 extension source
#   ├── snapshot_lib/     ← C++ library sources
#   ├── _exception_hook.py
#   ├── _sampling.py
#   ├── _traceback.py
#   └── ...


def find_zstd():
    """Find zstd library by checking common system include/lib paths."""
    include_dirs = []
    library_dirs = []

    search_prefixes = ["/usr", "/usr/local", "/opt/homebrew"]

    # Also check conda environment if active
    conda_prefix = os.environ.get("CONDA_PREFIX")
    if conda_prefix:
        search_prefixes.insert(0, conda_prefix)

    for prefix in search_prefixes:
        header = os.path.join(prefix, "include", "zstd.h")
        if os.path.exists(header):
            include_dirs.append(os.path.join(prefix, "include"))
            library_dirs.append(os.path.join(prefix, "lib"))
            break

    return include_dirs, library_dirs


def set_macos_deployment_target():
    """Raise the macOS deployment target to at least 10.15.

    pybind11's setup_helpers pins `-mmacosx-version-min=10.14` for C++17 and
    above whenever MACOSX_DEPLOYMENT_TARGET is unset, but libc++ marks
    <filesystem> unavailable before 10.15. Honor the interpreter's own target
    when it is already higher so the wheel's platform tag stays consistent.
    """
    if sys.platform != "darwin" or "MACOSX_DEPLOYMENT_TARGET" in os.environ:
        return

    minimum = (10, 15)
    target = str(sysconfig.get_config_var("MACOSX_DEPLOYMENT_TARGET") or "")
    try:
        parsed = tuple(int(part) for part in target.split("."))
    except ValueError:
        parsed = ()

    if parsed < minimum:
        target = ".".join(str(part) for part in minimum)

    os.environ["MACOSX_DEPLOYMENT_TARGET"] = target


# Must run before Pybind11Extension is constructed: its `cxx_std` setter reads
# MACOSX_DEPLOYMENT_TARGET at construction time.
set_macos_deployment_target()

zstd_include_dirs, zstd_library_dirs = find_zstd()

if not zstd_include_dirs:
    import warnings
    warnings.warn(
        "zstd headers not found in /usr, /usr/local, or /opt/homebrew. "
        "Install zstd (e.g., apt install libzstd-dev, brew install zstd) "
        "or set CFLAGS/LDFLAGS to point to your zstd installation.",
        stacklevel=1,
    )

ext_modules = [
    Pybind11Extension(
        "tintype._snapshot",
        sources=[
            "_snapshot.cpp",
            "snapshot_lib/SnapshotCapture.cpp",
            "snapshot_lib/SnapshotReader.cpp",
            "snapshot_lib/SnapshotStats.cpp",
            "snapshot_lib/SnapshotWriter.cpp",
        ],
        include_dirs=["."] + zstd_include_dirs,
        library_dirs=zstd_library_dirs,
        libraries=["zstd"],
        cxx_std=20,
        define_macros=[("PYBIND11_DETAILED_ERROR_MESSAGES", None)],
    ),
]

setup(
    ext_modules=ext_modules,
    cmdclass={"build_ext": build_ext},
    packages=["tintype", "tintype.dap", "tintype.tests", "tintype.utils"],
    package_dir={
        "tintype": ".",
        "tintype.dap": "dap",
        "tintype.tests": "tests",
        "tintype.utils": "utils",
    },
    package_data={
        "tintype": ["_snapshot.pyi", "py.typed"],
    },
)