X-Git-Url: https://git.madduck.net/etc/zsh.git/blobdiff_plain/f47f496110c9cad7ed628f28a02c73303cc521a3..1eff3f3673d73739a8ee6f32d4ee1f7b36817013:/.zsh/func/vcsh

diff --git a/.zsh/func/vcsh b/.zsh/func/vcsh
index 0959784..c3c6357 100755
--- a/.zsh/func/vcsh
+++ b/.zsh/func/vcsh
@@ -10,28 +10,36 @@
 # Source repository: http://git.madduck.net/v/etc/zsh.git
 #
 
-FGIT_BASE="$HOME/.fgits"
+local FGIT_BASE="$HOME/.fgits"
 
 if [ "${1:---help}" = '--help' ] || [ $# -gt 1 ]; then
-  echo "usage: ${0%/*} reponame" >&2
-  echo "usage: ${0%/*} -l" >&2
+  echo "usage: ${0##*/} reponame" >&2
+  echo "usage: ${0##*/} -l" >&2
   [ "$1" = '--help' ]
-  exit $?
+  return $?
 
 elif [ "$1" = '-l' ]; then
   for i in $FGIT_BASE/*.git; do
     i="${i#$FGIT_BASE/}"
     echo "${i%.git}"
   done
-  exit 0
+  return 0
 fi
 
 if [ ! -d "$FGIT_BASE/${1}.git" ]; then
   echo E: no repository found for "$1" >&2
-  exit 2
+  return 2
 fi
 
-export GIT_DIR=$FGIT_BASE/${1}.git
-git status
+old_GIT_DIR="${GIT_DIR:-}"
+old_GIT_WORK_TREE="${GIT_WORK_TREE:-}"
 
-PS1="{VCSH:$1}$PS1" $SHELL -i
+export GIT_DIR="$FGIT_BASE/${1}.git"
+export GIT_WORK_TREE="$GIT_DIR/$(git config --get core.worktree)"
+
+PS1="%S{${0##*/}:$1}%s$PS1" $SHELL -i || :
+
+GIT_DIR="$old_GIT_DIR"
+[ -z "$GIT_DIR" ] && unset GIT_DIR
+GIT_WORK_TREE="$old_GIT_WORK_TREE"
+[ -z "$GIT_WORK_TREE" ] && unset GIT_WORK_TREE