import os

import re

import sys

from distutils.command.install import INSTALL_SCHEMES

from distutils.command.install_data import install_data

from shutil import move

from tempfile import mkstemp

from setuptools import Command

try:

    from setuptools import setup

except ImportError:

    from distutils.core import setup

"""

See: https://dankeder.com/posts/adding-custom-commands-to-setup-py/

"""

class SetVersion(Command):

    """

    Set the version for the project

    """

    description = "Set the version for the project"

    # command_consumes_arguments = True

    user_options = [

        ('version=', 'v', 'version str')

    ]

    def initialize_options(self):

        """Set default values for options."""

        # Each user option must be listed here with their default value.

        self.version = None

    def finalize_options(self):

        """Post-process options."""

        if self.version is None:

            raise Exception("Parameter --version is missing")

        version_matches = re.match(r"(\d+).(\d+)(?:\.(\d+))?(?:(?:.(dev))|(?:(a|b|rc)(.*)))?", self.version)

        if version_matches is None:

            raise Exception("Parameter --version is not a version string (see )")

        version_list = version_matches.groups()

        version_tag = {'a': 'alpha', 'b': 'beta', 'rc': 'rc'}.get(version_list[4],'final')

        version_ext = int(version_list[5] or 0)

        if version_list[3] and version_list[3] == 'dev':

            version_tag = 'alpha'

            version_ext = 0

        self.version_tuple = (

            int(version_list[0]),

            int(version_list[1]),

            int(version_list[2] or 0),

            version_tag,

            version_ext,

        )

    def run(self):

        filepath = os.path.join(os.path.dirname(os.path.realpath(__file__)),"irinotes","__init__.py")

        print("Changing VERSION in %s to %r" % (filepath, self.version_tuple))

        fh, abs_path = mkstemp()

        with os.fdopen(fh,'w') as temp_file:

            with open(filepath, 'r') as initfiles:

                for l in initfiles:

                    if re.match(r"\s*VERSION\s*=\s*\(\d+\s*,\s*\d+\s*,\s*\d+,\s*\".+\",\s*.+\s*\)", l):

                        temp_file.write("VERSION = %r" % (self.version_tuple,))

                    else:

                        temp_file.write(l)

        os.remove(filepath)

        move(abs_path, filepath)

class osx_install_data(install_data):

    """

    On MacOS, the platform-specific lib dir is /System/Library/Framework/Python/.../

    which is wrong. Python 2.5 supplied with MacOS 10.5 has an Apple-specific fix

    for this in distutils.command.install_data#306. It fixes install_lib but not

    install_data, which is why we roll our own install_data class.

    """

    def finalize_options(self):

        """

        By the time finalize_options is called, install.install_lib is set to the

        fixed directory, so we set the installdir to install_lib. The

        install_data class uses ('install_data', 'install_dir') instead.

        """

        self.set_undefined_options('install', ('install_lib', 'install_dir'))

        install_data.finalize_options(self)

def fullsplit(path, result=None):

    """

    Split a pathname into components (the opposite of os.path.join) in a

    platform-neutral way.

    """

    if result is None:

        result = []

    head, tail = os.path.split(path)

    if head == '':

        return [tail] + result

    if head == path:

        return result

    return fullsplit(head, [tail] + result)

def launch_setup(setup_script_name, setup_script_args):

    """

    Start setup

    """

    if sys.platform == "darwin":

        cmdclasses = {'install_data': osx_install_data}

    else:

        cmdclasses = {'install_data': install_data}

    root_dir = os.path.dirname(__file__)

    if root_dir != '':

        os.chdir(root_dir)

    source_dirs = ['irinotes', 'notes', 'protocols']

    version_variables = {}

    try:

        with open(os.path.join(source_dirs[0], "__init__.py")) as f:

            code = compile(f.read(), "__init__.py", 'exec')

            exec(code, version_variables)

    except:

        pass

    version = version_variables['__version__']

    packages, data_files = [], []

    for source_dir in source_dirs:

        for dirpath, dirnames, filenames in os.walk(source_dir):

            # Ignore dirnames that start with '.'

            for i, dirname in enumerate(dirnames):

                if dirname.startswith('.') or dirname.startswith('__pycache__'): del dirnames[i]

            if '__init__.py' in filenames:

                packages.append('.'.join(fullsplit(dirpath)))

            elif filenames:

                data_files.append([dirpath, [os.path.join(dirpath, f) for f in filenames]])

    # Tell distutils to put the data_files in platform-specific installation

    # locations. See here for an explanation:

    # http://groups.google.com/group/comp.lang.python/browse_thread/thread/35ec7b2fed36eaec/2105ee4d9e8042cb

    for scheme in INSTALL_SCHEMES.values():

        scheme['data'] = scheme['purelib']

    # Small hack for working with bdist_wininst.

    # See http://mail.python.org/pipermail/distutils-sig/2004-August/004134.html

    if len(sys.argv) > 1 and sys.argv[1] == 'bdist_wininst':

        for file_info in data_files:

            file_info[0] = '\\PURELIB\\%s' % file_info[0]

    #write MANIFEST.in

    with open("MANIFEST.in", "w") as m:

        m.write("include CHANGES\n")

        m.write("include LICENSE\n")

        m.write("include README\n")

        m.write("include MANIFEST.in\n")

        m.write("include requirements/base.txt\n")

        m.write("include requirements/dev.txt\n")

        m.write("include requirements/prod.txt\n")

        for entry in data_files:

            file_list = entry[1]

            for filename in file_list:

                m.write("include %s\n" % (filename))

    long_description = ''

    with open('README', 'r') as f:

        long_description = f.read()

    cmdclasses = {**cmdclasses, **{ 'set_version': SetVersion }}

    setup(

        script_name=setup_script_name,

        script_args=setup_script_args,

        name='irinotes',

        version=version,

        author='IRI',

        author_email='contact@iri.centrepompidou.fr',

        packages=packages,

        data_files=data_files,

        cmdclass=cmdclasses,

        scripts=[],

        url='http://www.iri.centrepompidou.fr/dev/hg/irinotes',

        license='CECILL-C',

        description='projet Irinotes',

        long_description=long_description,

        classifiers=[

            'Development Status :: 4 - Beta',

            'Environment :: Web Environment',

            'Framework :: Django',

            'Intended Audience :: Developers',

            'License :: Ceccil-B',

            'Operating System :: OS Independent',

            'Programming Language :: Python',

            'Topic :: Utilities'

        ],

        install_requires=[

            "Django >= 1.11",

            "python-decouple",

            "Unipath",

            "dj-database-url",

            "six",

            "django-auditlog @ https://github.com/IRI-Research/django-auditlog/tarball/master#egg=django-auditlog-0.4.5+IRI",

            "django-extensions",

            "djangorestframework >= 3.6",

            "django-rest-auth[with_social]",

            "djangorestframework-jwt",

            "django-guardian >= 1.4",

            "django-colorful",

            "django-cors-headers",

            "django-concurrency",

            "django-filter",

            "drf-nested-routers",

            "markdown"

        ],

        # dependency_links=[

        #     "https://github.com/IRI-Research/django-auditlog/tarball/master#egg=django-auditlog-0.4.5dev"

        # ]

    )

if __name__ == "__main__":

    script_name = os.path.basename(sys.argv[0])

    script_args = sys.argv[1:]

    launch_setup(script_name, script_args)