WRAPPER_FOUND=0
for wrapsource in "/usr/local/bin/virtualenvwrapper.sh" "/etc/bash_completion.d/virtualenvwrapper" ; do
  if [[ -e $wrapsource ]] ; then
    WRAPPER_FOUND=1
    source $wrapsource

    if [[ ! $DISABLE_VENV_CD -eq 1 ]]; then
      # Automatically activate Git projects' virtual environments based on the
      # directory name of the project. Virtual environment name can be overridden
      # by placing a .venv file in the project root with a virtualenv name in it
      function _workon_cwd {
        # Check that this is a Git repo and get its root
        REPO_ROOT=`git_get_root`
        if [[ -n "$REPO_ROOT" ]]; then
          ENV_NAME=`basename "$REPO_ROOT"`
          if [[ -f "$REPO_ROOT/.venv" ]]; then
            ENV_NAME=`cat "$REPO_ROOT/.venv"`
          fi
          # Activate the environment only if it is not already active
          if [[ "$VIRTUAL_ENV" != "$WORKON_HOME/$ENV_NAME" ]]; then
            if [[ -e "$WORKON_HOME/$ENV_NAME/bin/activate" ]]; then
              workon "$ENV_NAME" && export CD_VIRTUAL_ENV="$ENV_NAME"
            else
              _deactivate
            fi
          fi
        else
          # We've just left the repo, deactivate the environment
          # Note: this only happens if the virtualenv was activated automatically
          _deactivate
        fi
        unset REPO_ROOT
        unset ENV_NAME
      }

      function _deactivate() {
        if [[ -n $CD_VIRTUAL_ENV ]]; then
          deactivate && unset CD_VIRTUAL_ENV
        fi
      }

      # New cd function that does the virtualenv magic
      function cd {
        builtin cd "$@" && _workon_cwd
      }
    fi

    break
  fi
done

if [ $WRAPPER_FOUND -eq 0 ] ; then
  print "zsh virtualenvwrapper plugin: Couldn't activate virtualenvwrapper. Please run \`pip install virtualenvwrapper\`."
fi

unset WRAPPER_FOUND