dev.env

# 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.

# 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

export VTTOP=$(pwd)
export VTROOT="${VTROOT:-${VTTOP/\/src\/vitess.io\/vitess/}}"
# VTTOP sanity check
if [[ "$VTTOP" == "${VTTOP/\/src\/vitess.io\/vitess/}" ]]; then
  echo "WARNING: VTTOP($VTTOP) does not contain src/vitess.io/vitess"
fi
export GOTOP=$VTTOP/go
export PYTOP=$VTTOP/py

export VTDATAROOT="${VTDATAROOT:-${VTROOT}/vtdataroot}"
mkdir -p $VTDATAROOT

export VTPORTSTART=15000

for pypath in $(find $VTROOT/dist -name site-packages -or -name dist-packages | grep -v src/python/grpcio/.tox/py27/lib/python2.7/site-packages)
do
  export PYTHONPATH=$(prepend_path "$PYTHONPATH" "$pypath")
done

export PYTHONPATH=$(prepend_path "$PYTHONPATH" "$VTROOT/py-vtdb")
export PYTHONPATH=$(prepend_path "$PYTHONPATH" "$VTTOP/test")
export PYTHONPATH=$(prepend_path "$PYTHONPATH" "$VTTOP/test/cluster/sandbox")

# Ensure bootstrap and install_grpc use python2 on systems which default to python3
command -v python2 >/dev/null && PYTHON=python2 || PYTHON=python
export PYTHON
command -v pip2 >/dev/null && PIP=pip2 || PIP=pip
export PIP
command -v virtualenv2 >/dev/null && VIRTUALENV=virtualenv2 || VIRTUALENV=virtualenv
export VIRTUALENV

selenium_dist=$VTROOT/dist/selenium
export PYTHONPATH=$(prepend_path "$PYTHONPATH" "$selenium_dist")

# Add the current GOBIN
if [ "$GOBIN" ]; then
  export PATH=$(prepend_path "$PATH" "$GOBIN")
fi

# Many tests rely on "go install" and assume GOBIN is really $VTROOT/bin.
# Make sure these take precedence.
export GOBIN=$VTROOT/bin
export PATH=$(prepend_path "$PATH" "$GOBIN")

# If we have a custom built mysql, prefer it over the default installation.
export PATH=$(prepend_path "$PATH" "$VTROOT/dist/mysql/bin")

# Add chromedriver to path for selenium tests
export PATH=$(prepend_path "$PATH" "$VTROOT/dist/chromedriver")

# Node path
export PATH=$(prepend_path "$PATH" "$VTROOT/dist/node/bin")

# GOROOT sanity
go_bin=`which go`
go_env=`go env | grep GOROOT | cut -f 2 -d\"`
if [ "$go_bin" -a "$go_bin" != "$go_env/bin/go" ]; then
  echo "WARNING: \$GOROOT may not be compatible with the used go binary"
  echo "Please make sure 'go' comes from \$GOROOT/bin"
  echo "go_env: $go_env"
  echo "go_bin: $go_bin"
fi

# mysql install location. Please set based on your environment.
# Build will not work if this is incorrect.

if [[ "$VT_MYSQL_ROOT" == "" ]]; then
  if [[ "$(which mysql)" == "" ]]; then
     echo "WARNING: VT_MYSQL_ROOT unset because mysql not found. Did you install a client package?"
  else
    export VT_MYSQL_ROOT=$(dirname $(dirname $(which mysql)))
  fi
fi

# restore MYSQL_FLAVOR, saved by bootstrap.sh
if [ -r $VTROOT/dist/MYSQL_FLAVOR ]; then
  export MYSQL_FLAVOR=`cat $VTROOT/dist/MYSQL_FLAVOR`
fi

# mysql cgo library config
if [ -z "$MYSQL_FLAVOR" ]; then
  export MYSQL_FLAVOR=MariaDB
fi
export PKG_CONFIG_PATH=$(prepend_path "$PKG_CONFIG_PATH" "$VTROOT/lib")

# needed to correctly import grpc if it's not installed globally
grpc_dist=$VTROOT/dist/grpc
if [ -f $grpc_dist/.build_finished ]; then
  export LD_LIBRARY_PATH=$(prepend_path "$LD_LIBRARY_PATH" "$grpc_dist/usr/local/lib")
fi

export GOPATH=$(prepend_path "$GOPATH" "$VTROOT")

# Useful aliases. Remove if inconvenient.
alias gt="cd $GOTOP"
alias pt="cd $PYTOP"
alias vt="cd $VTTOP"