import os
import shutil
from setuptools import setup, Extension
from setuptools.command.build_ext import build_ext


class CMakeExtension(Extension):
    def __init__(self, name):
        super().__init__(name, sources=[])


class CMakeBuild(build_ext):
    def run(self):
        # build and install
        cwd = os.getcwd()
        build_dir = os.path.join(cwd, 'build')
        os.makedirs(build_dir, exist_ok=True)
        os.chdir(build_dir)
        install_dir = os.path.join(build_dir, 'install')
        if not self.dry_run:
            self.spawn(
                ['cmake', '..', '-DCMAKE_INSTALL_PREFIX=%s' % install_dir])
            if self.debug:
                self.spawn(['cmake', '--build', '.', '--config',
                           'Debug', '--target', 'install'])
            else:
                self.spawn(['cmake', '--build', '.', '--config',
                           'Release', '--target', 'install'])
        os.chdir(cwd)
        # move
        for ext in self.extensions:
            dst_dir = os.path.dirname(self.get_ext_fullpath(ext.name))
            lib_dir = os.path.join(dst_dir, 'NDIlib')
            os.makedirs(lib_dir, exist_ok=True)
            files = os.listdir(install_dir)
            for filename in files:
                filepath = os.path.join(install_dir, filename)
                if os.path.isfile(filepath):
                    shutil.copy(filepath, lib_dir)


setup(
    name='ndi-python',
    version='5.1.1',
    description='Wrapper package for NDI SDK python bindings.',
    author='Naoto Kondo <cgigcp3yqt@gmail.com>',
    url='https://github.com/buresu/ndi-python',
    license="MIT",
    python_requires='>=3.6',
    ext_modules=[CMakeExtension('NDIlib')],
    cmdclass={'build_ext': CMakeBuild},
    packages=['NDIlib'],
    package_data={'NDIlib': ['*.so*', '*.pyd', '*.dll', '*.dylib', '*.txt']},
    zip_safe=False
)