#!/usr/bin/env bash

ROOT_DIR=$PWD
VENDOR_DIR="$ROOT_DIR/vendor"
BAZEL_DIR=`cd $VENDOR_DIR; bazel info output_base`
TENSORFLOW_DIR="$BAZEL_DIR/external/org_tensorflow"

# tensorflow: initialize bazel directories
eval "cd $VENDOR_DIR; bazel build @org_tensorflow//tensorflow:all_files"

# tensorflow: do not use interactive configure so can be installed automatically
GEN_GIT_SOURCE=tensorflow/tools/git/gen_git_source.py
eval "cd $TENSORFLOW_DIR; chmod a+x ${GEN_GIT_SOURCE}"
eval "cd $TENSORFLOW_DIR; python ${GEN_GIT_SOURCE} --configure ${TENSORFLOW_DIR}"
# tensorflow: run interactive configure instead of automatic
# if [ ! -d "$TENSORFLOW_DIR/bazel-bin" ]; then
#   eval "cd $TENSORFLOW_DIR; $TENSORFLOW_DIR/configure"
# fi

# tensorflow: build
eval "cd $TENSORFLOW_DIR; bazel build -c opt //tensorflow:libtensorflow_cc.so"
eval "cd $VENDOR_DIR; rm -rf bazel-*"

# protobuf: copy files
mkdir -p $VENDOR_DIR/protobuf/Headers
cp -rf $TENSORFLOW_DIR/bazel-org_tensorflow/external/protobuf/src/ $VENDOR_DIR/protobuf/Headers/
cp -f $TENSORFLOW_DIR/bazel-out/host/bin/external/protobuf/libprotobuf.a $VENDOR_DIR/protobuf/libprotobuf.a

# eigen: copy files
mkdir -p $VENDOR_DIR/eigen/Headers
cp -rf $TENSORFLOW_DIR/bazel-org_tensorflow/external/eigen_archive/ $VENDOR_DIR/eigen/Headers/
mkdir -p $VENDOR_DIR/eigen/Generated
cp -rf $TENSORFLOW_DIR/bazel-org_tensorflow/third_party/ $VENDOR_DIR/eigen/Generated/third_party/

# tensorflow: copy files
mkdir -p $VENDOR_DIR/tensorflow/Headers/tensorflow
cp -rf $TENSORFLOW_DIR/bazel-org_tensorflow/tensorflow/ $VENDOR_DIR/tensorflow/Headers/tensorflow/
mkdir -p $VENDOR_DIR/tensorflow/Generated
cp -rf $TENSORFLOW_DIR/bazel-genfiles/tensorflow/ $VENDOR_DIR/tensorflow/Generated/tensorflow/

# tensorflow: rename libtensorflow_cc.so to libtensorflow_cc.dylib to for xcode debugging
TENSORFLOW_BAZEL_DIR="bazel-out/local-opt/bin/tensorflow"
TENSORFLOW_LIB_DIR="$VENDOR_DIR/tensorflow/$TENSORFLOW_BAZEL_DIR"
mkdir -p $TENSORFLOW_LIB_DIR
cp -f $TENSORFLOW_DIR/$TENSORFLOW_BAZEL_DIR/libtensorflow_cc.so $TENSORFLOW_LIB_DIR/libtensorflow_cc.so
ln -s $TENSORFLOW_LIB_DIR/libtensorflow_cc.so $VENDOR_DIR/tensorflow/libtensorflow_cc.dylib > /dev/null 2>&1 || :

# # update the hardcoded library path in bazel
# # TODO: figure out a way to do this that 1) doesn't require sudo and 2) handles relative paths
sudo install_name_tool -id $TENSORFLOW_LIB_DIR/libtensorflow_cc.so $TENSORFLOW_LIB_DIR/libtensorflow_cc.so > /dev/null 2>&1 || :