From 49ef9da18be155464deb31bda806a1fd7728e52f Mon Sep 17 00:00:00 2001
From: DustVoice <info@dustvoice.de>
Date: Mon, 29 Mar 2021 14:49:10 +0200
Subject: [PATCH] Test out zsh-vi-mode

---
 .antigenrc |  2 +-
 .zshrc     | 30 ++++++++++++++++++++++++++++--
 2 files changed, 29 insertions(+), 3 deletions(-)

diff --git a/.antigenrc b/.antigenrc
index 95b8dbd..0dcec1a 100644
--- a/.antigenrc
+++ b/.antigenrc
@@ -9,7 +9,7 @@ antigen bundle history-substring-search
 antigen bundle zsh-users/zsh-completions
 antigen bundle zsh-users/zsh-autosuggestions
 antigen bundle zsh-users/zsh-syntax-highlighting
-antigen bundle softmoth/zsh-vim-mode
+antigen bundle jeffreytse/zsh-vi-mode
 
 antigen theme $ADOTDIR/themes dustvoice
 
diff --git a/.zshrc b/.zshrc
index bca7844..2b0cb0f 100644
--- a/.zshrc
+++ b/.zshrc
@@ -49,8 +49,34 @@ if [[ -a $ANTIGEN_ZSH && -a $ANTIGENRC ]]; then
 
     antigen init $ANTIGENRC
 
-    bindkey -M vicmd '^K' history-substring-search-up
-    bindkey -M vicmd '^J' history-substring-search-down
+    function zvm_after_lazy_keybindings() {
+        bindkey -M vicmd '^K' history-substring-search-up
+        bindkey -M vicmd '^J' history-substring-search-down
+    }
+    ZVM_NORMAL_MODE_CURSOR=$ZVM_CURSOR_BLOCK
+    ZVM_INSERT_MODE_CURSOR=$ZVM_CURSOR_BLINKING_BEAM
+    ZVM_VISUAL_MODE_CURSOR=$ZVM_CURSOR_BLOCK
+    ZVM_VISUAL_LINE_MODE_CURSOR=$ZVM_CURSOR_BLOCK
+    ZVM_OPPEND_MODE_CURSOR=$ZVM_CURSOR_BLINKING_UNDERLINE
+
+    ZVM_LINE_INIT_MODE=$ZVM_MODE_INSERT
+
+    function zvm_after_select_vi_mode() {
+        case $ZVM_MODE in
+            $ZVM_MODE_NORMAL)
+                RPROMPT=$'%{$fg[blue]%}[NORMAL]%{$reset_color%}'
+            ;;
+            $ZVM_MODE_INSERT)
+                RPROMPT=$''
+            ;;
+            $ZVM_MODE_VISUAL)
+                RPROMPT=$'%{$fg[yellow]%}[VISUAL]%{$reset_color%}'
+            ;;
+            $ZVM_MODE_NORMAL)
+                RPROMPT=$'%{$fg[yellow]%}[V-LINE]%{$reset_color%}'
+            ;;
+        esac
+    }
 
     KEYTIMEOUT=5
 else