From 05f31a587331bd0505c078eb098c2f1df8967c5e Mon Sep 17 00:00:00 2001
From: Arnaud Venturi <arnaud.venturi@rez-gif.supelec.fr>
Date: Thu, 21 Dec 2017 11:01:23 +0100
Subject: [PATCH] Fixed a bug calling end-of-line for every forward-char call

---
 src/widgets.zsh         | 25 +++++++++++++++++++------
 zsh-autosuggestions.zsh | 25 +++++++++++++++++++------
 2 files changed, 38 insertions(+), 12 deletions(-)

diff --git a/src/widgets.zsh b/src/widgets.zsh
index f470e71..1ccf2c2 100644
--- a/src/widgets.zsh
+++ b/src/widgets.zsh
@@ -114,12 +114,25 @@ _zsh_autosuggest_suggest() {
 
 # Accept the entire suggestion
 _zsh_autosuggest_accept() {
-  # Accepting the whole line is basically a specific case of
-  # accepting partially with "end-of-line" widget
-  # NB : we can't directly call end-of-line because since we wrap it,
-  # it would cause a recursive call to it
-  # TODO : find a non-hardcoded way to call this widget
-  _zsh_autosuggest_partial_accept "autosuggest-orig-1-end-of-line"
+	local -i max_cursor_pos=$#BUFFER
+
+	# When vicmd keymap is active, the cursor can't move all the way
+	# to the end of the buffer
+	if [[ "$KEYMAP" = "vicmd" ]]; then
+		max_cursor_pos=$((max_cursor_pos - 1))
+	fi
+
+	# Only accept if the cursor is at the end of the buffer
+	if [[ $CURSOR = $max_cursor_pos ]]; then
+    # Accepting the whole line is basically a specific case of
+    # accepting partially with "end-of-line" widget
+    # NB : we can't directly call end-of-line because since we wrap it,
+    # it would cause a recursive call to it
+    # TODO : find a non-hardcoded way to call this widget
+    _zsh_autosuggest_partial_accept "autosuggest-orig-1-end-of-line"
+	fi
+
+	_zsh_autosuggest_invoke_original_widget $@
 }
 
 # Accept the entire suggestion and execute it
diff --git a/zsh-autosuggestions.zsh b/zsh-autosuggestions.zsh
index 23a13b0..8524f7d 100644
--- a/zsh-autosuggestions.zsh
+++ b/zsh-autosuggestions.zsh
@@ -379,12 +379,25 @@ _zsh_autosuggest_suggest() {
 
 # Accept the entire suggestion
 _zsh_autosuggest_accept() {
-  # Accepting the whole line is basically a specific case of
-  # accepting partially with "end-of-line" widget
-  # NB : we can't directly call end-of-line because since we wrap it,
-  # it would cause a recursive call to it
-  # TODO : find a non-hardcoded way to call this widget
-  _zsh_autosuggest_partial_accept "autosuggest-orig-1-end-of-line"
+	local -i max_cursor_pos=$#BUFFER
+
+	# When vicmd keymap is active, the cursor can't move all the way
+	# to the end of the buffer
+	if [[ "$KEYMAP" = "vicmd" ]]; then
+		max_cursor_pos=$((max_cursor_pos - 1))
+	fi
+
+	# Only accept if the cursor is at the end of the buffer
+	if [[ $CURSOR = $max_cursor_pos ]]; then
+    # Accepting the whole line is basically a specific case of
+    # accepting partially with "end-of-line" widget
+    # NB : we can't directly call end-of-line because since we wrap it,
+    # it would cause a recursive call to it
+    # TODO : find a non-hardcoded way to call this widget
+    _zsh_autosuggest_partial_accept "autosuggest-orig-1-end-of-line"
+	fi
+
+	_zsh_autosuggest_invoke_original_widget $@
 }
 
 # Accept the entire suggestion and execute it