Merge pull request #117 from dritter/vi_customizations

Make the vi_mode segement more customizable.

2 files changed, 10 insertions, 2 deletions

diff --git a/README.md b/README.md
index 958889f2..e5c506ae 100644
--- a/README.md
+++ b/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"
diff --git a/powerlevel9k.zsh-theme b/powerlevel9k.zsh-theme
index 28e65a25..ad7ce272 100755
--- a/powerlevel9k.zsh-theme
+++ b/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
 }