#!/usr/bin/env python
import codecs
import os
import sys

import setuptools
import setuptools.command.test

here = os.path.abspath(os.path.dirname(__file__))


class PyTest(setuptools.command.test.test):
    user_options = [("pytest-args=", "a", "Arguments to pass into py.test")]

    def initialize_options(self):
        setuptools.command.test.test.initialize_options(self)
        try:
            from multiprocessing import cpu_count

            self.pytest_args = [
                "-n",
                str(cpu_count()),
                "--boxed",
                "--mypy",
                "--ignore=setup.py",
            ]
        except (ImportError, NotImplementedError):
            self.pytest_args = ["-n", "1", "--boxed", "--mypy", "--ignore=setup.py"]

    def finalize_options(self):
        setuptools.command.test.test.finalize_options(self)
        self.test_args = []
        self.test_suite = True

    def run_tests(self):
        import pytest

        errno = pytest.main(self.pytest_args)
        sys.exit(errno)


# 'setup.py publish' shortcut.
if sys.argv[-1] == "publish":
    os.system("python -m mypy.stubgen -m defer -o .")
    os.system("python setup.py sdist bdist_wheel")
    os.system("twine upload dist/*")
    sys.exit()

packages = ["defer"]

requires = []
test_requirements = [
    "pytest-cov>=2.6.1",
    "pytest>=4.3.1",
    "pytest-xdist>=1.27.0",
    "pytest-cov>=2.6.1",
]

with codecs.open("README.md", "r", "utf-8") as f:
    readme = f.read()
with codecs.open("HISTORY.md", "r", "utf-8") as f:
    history = f.read()

__title__: str = "py-defer"
__description__: str = "Golang-esque defer functionality"
__url__: str = "http://github.com/naphta/with-defer"
__version__: str = "1.1.4"
__author__: str = "<NAME>"
__author_email__: str = "<EMAIL>"
__license__: str = "MIT"
__copyright__: str = "Copyright 2020 <NAME>"


setuptools.setup(
    name=__title__,
    version=__version__,
    description=__description__,
    long_description=readme,
    long_description_content_type="text/markdown",
    author=__author__,
    author_email=__author_email__,
    url=__url__,
    packages=packages,
    package_data={"": ["LICENSE"], "defer": ["*.pyi"]},
    package_dir={"defer": "defer"},
    include_package_data=True,
    python_requires=">=3.6",
    install_requires=requires,
    license=__license__,
    zip_safe=False,
    classifiers=[
        "Intended Audience :: Developers",
        "License :: OSI Approved :: MIT License",
        "Programming Language :: Python",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.6",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: Implementation :: CPython",
    ],
    cmdclass={"test": PyTest},
    tests_require=test_requirements,
)
