#!/usr/bin/env python
#-*- coding:utf-8 -*-

from setuptools import setup, find_packages
import sys
import os
import shutil
import platform

def print_help():
    print("python setup.py help        -- 查看帮助")
    print("python setup.py             -- 执行编译")
    print("python setup.py build       -- 执行编译")
    print("python setup.py install     -- 执行安装")
    print("python setup.py uninstall   -- 删除已安装的Hikyuu")
    print("python setup.py clear       -- 清除本地编译结果")
    print("python setup.py bdist_wheel -- 生成wheel安装包")

current_plat = sys.platform
current_bits = 64 if sys.maxsize > 2**32 else 32

if len(sys.argv) < 1 or (len(sys.argv) > 1 and sys.argv[1] not in ('bdist_wheel', 'build', 'install', 'uninstall', 'clear')):
    print("参数错误，请参考以下命令执行：")
    print_help()
    exit(0)

print("\nchecking python version ...")

py_version = platform.python_version_tuple() 
py_version = int(py_version[0])*10 + int(py_version[1])
if py_version < 31:
    print("Python version must >= 3.1 !")
    sys.exit(0)

py_version_changed = True
if os.path.exists('py_version'):
    with open('py_version', 'r+') as f:
        old_py_version = f.read()
        if old_py_version == str(py_version):
            py_version_changed = False
        else:
            f.seek(0)
            f.write(str(py_version))
            print('old python version:', int(old_py_version)*0.1)
else:
    with open('py_version', 'w') as f:
        f.write(str(py_version))

print('current python version:', int(py_version)*0.1, '\n')

if sys.argv[-1] == 'uninstall':
    if current_plat == 'win32':
        site_lib_dir = sys.base_prefix + "/lib/site-packages"
    else:
        usr_dir = os.path.expanduser('~')
        site_lib_dir = '{}/.local/lib/python{:>.1f}/site-packages'.format(usr_dir, py_version*0.1)
    for dir in os.listdir(site_lib_dir):
        if dir == 'hikyuu' or (len(dir) > 6 and dir[:6] == 'Hikyuu'):
            print('delete', site_lib_dir + '/' + dir)
            shutil.rmtree(site_lib_dir + '/' + dir)
    print("Uninstall finished!")
    exit(0)
    
if sys.argv[-1] == 'clear':
    if os.path.lexists('.xmake'):
        print('delete .xmake')
        shutil.rmtree('.xmake')
    if os.path.lexists('build'):
        print('delete build')
        shutil.rmtree('build')
    if os.path.lexists('Hikyuu.egg-info'):
        print('delete Hikyuu.egg-info')
        shutil.rmtree('Hikyuu.egg-info')
    if os.path.exists('py_version'):
        print('delete py_version')
        os.remove('py_version')
    for r, d, f_list in os.walk('hikyuu'):
        for name in f_list:
            if (len(name) > 4 and name[-4:] in ('.dll','.exe','.pyd')) \
                   or (len(name) > 3 and name[-3:] == '.so')  \
                   or (len(name) > 6 and name[-6:] == '.dylib'):
                print('delete', r + '/' + name)
                os.remove(os.path.join(r, name))
    print('clear finished!')
    exit(0)

print("checking xmake ...")
xmake = os.system("xmake --version")

if xmake != 0:
    print("Please read https://hikyuu.readthedocs.io/en/latest/developer.html")
    exit(0)

print("checking boost ...")
current_dir = os.getcwd()
current_boost_root = None
current_boost_lib = None
if 'BOOST_ROOT' not in os.environ or 'BOOST_LIB' not in os.environ:
    for dir in os.listdir():
        if len(dir) >= 5 and dir[:5] == 'boost' and os.path.isdir(dir):
            current_boost_root = current_dir + '/' + dir
            current_boost_lib = current_dir + '/' + dir + '/stage/lib'
            os.environ['BOOST_ROOT'] = current_boost_root
            os.environ['BOOST_LIB'] = current_boost_lib
else:
    current_boost_root = os.environ['BOOST_ROOT']
    current_boost_lib = os.environ['BOOST_LIB']
    
if current_boost_root is None or current_boost_lib is None:
    print("Please read https://hikyuu.readthedocs.io/en/latest/developer.html")
    exit(0)
    
