X-Git-Url: https://git.r.bdr.sh/rbdr/dotfiles/blobdiff_plain/9041bd938fcce0cf91032c67d529bc21fedeeed3..07c693d23065b9371ba3aa713e8fd4b0e9874c9d:/ben-prose.zsh-theme

diff --git a/ben-prose.zsh-theme b/ben-prose.zsh-theme
index b6825ac..3573c24 100644
--- a/ben-prose.zsh-theme
+++ b/ben-prose.zsh-theme
@@ -18,6 +18,15 @@ function box_name {
     [ -f ~/.box-name ] && cat ~/.box-name || hostname -s
 }
 
+function zle-line-init zle-keymap-select {
+    local return_status="%{$fg[red]%}%(?..✘)%{$reset_color%}"
+    RPS1="${${KEYMAP/vicmd/-- NORMAL --}/(main|viins)/-- INSERT --}${return_status}%{$reset_color%}"
+    RPS2=$RPS1
+    zle reset-prompt
+}
+zle -N zle-line-init
+zle -N zle-keymap-select
+
 PROMPT='
 %{$fg[red]%}%n%{$reset_color%} at %{$fg[yellow]%}$(box_name)%{$reset_color%} in %{$fg_bold[green]%}${PWD/#$HOME/~}%{$reset_color%}$(hg_prompt_info)$(git_prompt_info) %{$fg[black]%}$(rvm_info)%{$reset_color%}
 %(?,,%{${fg_bold[white]}%}[%?]%{$reset_color%} )$ '
@@ -28,5 +37,3 @@ ZSH_THEME_GIT_PROMPT_DIRTY="%{$fg[green]%}!"
 ZSH_THEME_GIT_PROMPT_UNTRACKED="%{$fg[green]%}?"
 ZSH_THEME_GIT_PROMPT_CLEAN=""
 
-local return_status="%{$fg[red]%}%(?..✘)%{$reset_color%}"
-RPROMPT='${return_status}%{$reset_color%}'