#!/bin/bash

# Copyright 2017 Google Inc.
# 
# 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.

# This script downloads and installs the grpc library, for
# go and python, in the root of the image. It assumes we're running
# as root in the image.
set -ex

# Import prepend_path function.
dir="$(dirname "${BASH_SOURCE[0]}")"
source "${dir}/../tools/shell_functions.inc"
if [ $? -ne 0 ]; then
  echo "failed to load ../tools/shell_functions.inc"
  return 1
fi

# grpc_dist can be empty, in which case we just install to the default paths
grpc_dist="$1"
if [ -n "$grpc_dist" ]; then
  cd $grpc_dist
fi

# Python requires a very recent version of virtualenv.
# We also require a recent version of pip, as we use it to
# upgrade the other tools.
# For instance, setuptools doesn't work with pip 6.0:
# https://github.com/pypa/setuptools/issues/945
# (and setuptools is used by grpc install).
if [ -n "$grpc_dist" ]; then
  # Create a virtualenv, which also creates a virualenv-boxed pip.
  # Update both pip and virtualenv.
  $VIRTUALENV -v $grpc_dist/usr/local
  PIP=$grpc_dist/usr/local/bin/pip
  $PIP install --upgrade pip
  $PIP install --upgrade --ignore-installed virtualenv
else
  PIP=pip
  $PIP install --upgrade pip
  # System wide installations require an explicit upgrade of
  # certain gRPC Python dependencies e.g. "six" on Debian Jessie.
  $PIP install --upgrade --ignore-installed six
fi

# clone the repository, setup the submodules
git clone https://github.com/grpc/grpc.git
cd grpc
git checkout v1.0.0
git submodule update --init

# OSX specific setting + dependencies
if [ `uname -s` == "Darwin" ]; then
  export GRPC_PYTHON_BUILD_WITH_CYTHON=1
  $PIP install Cython

  # Work-around macOS Sierra blocker, see: https://github.com/youtube/vitess/issues/2115
  # TODO(mberlin): Remove this when the underlying issue is fixed and available
  #                in the gRPC version used by Vitess.
  #                See: https://github.com/google/protobuf/issues/2182
  export CPPFLAGS="-Wno-deprecated-declarations"
fi

# build everything
make

# install protobuf side (it was already built by the 'make' earlier)
cd third_party/protobuf
if [ -n "$grpc_dist" ]; then
  make install prefix=$grpc_dist/usr/local
else
  make install
fi

# now install grpc itself
cd ../..
if [ -n "$grpc_dist" ]; then
  make install prefix=$grpc_dist/usr/local

  # Add bin directory to the path such that gRPC python won't complain that
  # it cannot find "grpc_python_plugin".
  export PATH=$(prepend_path $PATH $grpc_dist/usr/local/bin)
else
  make install
fi

# Install gRPC python libraries from PyPI.
# Dependencies like protobuf python will be installed automatically.
grpcio_ver=1.0.4
$PIP install --upgrade grpcio==$grpcio_ver

# Build PHP extension, only if requested.
if [ -n "$INSTALL_GRPC_PHP" ]; then
  echo "Building gRPC PHP extension..."
  cd src/php/ext/grpc
  if [[ "$TRAVIS" == "true" ]]; then
    # On Travis, we get an old glibc version that doesn't yet
    # have clock_*() functions built in, so we need -lrt.
    # Temporarily turn off --as-needed because the lib doesn't
    # request rt, since it assumes a newer glibc.
    export LDFLAGS="-Wl,--no-as-needed -lrt -Wl,--as-needed"
  fi
  phpize
  if ! ./configure --enable-grpc=$grpc_dist/usr/local; then
    cat config.log
    echo "Failed to configure build for gRPC PHP extension."
    exit 1
  fi
  make
  mkdir -p $INSTALL_GRPC_PHP
  mv modules/grpc.so $INSTALL_GRPC_PHP
fi