Use main selection index as default when accessing only one register value

Fixes #117

1 files changed, 10 insertions, 0 deletions

diff --git a/src/context.cc b/src/context.cc
index 4f28b596..64bf8a4c 100644
--- a/src/context.cc
+++ b/src/context.cc
@@ -2,6 +2,7 @@
 
 #include "client.hh"
 #include "user_interface.hh"
+#include "register_manager.hh"
 #include "window.hh"
 
 namespace Kakoune
@@ -226,4 +227,13 @@ void Context::end_edition()
     --m_edition_level;
 }
 
+StringView Context::main_sel_register_value(StringView reg) const
+{
+    auto strings = RegisterManager::instance()[reg].values(*this);
+    size_t index = m_selections ? (*m_selections).main_index() : 0;
+    if (strings.size() < index)
+        index = strings.size() - 1;
+   return strings[index];
+}
+
 }