Merge pull request #117 from dritter/vi_customizations

Make the vi_mode segement more customizable.

README.md

@@ -233,6 +233,12 @@ VI-Mode, you need to configure it separately in your `~/.zshrc`:
     # make it more responsive
     export KEYTIMEOUT=1
 
+If you want to display another string than "NORMAL" or "INSERT" in command or
+insert-mode, you can do that by setting environment variables like:
+
+    POWERLEVEL9K_VI_INSERT_MODE_STRING="INS"
+    POWERLEVEL9K_VI_COMMAND_MODE_STRING="CMD"
+
 #### Unit Test Ratios
 
 The `symfony2_tests` and `rspec_stats` segments both show a ratio of "real"

powerlevel9k.zsh-theme

@@ -871,13 +871,15 @@ prompt_todo() {
 }
 
 # Vi Mode: show editing mode (NORMAL|INSERT)
+set_default POWERLEVEL9K_VI_INSERT_MODE_STRING "INSERT"
+set_default POWERLEVEL9K_VI_COMMAND_MODE_STRING "NORMAL"
 prompt_vi_mode() {
   case ${KEYMAP} in
     main|viins)
-      "$1_prompt_segment" "$0_INSERT" "$DEFAULT_COLOR" "blue" "INSERT"
+      "$1_prompt_segment" "$0_INSERT" "$DEFAULT_COLOR" "blue" "$POWERLEVEL9K_VI_INSERT_MODE_STRING"
     ;;
     vicmd)
-      "$1_prompt_segment" "$0_NORMAL" "$DEFAULT_COLOR" "default" "NORMAL"
+      "$1_prompt_segment" "$0_NORMAL" "$DEFAULT_COLOR" "default" "$POWERLEVEL9K_VI_COMMAND_MODE_STRING"
     ;;
   esac
 }