#!/usr/bin/env python
import glob
import os
import subprocess
import sys
import setuptools
from distutils.sysconfig import get_config_vars
from packaging import version
from setuptools import Distribution as _Distribution, setup
from setuptools.command.build_ext import build_ext as _build_ext
import errno
parse_version = version.parse
try:
 from setuptools.command.build_clib import build_clib as _build_clib
except ImportError:
 # We ignore type checking because mypy shows an import error
 from distutils.command.build_clib import build_clib as _build_clib # type: ignore

base_dir = os.path.dirname(__file__)
src_dir = os.path.join(base_dir, "src")
# sys.path.insert(0, base_dir)

use_system_lib = True
if os.environ.get("BUILD_LIB") == "1":
 use_system_lib = False


class BuildClib(_build_clib):
 def build_libraries(self, libraries):
 raise Exception("build_libraries")

 def check_library_list(self, libraries):
 raise Exception("check_library_list")

 def get_library_names(self):
 return ["fuzzy"]

 def get_source_files(self):
 files = glob.glob(os.path.relpath("src/ssdeep-lib/*"))
 files += glob.glob(os.path.relpath("src/ssdeep-lib/m4/*"))
 return files

 def run(self):
 if use_system_lib:
 return

 build_env = {
 key: val
 for key, val in get_config_vars().items()
 if key in ("LDFLAGS", "CFLAGS", "CC", "CCSHARED", "LDSHARED")
 and key not in os.environ
 }
 os.environ.update(build_env)

 build_temp = os.path.abspath(self.build_temp)
 try:
 os.makedirs(build_temp)
 except OSError as e:
 if e.errno != errno.EEXIST:
 raise

 # libtoolize: Install required files for automake
 returncode = subprocess.call(
 "(cd src/ssdeep-lib && libtoolize && autoreconf --force)",
 shell=True
 )
 if returncode != 0:
 # try harder
 returncode = subprocess.call(
 "(cd src/ssdeep-lib && automake --add-missing && autoreconf --force)",
 shell=True
 )
 if returncode != 0:
 sys.exit("Failed to reconfigure the project build.")

 configure_cmd = os.path.abspath(os.path.relpath("src/ssdeep-lib/configure"))

 configure_flags = [
 "--disable-shared",
 "--enable-static",
 "--disable-debug",
 "--disable-dependency-tracking",
 "--with-pic",
 ]

 returncode = subprocess.call(
 [configure_cmd]
 + configure_flags
 + ["--prefix", os.path.abspath(self.build_clib)],
 cwd="src/ssdeep-lib"
 )
 returncode = subprocess.call(
 ["make"],
 cwd="src/ssdeep-lib"
 )
 returncode = subprocess.call(
 ["make", "install"],
 cwd="src/ssdeep-lib"
 )

 if returncode != 0:
 sys.exit("Failed while building ssdeep lib.")


class BuildExt(_build_ext):
 def run(self):
 if self.distribution.has_c_libraries():
 build_clib = self.get_finalized_command("build_clib")
 self.include_dirs.append(
 os.path.join(build_clib.build_clib, "include"),
 )
 self.library_dirs.insert(
 0,
 os.path.join(build_clib.build_clib, "lib64"),
 )
 self.library_dirs.insert(
 0,
 os.path.join(build_clib.build_clib, "lib"),
 )

 # ToDo: Is there a better way?
 for ext in self.extensions:
 ext.extra_objects += get_objects()

 return _build_ext.run(self)


class Distribution(_Distribution):
 def has_c_libraries(self):
 return not use_system_lib


def get_objects():
 objects = glob.glob("src/ssdeep-lib/.libs/*.o")
 if len(objects) > 0:
 return objects
 return glob.glob("src/ssdeep-lib/.libs/*.obj")


about = {}
with open(os.path.join(src_dir, "ssdeep", "__about__.py")) as f:
 exec(f.read(), about)

with open(os.path.join(base_dir, "README.rst")) as f:
 long_description = f.read()

# On some systems(e.g. Debian 8, CentOS 7) the setuptools package is very old.
# We try to install an old version of pytest-runner without setuptools_scm dependency.
# See: https://github.com/pytest-dev/pytest-runner/blob/master/CHANGES.rst
if parse_version(setuptools.__version__) < parse_version("12"):
 setup_requires = ["pytest-runner<2.4"]
else:
 setup_requires = ["pytest-runner"]


setup(
 name=about["__title__"],
 version=about["__version__"],

 description=about["__summary__"],
 long_description=long_description,
 license=about["__license__"],
 url=about["__uri__"],

 zip_safe=False,
 author=about["__author__"],
 classifiers=[
 "Development Status :: 5 - Production/Stable",
 "Intended Audience :: Developers",
 "License :: OSI Approved :: GNU Lesser General Public License v3 or later (LGPLv3+)",
 "Operating System :: OS Independent",
 "Programming Language :: Python",
 "Programming Language :: Python :: 3.6",
 "Programming Language :: Python :: 3.7",
 "Programming Language :: Python :: 3.8",
 "Programming Language :: Python :: 3.9",
 "Programming Language :: Python :: 3.10",
 "Programming Language :: Python :: Implementation :: CPython",
 "Programming Language :: Python :: Implementation :: PyPy",
 "Topic :: Software Development :: Libraries :: Python Modules",
 ],
 keywords="ssdeep",
 install_requires=[
 "cffi>=1.0.0",
 ],
 setup_requires=[
 "cffi>=1.0.0",
 ] + setup_requires,
 tests_require=[
 "pytest",
 ],
 extras_require={
 "docstest": [
 "doc8",
 "readme_renderer >= 16.0",
 "sphinx",
 "sphinx_rtd_theme",
 ],
 },
 packages=[
 "ssdeep"
 ],
 include_package_data=True,
 cffi_modules=["src/ssdeep/_build.py:ffi"],
 package_dir={"": "src"},
 cmdclass={
 "build_clib": BuildClib,
 "build_ext": BuildExt
 },
 distclass=Distribution,
)