linux下shell显示git当前分支

function git-branch-name {
  git symbolic-ref HEAD 2>/dev/null | cut -d"/" -f 3
}

function git-branch-prompt {
  local branch=`git-branch-name`
  if [ $branch ]; then printf " [%s]" $branch; fi
}

PS1="u@h [33[0;36m]W[33[0m][33[0;32m]$(git-branch-prompt)[33[0m] $ "