import os
import sys

from glob import glob

try:
    import setuptools
except ImportError:
    sys.stderr.write("Network Runner needs setuptools to build.\n")
    sys.exit(1)


def get_data_files(directory):
    # get directory contents
    lst = glob(os.path.join(directory, '*'))
    # files list with out dirs in it.
    files = [f for f in lst if os.path.isfile(f)]
    # build data structure to return [('dir', ['file', 'file'])]
    data_files = [('{}{}'.format(os.sep, directory), files)] if files else []
    # recurse to build data structure for dirs
    for d in lst:
        if os.path.isdir(d):
            data_files.extend(get_data_files(d))
    return data_files


def main():

    with open("README.rst", "r") as fh:
        long_description = fh.read()

    github = 'https://github.com/ansible-network/network-runner/'
    github_issues = '{}issues'.format(github)

    setuptools.setup(
        name='network-runner',
        version='0.3.6',
        description='Abstracton and Python API for Ansible Networking',
        long_description=long_description,
        author='Ansible',
        author_email='info@ansible.com',
        url='https://github.com/ansible-network/network-runner/',
        project_urls={
            'Bug Tracker': github,
            'Source Code': github_issues,
        },
        license='Apache 2',
        python_requires='>=2.7',
        packages=setuptools.find_packages(),
        classifiers=[
            'Environment :: Console',
            'Intended Audience :: Developers',
            'Intended Audience :: Information Technology',
            'Intended Audience :: System Administrators',
            'Natural Language :: English',
            'Operating System :: POSIX',
            'Programming Language :: Python :: 2',
            'Programming Language :: Python :: 2.7',
            'Programming Language :: Python :: 3',
            'Programming Language :: Python :: 3.5',
            'Programming Language :: Python :: 3.6',
            'Programming Language :: Python :: 3.7',
            'Topic :: System :: Systems Administration',
            'Topic :: Utilities',
        ],
        data_files=get_data_files('etc/ansible/roles/network-runner'),
    )


if __name__ == '__main__':
    main()