vis: make <Escape> reset count in visual modes

1 files changed, 16 insertions, 0 deletions

diff --git a/main.c b/main.c
index de179ad..af15ffb 100644
--- a/main.c
+++ b/main.c
@@ -34,6 +34,8 @@ static const char *macro_replay(Vis*, const char *keys, const Arg *arg);
 static const char *suspend(Vis*, const char *keys, const Arg *arg);
 /* reset count if set, otherwise remove all but the primary selection */
 static const char *normalmode_escape(Vis*, const char *keys, const Arg *arg);
+/* reset count if set, otherwise switch to normal mode */
+static const char *visualmode_escape(Vis*, const char *keys, const Arg *arg);
 /* switch to mode indicated by arg->i */
 static const char *switchmode(Vis*, const char *keys, const Arg *arg);
 /* switch to insert mode after performing movement indicated by arg->i */
@@ -204,6 +206,7 @@ enum {
 	VIS_ACTION_MODE_NORMAL,
 	VIS_ACTION_MODE_NORMAL_ESCAPE,
 	VIS_ACTION_MODE_VISUAL,
+	VIS_ACTION_MODE_VISUAL_ESCAPE,
 	VIS_ACTION_MODE_VISUAL_LINE,
 	VIS_ACTION_MODE_INSERT,
 	VIS_ACTION_MODE_REPLACE,
@@ -613,6 +616,11 @@ static const KeyAction vis_action[] = {
 		VIS_HELP("Enter characterwise visual mode")
 		switchmode, { .i = VIS_MODE_VISUAL }
 	},
+	[VIS_ACTION_MODE_VISUAL_ESCAPE] = {
+		"vis-mode-visual-escape",
+		VIS_HELP("Reset count or switch to normal mode")
+		visualmode_escape,
+	},
 	[VIS_ACTION_MODE_VISUAL_LINE] = {
 		"vis-mode-visual-linewise",
 		VIS_HELP("Enter linewise visual mode")
@@ -2244,6 +2252,14 @@ static const char *normalmode_escape(Vis *vis, const char *keys, const Arg *arg)
 	return keys;
 }
 
+static const char *visualmode_escape(Vis *vis, const char *keys, const Arg *arg) {
+	if (vis_count_get(vis) == VIS_COUNT_UNKNOWN)
+		vis_mode_switch(vis, VIS_MODE_NORMAL);
+	else
+		vis_count_set(vis, VIS_COUNT_UNKNOWN);
+	return keys;
+}
+
 static const char *switchmode(Vis *vis, const char *keys, const Arg *arg) {
 	vis_mode_switch(vis, arg->i);
 	return keys;