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\/github.com\/youtube\/vitess/}}"
# VTTOP sanity check
if [[ "$VTTOP" == "${VTTOP/\/src\/github.com\/youtube\/vitess/}" ]]; then
  echo "WARNING: VTTOP($VTTOP) does not contain src/github.com/youtube/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_config)" == "" ]]; then
     echo "WARNING: VT_MYSQL_ROOT unset because mysql_config not found. Did you install a client dev package?"
  else
    export VT_MYSQL_ROOT=$(dirname $(dirname $(which mysql_config)))
  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"