#!/bin/bash
# usage: source ./hacking/env-setup [-q]
#    modifies environment for running Ansible from checkout

# When run using source as directed, $0 gets set to bash, so we must use $BASH_SOURCE
if [ -n "$BASH_SOURCE" ] ; then
    HACKING_DIR=`dirname $BASH_SOURCE`
elif [ $(basename $0) = "env-setup" ]; then
    HACKING_DIR=`dirname $0`
else
    HACKING_DIR="$PWD/hacking"
fi
# The below is an alternative to readlink -fn which doesn't exist on OS X
# Source: http://stackoverflow.com/a/1678636
FULL_PATH=`python -c "import os; print(os.path.realpath('$HACKING_DIR'))"`
ANSIBLE_HOME=`dirname "$FULL_PATH"`

PREFIX_PYTHONPATH="$ANSIBLE_HOME/lib"
PREFIX_PATH="$ANSIBLE_HOME/bin"
PREFIX_MANPATH="$ANSIBLE_HOME/docs/man"

[[ $PYTHONPATH != ${PREFIX_PYTHONPATH}* ]] && export PYTHONPATH=$PREFIX_PYTHONPATH:$PYTHONPATH
[[ $PATH != ${PREFIX_PATH}* ]] && export PATH=$PREFIX_PATH:$PATH
unset ANSIBLE_LIBRARY
export ANSIBLE_LIBRARY="$ANSIBLE_HOME/library:`python $HACKING_DIR/get_library.py`"
[[ $MANPATH != ${PREFIX_MANPATH}* ]] && export MANPATH=$PREFIX_MANPATH:$MANPATH

#
# Generate egg_info so that pkg_resources works
#

# Do the work in a function so we don't repeat ourselves later
gen_egg_info()
{
    python setup.py egg_info
    if [ -e $PREFIX_PYTHONPATH/ansible*.egg-info ] ; then
        rm -r $PREFIX_PYTHONPATH/ansible*.egg-info
    fi
    mv ansible*.egg-info $PREFIX_PYTHONPATH
}

# In some shells if pushd is a no-op then popd sends you to a previous
# directory in history
if [ "$ANSIBLE_HOME" != "$PWD" ] ; then
    pushd "$ANSIBLE_HOME"
    gen_egg_info
    popd
else
    gen_egg_info
fi

# Print out values unless -q is set

if [ $# -eq 0 -o "$1" != "-q" ] ; then
    echo ""
    echo "Setting up Ansible to run out of checkout..."
    echo ""
    echo "PATH=$PATH"
    echo "PYTHONPATH=$PYTHONPATH"
    echo "ANSIBLE_LIBRARY=$ANSIBLE_LIBRARY"
    echo "MANPATH=$MANPATH"
    echo ""

    echo "Remember, you may wish to specify your host file with -i"
    echo ""
    echo "Done!"
    echo ""
fi