print('BOOST_ROOT:', current_boost_root)
print('BOOST_LIB:', current_boost_lib)


if py_version_changed:
    if current_plat == 'win32' and current_bits == 64:
        build_hikyuu_pywrap_dir = 'build\\release\\windows\\x64\\.objs\\windows\\x64\\release\\hikyuu_pywrap'
    elif current_plat == 'win32' and current_bits == 32:
        build_hikyuu_pywrap_dir = 'build\\release\\windows\\x64\\.objs\\windows\\x64\\release\\hikyuu_pywrap'
    elif current_plat == 'linux' and current_bits == 64:
        build_hikyuu_pywrap_dir = 'build/release/linux/x86_64/.objs/linux/x86_64/release/hikyuu_pywrap'
    else:
        print("************不支持的平台打包**************")
        exit(0)
        
    if os.path.lexists(build_hikyuu_pywrap_dir):
        shutil.rmtree(build_hikyuu_pywrap_dir)
    
print('\ncompile boost ...')
if py_version_changed and os.path.lexists('{}/bin.v2/libs/python'.format(current_boost_root)):
    shutil.rmtree('{}/bin.v2/libs/python'.format(current_boost_root))
if current_plat == 'win32':
    os.chdir(current_boost_root)
    if not os.path.exists('b2.exe'):
        os.system('bootstrap.bat')
    os.system('b2 release link=shared address-model=64 -j 4 --with-python --with-date_time --with-filesystem --with-system --with-serialization --with-test')
    os.chdir(current_dir)
else:
    cmd = 'cd {boost} ; if [ ! -f "b2" ]; then ./bootstrap.sh ; fi; ./b2 release link=shared address-model=64 -j 4 --with-python --with-date_time --with-filesystem --with-system --with-serialization --with-test; cd {current}'.format(boost=current_boost_root, current=current_dir)
    #print(cmd)
    os.system(cmd)


if len(sys.argv) == 1 or sys.argv[-1] == 'build':
    if py_version_changed:
        os.system("xmake f -c")
        os.system("xmake f --with-unit-test=y")
    os.system("xmake")
    os.system("xmake r unit-test")
    sys.exit(0)

if sys.argv[-1] == 'install':
    if py_version_changed:
        os.system("xmake f -c")
    if current_plat == 'win32':
        install_dir = sys.base_prefix + "\\Lib\\site-packages\\hikyuu"
    else:
        usr_dir = os.path.expanduser('~')
        if not os.path.lexists(usr_dir + '/.local'):
            os.mkdir(usr_dir + '/.local')
        if not os.path.lexists('{}/.local/lib'.format(usr_dir)):
            os.mkdir('{}/.local/lib'.format(usr_dir))
        if not os.path.lexists('{}/.local/lib/python{:>.1f}'.format(usr_dir, py_version*0.1)):
            os.mkdir('{}/.local/lib/python{:>.1f}'.format(usr_dir, py_version*0.1))
            os.mkdir('{}/.local/lib/python{:>.1f}/site-packages'.format(usr_dir, py_version*0.1))
        if not os.path.lexists('{}/.local/lib/python{:>.1f}/site-packages'.format(usr_dir, py_version*0.1)):
            os.mkdir('{}/.local/lib/python{:>.1f}/site-packages'.format(usr_dir, py_version*0.1))
        install_dir = '{}/.local/lib/python{:>.1f}/site-packages/hikyuu'.format(usr_dir, py_version*0.1)
    os.system("xmake install -o " + install_dir)
    sys.exit(0)

if sys.argv[-1] == 'bdist_wheel':
    if py_version_changed:
        os.system("xmake f -c")
        os.system("xmake")
    sys.argv.append("--python-tag")
    sys.argv.append("cp{}".format(py_version))
    sys.argv.append("-p")
    if current_plat == 'win32' and current_bits == 64:
        sys.argv.append("win-amd64")
    elif current_plat == 'win32' and current_bits == 32:
        sys.argv.append("win32")
    elif current_plat == 'linux' and current_bits == 64:
        sys.argv.append("manylinux1_x86_64")
    elif current_plat == 'linux' and current_bits == 32:
        sys.argv.append("manylinux1_i386")
    else:
        print("*********尚未实现*******")
        

