from __future__ import print_function

import os
import subprocess

from setuptools import setup, find_packages
import setuptools.command.install as _install

subprocess.check_call(["../../build-webui.sh"])
datafiles = [(root, [os.path.join(root, f) for f in files])
    for root, dirs, files in os.walk("./webui")]

class install(_install.install):
  def run(self):
    subprocess.check_call(["../../build.sh"])
    # Calling _install.install.run(self) does not fetch required packages and
    # instead performs an old-style install. See command/install.py in
    # setuptools. So, calling do_egg_install() manually here.
    self.do_egg_install()

setup(name="ray",
      version="0.0.1",
      packages=find_packages(),
      package_data={"common": ["thirdparty/redis/src/redis-server"],
                    "plasma": ["build/plasma_store",
                               "build/plasma_manager",
                               "lib/python/libplasma.so"],
                    "photon": ["build/photon_scheduler",
                               "photon/libphoton.so"],
                    "global_scheduler": ["build/global_scheduler"]},
      data_files=datafiles,
      cmdclass={"install": install},
      install_requires=["numpy",
                        "funcsigs",
                        "colorama",
                        "psutil",
                        "redis",
                        "cloudpickle"],
      include_package_data=True,
      zip_safe=False,
      license="Apache 2.0")