Remember count when repeating last insert

Fixes #1465

2 files changed, 4 insertions, 3 deletions

diff --git a/src/input_handler.cc b/src/input_handler.cc
index 1ac60905..217f3a5c 100644
--- a/src/input_handler.cc
+++ b/src/input_handler.cc
@@ -1057,6 +1057,7 @@ public:
         last_insert().mode = mode;
         last_insert().keys.clear();
         last_insert().disable_hooks = context().hooks_disabled();
+        last_insert().count = count;
         context().hooks().run_hook("InsertBegin", "", context());
         prepare(mode, count);
 
@@ -1435,7 +1436,7 @@ void InputHandler::repeat_last_insert()
                                 m_last_insert.disable_hooks);
     // context.last_insert will be refilled by the new Insert
     // this is very inefficient.
-    push_mode(new InputModes::Insert(*this, m_last_insert.mode, 1));
+    push_mode(new InputModes::Insert(*this, m_last_insert.mode, m_last_insert.count));
     for (auto& key : keys)
         current_mode().handle_key(key);
     kak_assert(dynamic_cast<InputModes::Normal*>(&current_mode()) != nullptr);
diff --git a/src/input_handler.hh b/src/input_handler.hh
index 17c655d9..4287f172 100644
--- a/src/input_handler.hh
+++ b/src/input_handler.hh
@@ -116,8 +116,8 @@ private:
     void push_mode(InputMode* new_mode);
     void pop_mode(InputMode* current_mode);
 
-    struct Insertion{ InsertMode mode; Vector<Key> keys; bool disable_hooks; };
-    Insertion m_last_insert = { InsertMode::Insert, {}, false };
+    struct Insertion{ InsertMode mode; Vector<Key> keys; bool disable_hooks; int count; };
+    Insertion m_last_insert = { InsertMode::Insert, {}, false, 1 };
 
     char   m_recording_reg = 0;
     String m_recorded_keys;