Визуализация режима для Zsh Vi Mode

Как известно, для zsh можно включить vim-режим для ввода команд. Однако, существует проблема: не всегда можно быстро понять в каком режиме находишься  - NORMAL или INSERT. Это проблему можно побороть, например, добавив в правую часть приглашения визуализацию текущего режима. Для этого надо добавить в .zshrc:

function zle-line-init zle-keymap-select {
    VIM_PROMPT="%{$fg_bold[yellow]%} [% NORMAL]%  %{$reset_color%}"
    RPROMPT="${${KEYMAP/vicmd/$VIM_PROMPT}/(main|viins)/} $EPS1"
    zle reset-prompt
}
zle -N zle-line-init
zle -N zle-keymap-select

Идея взята здесь