From c69d21016caa712481f3738a62ba2722d5583047 Mon Sep 17 00:00:00 2001
From: Matthew Hoener <matthewh@ca.ibm.com>
Date: Wed, 7 Feb 2018 14:56:48 -0800
Subject: [PATCH 1/2] Fix vi mode plugin

---
 powerlevel9k.zsh-theme | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/powerlevel9k.zsh-theme b/powerlevel9k.zsh-theme
index 14d7bd81..15a0b68a 100755
--- a/powerlevel9k.zsh-theme
+++ b/powerlevel9k.zsh-theme
@@ -1522,6 +1522,11 @@ NEWLINE='
   [[ $POWERLEVEL9K_PROMPT_ADD_NEWLINE == true ]] && PROMPT="$NEWLINE$PROMPT"
 }
 
+zle-keymap-select () {
+	zle reset-prompt
+	zle -R
+}
+
 prompt_powerlevel9k_setup() {
   # The value below was set to better support 32-bit CPUs.
   # It's the maximum _signed_ integer value on 32-bit CPUs.
@@ -1590,6 +1595,8 @@ prompt_powerlevel9k_setup() {
   # prepare prompts
   add-zsh-hook precmd powerlevel9k_prepare_prompts
   add-zsh-hook preexec powerlevel9k_preexec
+
+  zle -N zle-keymap-select
 }
 
 prompt_powerlevel9k_teardown() {

From 162caba19cdc51ea52eb5fabd5d18f03cb20e467 Mon Sep 17 00:00:00 2001
From: Matthew Hoener <matthewh@ca.ibm.com>
Date: Wed, 7 Feb 2018 15:15:02 -0800
Subject: [PATCH 2/2] Don't show segment if POWERLEVEL9K_VI_INSERT_MODE_STRING
 == ''

---
 README.md              | 2 ++
 powerlevel9k.zsh-theme | 1 +
 2 files changed, 3 insertions(+)

diff --git a/README.md b/README.md
index 46d8ab61..fdb2f5d7 100644
--- a/README.md
+++ b/README.md
@@ -621,6 +621,8 @@ you are using the [ZSH Line Editor](http://zsh.sourceforge.net/Doc/Release/Zsh-L
 |`POWERLEVEL9K_VI_INSERT_MODE_STRING`|`"INSERT"`|String to display while in 'Insert' mode.|
 |`POWERLEVEL9K_VI_COMMAND_MODE_STRING`|`"NORMAL"`|String to display while in 'Command' mode.|
 
+To hide the segment entirely when in `INSERT` mode, set `POWERLEVEL9K_VI_INSERT_MODE_STRING=''`
+
 #### 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 15a0b68a..6e30efee 100755
--- a/powerlevel9k.zsh-theme
+++ b/powerlevel9k.zsh-theme
@@ -1366,6 +1366,7 @@ prompt_vi_mode() {
       "$1_prompt_segment" "$0_NORMAL" "$2" "$DEFAULT_COLOR" "default" "$POWERLEVEL9K_VI_COMMAND_MODE_STRING"
     ;;
     main|viins|*)
+      if [[ -z $POWERLEVEL9K_VI_INSERT_MODE_STRING ]]; then return; fi
       "$1_prompt_segment" "$0_INSERT" "$2" "$DEFAULT_COLOR" "blue" "$POWERLEVEL9K_VI_INSERT_MODE_STRING"
     ;;
   esac