setup.py

# This file is part of "echion" which is released under MIT.
#
# Copyright (c) 2023 Gabriele N. Tornetta <phoenix1987@gmail.com>.

import os
import sys
from pathlib import Path

from setuptools import Extension
from setuptools import find_packages
from setuptools import setup


PLATFORM = sys.platform.lower()

DISABLE_NATIVE = os.environ.get("UNWIND_NATIVE_DISABLE")

LDADD = {
    "linux": ["-l:libunwind.a", "-l:liblzma.a"] if not DISABLE_NATIVE else [],
}

# add option to colorize compiler output
COLORS = [
    "-fdiagnostics-color=always" if PLATFORM == "linux" else "-fcolor-diagnostics"
]

if PLATFORM == "darwin":
    CFLAGS = ["-mmacosx-version-min=10.15"]
else:
    CFLAGS = []

if DISABLE_NATIVE:
    CFLAGS += ["-DUNWIND_NATIVE_DISABLE"]

echionmodule = Extension(
    "echion.core",
    sources=["echion/coremodule.cc", "echion/frame.cc", "echion/render.cc", "echion/danger.cc"],
    include_dirs=["."],
    define_macros=[(f"PL_{PLATFORM.upper()}", None)],
    extra_compile_args=["-std=c++17", "-Wall", "-Wextra"] + CFLAGS + COLORS,
    extra_link_args=LDADD.get(PLATFORM, []),
    libraries=["unwind", "lzma"] if (PLATFORM != "darwin" and not DISABLE_NATIVE) else []
)

setup(
    name="echion",
    author="Gabriele N. Tornetta",
    description="In-process Python sampling profiler",
    long_description=Path("README.md")
    .read_text()
    .replace(
        'src="art/', 'src="https://raw.githubusercontent.com/P403n1x87/echion/main/art/'
    ),
    ext_modules=[echionmodule],
    entry_points={
        "console_scripts": ["echion=echion.__main__:main"],
    },
    packages=find_packages(exclude=["tests"]),
)