# make sure virtualenvwrapper is enabled if available

cite about-plugin
about-plugin 'pyenv-virtualenvwrapper helper functions'

export PYENV_VIRTUALENVWRAPPER_PREFER_PYVENV="true"

[[ `which pyenv` ]] && eval "$(pyenv init -)"
[[ `which pyenv-virtualenvwrapper` ]] && eval "$(pyenv virtualenvwrapper)"

# Activate autoenv
source /usr/local/opt/autoenv/activate.sh

function mkpvenv {
  about 'create a new virtualenv for this directory'
  group 'pyenv-virtualenv'

  eval "touch .env"
  eval "echo \"#!/bin/bash\" >> .env"
  eval "echo \"eval \"wopvenv\"\" >> .env"
  cwd=`basename \`pwd\``
  mkvirtualenv --distribute $cwd
}

function mkpvbranch {
  about 'create a new virtualenv for the current branch'
  group 'pyenv-virtualenv'

  mkvirtualenv --distribute "$(basename `pwd`)@$SCM_BRANCH"
}

function wopvbranch {
  about 'sets workon branch'
  group 'pyenv-virtualenv'

  workon "$(basename `pwd`)@$SCM_BRANCH"
}

function wopvenv {
  about 'works on the virtualenv for this directory'
  group 'virtualenv'

  workon "$(basename `pwd`)"
}

function rmpvenv {
  about 'removes virtualenv for this directory'
  group 'virtualenv'

  eval "deactivate"
  rmvirtualenv "$(basename `pwd`)"
  eval "rm .env"
}

function rmpvenvbranch {
  about 'removes virtualenv for this directory'
  group 'virtualenv'

  eval "deactivate"
  rmvirtualenv "$(basename `pwd`)@$SCM_BRANCH"
}