Prevent pollution of the global namespace & properly quote variables

gitprompt.sh

@@ -1,107 +1,85 @@
-if [ "x$__GIT_PROMPT_DIR" == "x" ]
-then
-  __GIT_PROMPT_DIR=~/.bash
-fi
 
-# Colors
+function git_prompt_config() {
-# Reset
+  # Colors
-ResetColor="\[\033[0m\]"       # Text Reset
+  local ResetColor="\[\033[0m\]"
-
+  local BoldGreen="\[\033[1;32m\]"
-# Regular Colors
+  local BoldBlue="\[\033[1;34m\]"
-Red="\[\033[0;31m\]"          # Red
+  local Magenta="\[\033[1;95m\]"
-Yellow="\[\033[0;33m\]"       # Yellow
+  local Yellow="\[\033[1;33m\]"
-Blue="\[\033[0;34m\]"         # Blue
+  local White="\[\033[37m\]"
-WHITE='\[\033[37m\]'
+  local Blue="\[\033[0;34m\]"
-
+  local Red="\[\033[0;31m\]"
-# Bold
+
-BGreen="\[\033[1;32m\]"       # Green
+  # Default values for the appearance of the prompt. Configure at will.
-
+  GIT_PROMPT_PREFIX="["
-# High Intensty
+  GIT_PROMPT_SUFFIX="]"
-IBlack="\[\033[0;90m\]"       # Black
+  GIT_PROMPT_SEPARATOR="|"
-
+  GIT_PROMPT_BRANCH="${Magenta}"
-# Bold High Intensty
+  GIT_PROMPT_STAGED="${Red}● "
-Magenta="\[\033[1;95m\]"     # Purple
+  GIT_PROMPT_CONFLICTS="${Red}✖ "
-
+  GIT_PROMPT_CHANGED="${Blue}✚ "
-# Various variables you might want for your PS1 prompt instead
+  GIT_PROMPT_REMOTE=" "
-Time12a="\@"
+  GIT_PROMPT_UNTRACKED="…"
-PathShort="\w"
+  GIT_PROMPT_CLEAN="${BoldGreen}✔"
-
+
-# Default values for the appearance of the prompt. Configure at will.
+  PROMPT_START="${Yellow}\w${ResetColor}"
-GIT_PROMPT_PREFIX="["
+  PROMPT_END=" \n${White}\@${ResetColor} $ "
-GIT_PROMPT_SUFFIX="]"
-GIT_PROMPT_SEPARATOR="|"
-GIT_PROMPT_BRANCH="${Magenta}"
-GIT_PROMPT_STAGED="${Red}● "
-GIT_PROMPT_CONFLICTS="${Red}✖ "
-GIT_PROMPT_CHANGED="${Blue}✚ "
-GIT_PROMPT_REMOTE=" "
-GIT_PROMPT_UNTRACKED="…"
-GIT_PROMPT_CLEAN="${BGreen}✔"
-
-PROMPT_START="$Yellow$PathShort$ResetColor"
-PROMPT_END=" \n$WHITE$Time12a$ResetColor $ "
-
-
-function update_current_git_vars() {
-    unset __CURRENT_GIT_STATUS
-    local gitstatus="${__GIT_PROMPT_DIR}/gitstatus.py"
-    
-    _GIT_STATUS=$(python $gitstatus)
-    __CURRENT_GIT_STATUS=($_GIT_STATUS)
-	GIT_BRANCH=${__CURRENT_GIT_STATUS[0]}
-	GIT_REMOTE=${__CURRENT_GIT_STATUS[1]}
-    if [[ "." == "$GIT_REMOTE" ]]; then
-		unset GIT_REMOTE
-	fi
-	GIT_STAGED=${__CURRENT_GIT_STATUS[2]}
-	GIT_CONFLICTS=${__CURRENT_GIT_STATUS[3]}
-	GIT_CHANGED=${__CURRENT_GIT_STATUS[4]}
-	GIT_UNTRACKED=${__CURRENT_GIT_STATUS[5]}
-	GIT_CLEAN=${__CURRENT_GIT_STATUS[6]}
 }
 
 function setGitPrompt() {
-	update_current_git_vars
+  local ResetColor="\[\033[0m\]"
-	set_virtualenv
+
-
+  local -a GitStatus
-	if [ -n "$__CURRENT_GIT_STATUS" ]; then
+  GitStatus=($("${__GIT_PROMPT_DIR:-${HOME}/.bash}/gitstatus.py" 2>/dev/null))
-	  STATUS=" $GIT_PROMPT_PREFIX$GIT_PROMPT_BRANCH$GIT_BRANCH$ResetColor"
+
-
+  local GIT_PROMPT_PREFIX
-	  if [ -n "$GIT_REMOTE" ]; then
+  local GIT_PROMPT_SUFFIX
-		  STATUS="$STATUS$GIT_PROMPT_REMOTE$GIT_REMOTE$ResetColor"
+  local GIT_PROMPT_SEPARATOR
-	  fi
+  local GIT_PROMPT_BRANCH
-
+  local GIT_PROMPT_STAGED
-	  STATUS="$STATUS$GIT_PROMPT_SEPARATOR"
+  local GIT_PROMPT_CONFLICTS
-	  if [ "$GIT_STAGED" -ne "0" ]; then
+  local GIT_PROMPT_CHANGED
-		  STATUS="$STATUS$GIT_PROMPT_STAGED$GIT_STAGED$ResetColor"
+  local GIT_PROMPT_REMOTE
-	  fi
+  local GIT_PROMPT_UNTRACKED
-
+  local GIT_PROMPT_CLEAN
-	  if [ "$GIT_CONFLICTS" -ne "0" ]; then
+  local PROMPT_START
-		  STATUS="$STATUS$GIT_PROMPT_CONFLICTS$GIT_CONFLICTS$ResetColor"
+  local PROMPT_END
-	  fi
+  git_prompt_config
-	  if [ "$GIT_CHANGED" -ne "0" ]; then
+
-		  STATUS="$STATUS$GIT_PROMPT_CHANGED$GIT_CHANGED$ResetColor"
+  if [[ -n "${GitStatus}" ]]; then
-	  fi
+    local Status="${Status} ${GIT_PROMPT_PREFIX}"
-	  if [ "$GIT_UNTRACKED" -ne "0" ]; then
+
-		  STATUS="$STATUS$GIT_PROMPT_UNTRACKED$GIT_UNTRACKED$ResetColor"
+    Status="${Status}${GIT_PROMPT_BRANCH}${GitStatus[0]}${ResetColor}"
-	  fi
+    if [[ "${GitStatus[1]}" != "." ]]; then
-	  if [ "$GIT_CLEAN" -eq "1" ]; then
+      Status="${Status}${GIT_PROMPT_REMOTE}${GitStatus[1]}${ResetColor}"
-		  STATUS="$STATUS$GIT_PROMPT_CLEAN"
+    fi
-	  fi
+
-	  STATUS="$STATUS$ResetColor$GIT_PROMPT_SUFFIX"
+    Status="${Status}${GIT_PROMPT_SEPARATOR}"
-
+    if [[ "${GitStatus[2]}" -ne "0" ]]; then
-	  PS1="$PYTHON_VIRTUALENV$PROMPT_START$STATUS$PROMPT_END"
+      Status="${Status}${GIT_PROMPT_STAGED}${GitStatus[2]}${ResetColor}"
-	else
+    fi
-	  PS1="$PROMPT_START$PROMPT_END"
+
-	fi
+    if [[ "${GitStatus[3]}" -ne "0" ]]; then
-}
+      Status="${Status}${GIT_PROMPT_CONFLICTS}${GitStatus[3]}${ResetColor}"
-
+    fi
-# Determine active Python virtualenv details.
+    if [[ "${GitStatus[4]}" -ne "0" ]]; then
-function set_virtualenv () {
+      Status="${Status}${GIT_PROMPT_CHANGED}${GitStatus[4]}${ResetColor}"
-  if test -z "$VIRTUAL_ENV" ; then
+    fi
-      PYTHON_VIRTUALENV=""
+    if [[ "${GitStatus[5]}" -ne "0" ]]; then
+      Status="${Status}${GIT_PROMPT_UNTRACKED}${GitStatus[5]}${ResetColor}"
+    fi
+    if [[ "${GitStatus[6]}" -eq "1" ]]; then
+      Status="${Status}${GIT_PROMPT_CLEAN}${ResetColor}"
+    fi
+    Status="${Status}${GIT_PROMPT_SUFFIX}"
+
+    PS1="${PROMPT_START}${Status}${PROMPT_END}"
+    if [[ -n "${VIRTUAL_ENV}" ]]; then
+      local Blue="\[\033[1;34m\]"
+      PS1="${Blue}($(basename "${VIRTUAL_ENV}"))${ResetColor} ${PS1}"
+    fi
   else
-      PYTHON_VIRTUALENV="${BLUE}(`basename \"$VIRTUAL_ENV\"`)${ResetColor} "
+    PS1="${PROMPT_START}${PROMPT_END}"
   fi
 }