drgn/setup.py
2019-04-11 15:51:20 -07:00

114 lines
3.9 KiB
Python
Executable File

#!/usr/bin/env python3
import re
import os.path
from distutils.dir_util import mkpath
from distutils.errors import DistutilsSetupError
from distutils.file_util import copy_file
from setuptools import setup, find_packages
from setuptools.extension import Extension
from setuptools.command.build_ext import build_ext
from setuptools.command.sdist import sdist
import subprocess
import sys
class my_build_ext(build_ext):
user_options = [
('inplace', 'i', 'put compiled extension into the source directory'),
('parallel=', 'j', 'number of parallel build jobs'),
]
boolean_options = ['inplace']
help_options = []
def _run_autotools(self):
if not os.path.exists('libdrgn/configure'):
subprocess.check_call(['autoreconf', '-i', 'libdrgn'])
mkpath(self.build_temp)
if not os.path.exists(os.path.join(self.build_temp, 'Makefile')):
args = [
os.path.relpath('libdrgn/configure', self.build_temp),
'--disable-static', '--with-python=' + sys.executable,
]
subprocess.check_call(args, cwd=self.build_temp)
def get_source_files(self):
self._run_autotools()
args = ['make', '-C', self.build_temp, 'distfiles', '-s']
return [
os.path.normpath(os.path.join(self.build_temp, path)) for path in
subprocess.check_output(args, universal_newlines=True).splitlines()
]
def run(self):
self._run_autotools()
args = ['make', '-C', self.build_temp, '_drgn.la']
if self.parallel:
args.append(f'-j{self.parallel}')
subprocess.check_call(args)
so = os.path.join(self.build_temp, '.libs/_drgn.so')
if self.inplace:
copy_file(so, self.get_ext_fullpath('_drgn'))
self.inplace = 0
build_path = self.get_ext_fullpath('_drgn')
mkpath(os.path.dirname(build_path))
copy_file(so, build_path)
class my_sdist(sdist):
user_options = sdist.user_options + [
('force', 'f',
'create the source distribution even if the repository is unclean'),
]
boolean_options = sdist.boolean_options + ['force']
def initialize_options(self):
super().initialize_options()
self.force = 0
def run(self):
# In order to avoid shipping a stale source distribution (e.g., due to
# pypa/setuptools#436 or the autotools output being out of date),
# require the repository to be clean (no unknown or ignored files).
# This check can be disabled with --force.
if (not self.force and subprocess.check_output(['git', 'clean', '-dnx'])):
raise DistutilsSetupError('repository has untracked or ignored files; '
'please run git clean -dfx or use --force')
super().run()
with open('libdrgn/drgn.h', 'r') as f:
drgn_h = f.read()
version_major = re.search('^#define DRGN_VERSION_MAJOR ([0-9])+$', drgn_h,
re.MULTILINE).group(1)
version_minor = re.search('^#define DRGN_VERSION_MINOR ([0-9])+$', drgn_h,
re.MULTILINE).group(1)
version_patch = re.search('^#define DRGN_VERSION_PATCH ([0-9])+$', drgn_h,
re.MULTILINE).group(1)
version = f'{version_major}.{version_minor}.{version_patch}'
setup(
name='drgn',
version=version,
packages=find_packages(exclude=['examples', 'scripts', 'tests']),
# This is here so that setuptools knows that we have an extension; it's
# actually built using autotools/make.
ext_modules=[Extension(name='_drgn', sources=[])],
cmdclass={
'build_ext': my_build_ext,
'sdist': my_sdist,
},
entry_points={
'console_scripts': ['drgn=drgn.internal.cli:main'],
},
author='Omar Sandoval',
author_email='osandov@osandov.com',
description='Scriptable debugger library',
license='GPL-3.0+',
url='https://github.com/osandov/drgn',
)