# Copyright (c) 2017 VisualDL Authors. All Rights Reserve.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# =======================================================================

from __future__ import absolute_import

import os
import sys
from sys import platform
from distutils.spawn import find_executable
from distutils import log
import setuptools.command.build_py
import setuptools
from setuptools import setup, Extension
import subprocess

TOP_DIR = os.path.realpath(os.path.dirname(__file__))
PYTHON_SDK_DIR = os.path.join(TOP_DIR, 'visualdl/python')
BUILD_DIR = os.path.join(TOP_DIR, 'build')
MODE = os.environ.get('VS_BUILD_MODE', 'RELEASE')


def read(name):
    return open(os.path.join(TOP_DIR, name)).read()


def readlines(name):
    return read(name).split('\n')


VERSION_NUMBER = read('VERSION_NUMBER')
LICENSE = readlines('LICENSE')[0].strip()

# use memcache to reduce disk read frequency.
install_requires = ['Flask', 'numpy', 'Pillow', 'protobuf >= 3.1.0', 'scipy']
execute_requires = ['npm', 'node', 'bash', 'cmake', 'unzip']
if platform == "win32":
    execute_requires = ['node', 'powershell', 'cmake']


def die(msg):
    log.error(msg)
    sys.exit(1)


def CHECK(cond, msg):
    if not cond:
        die(msg)


for exe in execute_requires:
    CHECK(find_executable(exe), "{} should be installed.".format(exe))


class BaseCommand(setuptools.Command):
    user_options = []

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass


class build_py(setuptools.command.build_py.build_py):
    def run(self):
        cmd = ['bash', 'scripts/build.sh']
        if platform == "win32":
            cmd = ['powershell', '-NoProfile', './scripts/build.ps1']
        env = dict(os.environ)
        if MODE == "travis-CI":
            cmd.append('travis-CI')
        if sys.version_info[0] >= 3:
            env["WITH_PYTHON3"] = "ON"
        subprocess.check_call(cmd, env=env)
        return setuptools.command.build_py.build_py.run(self)


cmdclass = {
    'build_py': build_py,
}

packages = [
    'visualdl',
    'visualdl.python',
    'visualdl.server',
    'visualdl.server.mock',
    'visualdl.server.model',
    'visualdl.server.model.onnx',
    'visualdl.server.model.paddle',
]

libraries = ['core.so']
if platform == 'win32':
    libraries = ['core.pyd', 'libprotobuf.dll']

scripts = ['visualdl/server/visualdl', 'demo/vdl_create_scratch_log']
if platform == 'win32':
    scripts.append('visualdl/server/visualDL.bat')

def frontend_package_files(directory):
    paths = []
    for (path, directories, filenames) in os.walk(directory):
        for filename in filenames:
            paths.append(os.path.join(path, filename))
    return paths

setup(
    name="visualdl",
    version=VERSION_NUMBER,
    author="PaddlePaddle and Echarts team",
    description="Visualize Deep Learning",
    license=LICENSE,
    keywords="visualization deeplearning",
    long_description=read('README.md'),
    install_requires=install_requires,
    package_data={
        'visualdl.server':
        [s[len('visualdl/server/'):] for s in frontend_package_files('visualdl/server/dist')],
        'visualdl':
        libraries,
        'visualdl.python':
        libraries + ['dog.jpg', 'testing.wav']
    },
    packages=packages,
    ext_modules=[Extension('_foo', ['stub.cc'])],
    scripts=scripts,
    cmdclass=cmdclass)