from __future__ import print_function

import os
import sys
import setuptools

from shutil import copyfile
from setuptools import setup, find_packages
from setuptools.extension import Extension
from setuptools.command.develop import develop
from setuptools.command.install import install
from wheel.bdist_wheel import bdist_wheel


VERSION = '1.1.0-beta'
__CWD__ = os.getcwd()
__DIR__ = os.path.dirname(os.path.abspath(__file__))
__URL__ = 'https://github.com/Hikari9/comfortable-swipe-ubuntu'

extension_names = ['service']



try:
    # make sure working directory is here
    os.chdir(__DIR__)


    # save README as long_description
    with open('README.md', 'r') as README_file:
        README = README_file.read()


    # match constants with source
    from comfortable_swipe.constants import *

    # include old conf paths to list from previous versions
    conf_paths = [
        DEFAULT_CONFIG,
        os.path.join(os.getenv('HOME', ''), '.config', 'comfortable-swipe', 'comfortable-swipe.conf'),
        os.path.join('/usr/local/share', 'comfortable-swipe', 'comfortable-swipe.conf'),
        CONFIG
    ]

    # for C++ library
    cpp_macros = dict(
        __COMFORTABLE_SWIPE__VERSION__='"{}"'.format(VERSION),
        __COMFORTABLE_SWIPE__CONFIG__='"{}"'.format(CONFIG)
    )

    # read C++ libraries for comfortable swipe
    extensions = [Extension(
        name='{}.cpp.{}'.format(PYTHON_NAME, extension_name),
        define_macros=list(cpp_macros.items()),
        sources=[os.path.join('cpp', '_python.cpp')],
        extra_compile_args=['-O2', '-Wno-unused-result'],
        libraries=['xdo']
    ) for extension_name in extension_names]


    # add post_install script to install method
    class Install(install):
        def run(self):
            self.pre_install()
            install.run(self)
            self.post_install()
        def pre_install(self):
            pass
        def post_install(self):
            print('running post_install')
            # create program/config directories
            if not os.path.exists(os.path.dirname(CONFIG)):
                os.makedirs(os.path.dirname(CONFIG))

            # copy any of the old config files
            conf_files = [path for path in conf_paths if os.path.exists(path) and os.path.isfile(path)]
            print('Using configuration file at', conf_files[-1])

            if conf_files[-1] != CONFIG:
                # new installation or upgrading from old version, copy to new location
                copyfile(conf_files[-1], CONFIG)
                if conf_files[-1] == DEFAULT_CONFIG:
                    # new installation - copy default configuration
                    print('Copying configuration file to', CONFIG)
                else:
                    # upgrading - delete the deprecated config file (failsafe)
                    print('warning: depcrecated configuration file at', conf_files[-1])
                    print('         you have to remove this manually')

            # enable autostart by default
            from comfortable_swipe import autostart
            autostart.set_status(autostart.ON)
            print('Autostart created at', autostart.target_path())
            print('\nInstallation successful\nTry running: {} start'.format(NAME))


    class Develop(develop):
        def run(self):
            self.pre_uninstall() if self.uninstall else Install.pre_install(self)
            develop.run(self)
            self.post_uninstall() if self.uninstall else Install.post_install(self)
        def pre_uninstall(self):
            print('running pre_uninstall')
            from comfortable_swipe import autostart
            if autostart.get_status() is autostart.ON:
                print('Removing autostart at', autostart.target_path())
                autostart.set_status(autostart.OFF)
        def post_uninstall(self):
            pass

    # Override command classes here
    cmdclass = dict(Install=Install, develop=Develop, bdist_wheel=bdist_wheel)

    # setup python script
    setup_script = setup(
        name=NAME,
        version=VERSION,
        description=DESCRIPTION,
        long_description=README,
        license='MIT',
        author='Rico Tiongson',
        author_email='thericotiongson@gmail.com',
        url=__URL__,
        zip_safe=False,
        packages=find_packages(),
        include_package_data=True,
        entry_points=dict(console_scripts=['{}={}.__main__:main'.format(NAME, PYTHON_NAME)]),
        ext_modules=extensions,
        cmdclass=cmdclass,
        install_requires=['psutil']
    )

finally:
    # move working directory back to where it was before
    os.chdir(__CWD__)