# Copyright 2014 TrilioData Inc.
# All Rights Reserved.


import os
try:
    from setuptools import setup, find_packages
except ImportError:
    from distutils.core import setup, find_packages


def get_version():
    return '4.2.64.17'


def get_package():
    return os.getenv('TVAULT_PACKAGE', 'tvault-contego')


ROOT = os.path.dirname(os.path.realpath(__file__))
TEST_REQUIRES = os.path.join(ROOT, 'test-requirements.txt')


def read_file_list(filename):
    with open(filename) as f:
        return [
            line.strip() for line in f.readlines() if len(line.strip()) > 0
        ]


PACKAGE = get_package()
VERSION = get_version()

# As of now not checking any package dependency , In future can uncomment
# if required
"""PIP_REQUIRES = os.path.join(ROOT, PACKAGE+'-requirements.txt')
INSTALL_REQUIRES = read_file_list(PIP_REQUIRES)"""

COMMON = dict(
    author='TrilioData',
    author_email='support@triliodata.com',
    url='http://www.triliodata.com/',
    version=VERSION,
    classifiers=[
        'Environment :: OpenStack',
        'Intended Audience :: Information Technology',
        'Intended Audience :: System Administrators',
        'License :: www.triliodata.com',
        'Operating System :: POSIX :: Linux',
        'Programming Language :: Python',
        'Programming Language :: Python :: 2',
        'Programming Language :: Python :: 2.7',
        'Programming Language :: Python :: 2.6',
    ],
)
if PACKAGE == 'tvault-contego':
    setup(
        name='tvault-contego',
        description='tvault contego extension for openstack compute.',
        packages=find_packages(exclude=['contego.tests','contego.tests.*']),
        scripts=['bin/tvault-contego'],
        package_data={PACKAGE: ['contego/nova/extension/driver/filemanager_scripts/fsfreeze.sh']},
        include_package_data=True,
        data_files=[],
        **COMMON
    )
elif PACKAGE == 'tvault-contego-api':
    import pkg_resources

    v21extensions = False
    for x in pkg_resources.working_set:
        if 'nova' in str(x):
            group = pkg_resources.get_entry_map(x, 'nova.api.v21.extensions')
            if group is not None:
                v21extensions = True
                setup(
                    name='tvault-contego-api',
                    description='tvault contego extension for openstack compute api.',
                    packages=['contego', 'contego.nova', 'contego.nova.osapi'],
                    include_package_data=True,
                    entry_points="""
                          [nova.api.v21.extensions]
                          contego_extension = contego.nova.osapi.contego_extension:ContegoExtension
                          """,
                    **COMMON
                )
                break

    if v21extensions is False:
        setup(
            name='tvault-contego-api',
            description='tvault contego extension for openstack compute api.',
            packages=['contego', 'contego.nova', 'contego.nova.osapi'],
            **COMMON
        )
else:
    setup(
        name=PACKAGE,
        description='tvault contego extension for openstack compute.',
        packages=find_packages(exclude=['contego.tests','contego.tests.*']),
	scripts=['bin/tvault-contego'],
        package_data={PACKAGE: ['contego/nova/extension/driver/filemanager_scripts/fsfreeze.sh']},
        include_package_data=True,
        data_files=[],
        **COMMON
    )