if os.path.lexists('Hikyuu.egg-info'):
    shutil.rmtree('Hikyuu.egg-info')

if os.path.lexists('build/lib'):
    shutil.rmtree('build/lib')
    
for bdist in os.listdir('build'):
    if len(bdist) >= 5 and bdist[:5] == 'bdist' and os.path.lexists(bdist):
        shutil.rmtree(bdist)

for x in os.listdir('hikyuu'):
    if x[:12] == 'boost_python':
        if x[12:14] != str(py_version):
            os.remove('hikyuu/{}'.format(x))

hku_version = ''
with open('xmake.lua', 'r') as f:
    for line in f:
        if len(line) > 15 and line[:11] == 'set_version':
            pre_pos = line.find('"') + 1
            end_pos = line.find('"', pre_pos)
            hku_version = line[pre_pos : end_pos]
            break
            
if not hku_version:
    print("无法在 xmake.lua 中找到 set_version 语句，获取版本号失败！")
    exit(0)

print('current hikyuu version:', hku_version)

hku_name = "Hikyuu"
#hku_version = "1.0.9"
hku_author = "fasiondog"
hku_author_email = "fasiondog@sina.com"

hku_license = "MIT"
hku_keywords = ("quant", "trade", "System Trading", "backtester", "量化", "程序化交易", "量化交易", "系统交易")
hku_platforms = "Independant"
hku_url = "http://hikyuu.org/"

hku_description = "Hikyuu Quant Framework for System Trading Analysis and backtester"
with open("README.rst", encoding='utf-8') as f:
    hku_long_description = f.read()

hku_data_files = []

packages = ['hikyuu',
            'hikyuu/config',
            'hikyuu/data',
            'hikyuu/data/mysql_upgrade',
            'hikyuu/data/sqlite_upgrade',
            'hikyuu/data_driver',
            'hikyuu/examples',
            'hikyuu/gui',
            'hikyuu/gui/data',
            'hikyuu/indicator',
            'hikyuu/interactive',
            'hikyuu/interactive/drawplot',
            #'hikyuu/test',
            'hikyuu/tools',
            'hikyuu/trade_instance',
            'hikyuu/trade_manage',
            'hikyuu/trade_sys',
            'hikyuu/util',
           ]

setup(
        name = hku_name,
        version = hku_version,
        description = hku_description,
        long_description = hku_long_description,
        author = hku_author,
        author_email = hku_author_email,

        license = hku_license,
        keywords = hku_keywords,
        platforms = hku_platforms,
        url = hku_url,

        packages = packages, #find_packages(),
        zip_safe = False,
        include_package_data=True,
        package_data = {
            '': ['*.rst', '*.pyd', '*.ini', '*.sql', '*.properties', '*.xml', 'LICENSE.txt', '*.dll', '*.exe', '*.ico'],
        },
        
        data_files = hku_data_files,
        
        classifiers=[
            # How mature is this project? Common values are
            #   3 - Alpha
            #   4 - Beta
            #   5 - Production/Stable
            'Development Status :: 3 - Alpha',

            # Indicate who your project is intended for
            'Intended Audience :: Developers',
            'Topic :: Software Development :: Libraries :: Python Modules',
            'Topic :: Office/Business :: Financial',
            'Topic :: Office/Business :: Financial :: Investment',
            'Topic :: Scientific/Engineering :: Mathematics',

            # Pick your license as you wish (should match "license" above)
             'License :: OSI Approved :: MIT License',

            'Operating System :: Microsoft :: Windows',
             
            # Specify the Python versions you support here. In particular, ensure
            # that you indicate whether you support Python 2, Python 3 or both.
            'Programming Language :: Python :: 3',
            'Programming Language :: Python :: 3.5',
        ],

        entry_points={
            'gui_scripts': ['HikyuuTDX=hikyuu.gui.HikyuuTDX:start',],
            'console_scripts': ['importdata=hikyuu.gui.importdata:main',]
        },

        install_requires=[
            'matplotlib>=1.5.0',
            'pandas>=0.17.1',
            #'tushare>=0.8.2',
            'pytdx',
            'pyperclip>=1.5.27',
            'pyecharts>=0.2.7',
            'pyecharts_snapshot',
            'PyQt5',
        ],        
)