Merge 122baa49fa into 11034217a7

lib/git.zsh

@@ -4,6 +4,10 @@ function git_prompt_info() {
   echo "$ZSH_THEME_GIT_PROMPT_PREFIX${ref#refs/heads/}$(parse_git_dirty)$ZSH_THEME_GIT_PROMPT_SUFFIX"
 }
 
+# returns root of current repo
+function git_get_root() {
+  echo "$(git rev-parse --show-toplevel 2> /dev/null)"
+}
 
 # Checks if working tree is dirty
 parse_git_dirty() {

plugins/git/git.plugin.zsh

@@ -60,6 +60,11 @@ function current_repository() {
   echo $(git remote -v | cut -d':' -f 2)
 }
 
+# change workdir to repo's root (or do `cd` with no arguments if not in a repo)
+function toroot() {
+  builtin cd "`git_get_root`"
+}
+
 # these aliases take advantage of the previous function
 alias ggpull='git pull origin $(current_branch)'
 compdef ggpull=git

plugins/virtualenvwrapper/virtualenvwrapper.plugin.zsh

@@ -8,32 +8,38 @@ for wrapsource in "/usr/bin/virtualenvwrapper.sh" "/usr/local/bin/virtualenvwrap
       # 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
-          PROJECT_ROOT=`git rev-parse --show-toplevel 2> /dev/null`
-          if (( $? == 0 )); then
-              # Check for virtualenv name override
-              ENV_NAME=`basename "$PROJECT_ROOT"`
-              if [[ -f "$PROJECT_ROOT/.venv" ]]; then
-                  ENV_NAME=`cat "$PROJECT_ROOT/.venv"`
+      function _workon_cwd {
+        # Check that this is a Git repo and get its root
+        local repo_root=`git_get_root`
+        if [[ -n "$repo_root" ]]; then
+          local 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"
+          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
-          elif [ $CD_VIRTUAL_ENV ]; then
+        else
           # We've just left the repo, deactivate the environment
           # Note: this only happens if the virtualenv was activated automatically
+          _deactivate
+        fi
+      }
+
+      function _deactivate() {
+        if [[ -n $CD_VIRTUAL_ENV ]]; then
           deactivate && unset CD_VIRTUAL_ENV
         fi
-          unset PROJECT_ROOT
       }
 
       # New cd function that does the virtualenv magic
       function cd {
-          builtin cd "$@" && workon_cwd
+        builtin cd "$@" && _workon_cwd
       }
     fi
 
@@ -44,3 +50,5 @@ done
 if [ $WRAPPER_FOUND -eq 0 ] ; then
   print "zsh virtualenvwrapper plugin: Couldn't activate virtualenvwrapper. Please run \`pip install virtualenvwrapper\`."
 fi
+
+unset WRAPPER_FOUND