Make numeric registers setable

Fixes #1214

6 files changed, 71 insertions, 57 deletions

diff --git a/src/commands.cc b/src/commands.cc
index 65a0b752..8d98e1e0 100644
--- a/src/commands.cc
+++ b/src/commands.cc
@@ -1487,32 +1487,24 @@ private:
 class RegisterRestorer
 {
 public:
-    RegisterRestorer(char name, const Context& context)
-      : m_name(name)
+    RegisterRestorer(char name, Context& context)
+      : m_context{context}, m_name{name}
     {
-        ConstArrayView<String> save = RegisterManager::instance()[name].values(context);
+        ConstArrayView<String> save = RegisterManager::instance()[name].get(context);
         m_save = Vector<String>(save.begin(), save.end());
     }
 
     RegisterRestorer(RegisterRestorer&& other) noexcept
-        : m_save(std::move(other.m_save)), m_name(other.m_name)
+        : m_context{other.m_context}, m_save{std::move(other.m_save)}, m_name{other.m_name}
     {
         other.m_name = 0;
     }
 
-    RegisterRestorer& operator=(RegisterRestorer&& other) noexcept
-    {
-        m_save = std::move(other.m_save);
-        m_name = other.m_name;
-        other.m_name = 0;
-        return *this;
-    }
-
     ~RegisterRestorer()
     {
         if (m_name != 0) try
         {
-            RegisterManager::instance()[m_name] = m_save;
+            RegisterManager::instance()[m_name].set(m_context, m_save);
         }
         catch (runtime_error& e)
         {
@@ -1523,6 +1515,7 @@ public:
 
 private:
     Vector<String> m_save;
+    Context&       m_context;
     char           m_name;
 };
 
@@ -2012,7 +2005,7 @@ const CommandDesc set_register_cmd = {
     CommandCompleter{},
     [](const ParametersParser& parser, Context& context, const ShellContext&)
     {
-        RegisterManager::instance()[parser[0]] = ConstArrayView<String>(parser[1]);
+        RegisterManager::instance()[parser[0]].set(context, {parser[1]});
     }
 };
 
diff --git a/src/context.cc b/src/context.cc
index dc8086fa..8d225052 100644
--- a/src/context.cc
+++ b/src/context.cc
@@ -211,7 +211,7 @@ void Context::end_edition()
 
 StringView Context::main_sel_register_value(StringView reg) const
 {
-    auto strings = RegisterManager::instance()[reg].values(*this);
+    auto strings = RegisterManager::instance()[reg].get(*this);
     size_t index = m_selections ? (*m_selections).main_index() : 0;
     if (strings.size() <= index)
         index = strings.size() - 1;
diff --git a/src/input_handler.cc b/src/input_handler.cc
index b112f7ad..3c72e66d 100644
--- a/src/input_handler.cc
+++ b/src/input_handler.cc
@@ -1124,7 +1124,7 @@ public:
             on_next_key_with_autoinfo(context(), KeymapMode::None,
                 [this](Key key, Context&) {
                     if (auto cp = key.codepoint())
-                        insert(RegisterManager::instance()[*cp].values(context()));
+                        insert(RegisterManager::instance()[*cp].get(context()));
                 }, "Enter register name", register_doc);
             update_completions = false;
         }
@@ -1491,7 +1491,8 @@ void InputHandler::stop_recording()
     kak_assert(m_recording_reg != 0);
 
     if (not m_recorded_keys.empty())
-        RegisterManager::instance()[m_recording_reg] = {m_recorded_keys};
+        RegisterManager::instance()[m_recording_reg].set(
+            context(), {m_recorded_keys});
 
     m_recording_reg = 0;
     m_recording_level = -1;
diff --git a/src/main.cc b/src/main.cc
index 7aba8f26..09902b5d 100644
--- a/src/main.cc
+++ b/src/main.cc
@@ -186,6 +186,19 @@ void register_registers()
                 for (auto& sel : context.selections())
                     result.emplace_back(i < sel.captures().size() ? sel.captures()[i] : "");
                 return result;
+            },
+            [i](Context& context, ConstArrayView<String> values) {
+                if (values.empty())
+                    return;
+
+                auto& sels = context.selections();
+                for (size_t sel_index = 0; sel_index < sels.size(); ++sel_index)
+                {
+                    auto& sel = sels[sel_index];
+                    if (sel.captures().size() < i+1)
+                        sel.captures().resize(i+1);
+                    sel.captures()[i] = values[std::min(sel_index, values.size()-1)];
+                }
             }));
     }
 
diff --git a/src/normal.cc b/src/normal.cc
index 6a34f6ac..3f816b02 100644
--- a/src/normal.cc
+++ b/src/normal.cc
@@ -417,7 +417,7 @@ void pipe(Context& context, NormalParams)
                 real_cmd = context.main_sel_register_value("|");
             else
             {
-                RegisterManager::instance()['|'] = cmdline.str();
+                RegisterManager::instance()['|'].set(context, cmdline.str());
                 real_cmd = cmdline;
             }
 
@@ -479,7 +479,7 @@ void insert_output(Context& context, NormalParams)
                 real_cmd = context.main_sel_register_value("|");
             else
             {
-                RegisterManager::instance()['|'] = cmdline.str();
+                RegisterManager::instance()['|'].set(context, cmdline.str());
                 real_cmd = cmdline;
             }
 
@@ -496,7 +496,7 @@ void insert_output(Context& context, NormalParams)
 void yank(Context& context, NormalParams params)
 {
     const char reg = params.reg ? params.reg : '"';
-    RegisterManager::instance()[reg] = context.selections_content();
+    RegisterManager::instance()[reg].set(context, context.selections_content());
     context.print_status({ format("yanked {} selections to register {}",
                                   context.selections().size(), reg),
                            get_face("Information") });
@@ -505,7 +505,7 @@ void yank(Context& context, NormalParams params)
 void erase_selections(Context& context, NormalParams params)
 {
     const char reg = params.reg ? params.reg : '"';
-    RegisterManager::instance()[reg] = context.selections_content();
+    RegisterManager::instance()[reg].set(context, context.selections_content());
     ScopedEdition edition(context);
     context.selections().erase();
     context.selections().avoid_eol();
@@ -514,7 +514,7 @@ void erase_selections(Context& context, NormalParams params)
 void change(Context& context, NormalParams params)
 {
     const char reg = params.reg ? params.reg : '"';
-    RegisterManager::instance()[reg] = context.selections_content();
+    RegisterManager::instance()[reg].set(context, context.selections_content());
     enter_insert_mode<InsertMode::Replace>(context, params);
 }
 
@@ -533,7 +533,7 @@ template<InsertMode mode>
 void paste(Context& context, NormalParams params)
 {
     const char reg = params.reg ? params.reg : '"';
-    auto strings = RegisterManager::instance()[reg].values(context);
+    auto strings = RegisterManager::instance()[reg].get(context);
     const bool linewise = contains_that(strings, [](StringView str) {
         return not str.empty() and str.back() == '\n';
     });
@@ -547,7 +547,7 @@ template<InsertMode mode>
 void paste_all(Context& context, NormalParams params)
 {
     const char reg = params.reg ? params.reg : '"';
-    auto strings = RegisterManager::instance()[reg].values(context);
+    auto strings = RegisterManager::instance()[reg].get(context);
     InsertMode effective_mode = mode;
     String all;
     Vector<ByteCount> offsets;
@@ -645,7 +645,7 @@ void search(Context& context, NormalParams params)
     const char reg = to_lower(params.reg ? params.reg : '/');
     const int count = params.count;
 
-    auto reg_content = RegisterManager::instance()[reg].values(context);
+    auto reg_content = RegisterManager::instance()[reg].get(context);
     Vector<String> saved_reg{reg_content.begin(), reg_content.end()};
     const int main_index = std::min(context.selections().main_index(), saved_reg.size()-1);
 
@@ -654,13 +654,13 @@ void search(Context& context, NormalParams params)
                  (Regex regex, PromptEvent event, Context& context) {
                      if (event == PromptEvent::Abort)
                      {
-                         RegisterManager::instance()[reg] = saved_reg;
+                         RegisterManager::instance()[reg].set(context, saved_reg);
                          return;
                      }
 
                      if (regex.empty())
                          regex = Regex{saved_reg[main_index]};
-                     RegisterManager::instance()[reg] = regex.str();
+                     RegisterManager::instance()[reg].set(context, regex.str());
 
                      if (not regex.empty() and not regex.str().empty())
                      {
@@ -741,7 +741,7 @@ void use_selection_as_search_pattern(Context& context, NormalParams params)
         format("register '{}' set to '{}'", reg, patterns[sels.main_index()]),
         get_face("Information") });
 
-    RegisterManager::instance()[reg] = patterns;
+    RegisterManager::instance()[reg].set(context, patterns);
 
     // Hack, as Window do not take register state into account
     if (context.has_window())
@@ -754,7 +754,7 @@ void select_regex(Context& context, NormalParams params)
     const int capture = params.count;
     auto prompt = capture ? format("select (capture {}):", capture) :  "select:"_str;
 
-    auto reg_content = RegisterManager::instance()[reg].values(context);
+    auto reg_content = RegisterManager::instance()[reg].get(context);
     Vector<String> saved_reg{reg_content.begin(), reg_content.end()};
     const int main_index = std::min(context.selections().main_index(), saved_reg.size()-1);
 
@@ -762,13 +762,13 @@ void select_regex(Context& context, NormalParams params)
                  [reg, capture, saved_reg, main_index](Regex ex, PromptEvent event, Context& context) {
          if (event == PromptEvent::Abort)
          {
-             RegisterManager::instance()[reg] = saved_reg;
+             RegisterManager::instance()[reg].set(context, saved_reg);
              return;
          }
 
         if (ex.empty())
             ex = Regex{saved_reg[main_index]};
-        RegisterManager::instance()[reg] = ex.str();
+        RegisterManager::instance()[reg].set(context, ex.str());
 
         if (not ex.empty() and not ex.str().empty())
             select_all_matches(context.selections(), ex, capture);
@@ -781,7 +781,7 @@ void split_regex(Context& context, NormalParams params)
     const int capture = params.count;
     auto prompt = capture ? format("split (on capture {}):", (int)capture) :  "split:"_str;
 
-    auto reg_content = RegisterManager::instance()[reg].values(context);
+    auto reg_content = RegisterManager::instance()[reg].get(context);
     Vector<String> saved_reg{reg_content.begin(), reg_content.end()};
     const int main_index = std::min(context.selections().main_index(), saved_reg.size()-1);
 
@@ -789,13 +789,13 @@ void split_regex(Context& context, NormalParams params)
                  [reg, capture, saved_reg, main_index](Regex ex, PromptEvent event, Context& context) {
          if (event == PromptEvent::Abort)
          {
-             RegisterManager::instance()[reg] = saved_reg;
+             RegisterManager::instance()[reg].set(context, saved_reg);
              return;
          }
 
         if (ex.empty())
             ex = Regex{saved_reg[main_index]};
-        RegisterManager::instance()[reg] = ex.str();
+        RegisterManager::instance()[reg].set(context, ex.str());
 
         if (not ex.empty() and not ex.str().empty())
             split_selections(context.selections(), ex, capture);
@@ -1254,7 +1254,7 @@ void replay_macro(Context& context, NormalParams params)
     if (running_macros[idx])
         throw runtime_error("recursive macros call detected");
 
-    ConstArrayView<String> reg_val = RegisterManager::instance()[reg].values(context);
+    ConstArrayView<String> reg_val = RegisterManager::instance()[reg].get(context);
     if (reg_val.empty() or reg_val[0].empty())
         throw runtime_error(format("Register '{}' is empty", reg));
 
@@ -1445,7 +1445,7 @@ SelectionList read_selections_from_register(char reg, Context& context)
     if (not is_basic_alpha(reg) and reg != '^')
         throw runtime_error("selections can only be saved to the '^' and alphabetic registers");
 
-    auto content = RegisterManager::instance()[reg].values(context);
+    auto content = RegisterManager::instance()[reg].get(context);
 
     if (content.size() != 1)
         throw runtime_error(format("Register {} does not contain a selections desc", reg));
@@ -1496,7 +1496,7 @@ void save_selections(Context& context, NormalParams params)
                          context.buffer().name(),
                          context.buffer().timestamp());
 
-    RegisterManager::instance()[reg] = desc;
+    RegisterManager::instance()[reg].set(context, desc);
 
     context.print_status({format("{} selections to register '{}'", add ? "Added" : "Saved", reg), get_face("Information")});
 }
diff --git a/src/register_manager.hh b/src/register_manager.hh
index d5bc1ec3..4a8e7f2c 100644
--- a/src/register_manager.hh
+++ b/src/register_manager.hh
@@ -17,9 +17,9 @@ class Register
 {
 public:
     virtual ~Register() = default;
-    virtual Register& operator=(ConstArrayView<String> values) = 0;
 
-    virtual ConstArrayView<String> values(const Context& context) = 0;
+    virtual void set(Context& context, ConstArrayView<String> values) = 0;
+    virtual ConstArrayView<String> get(const Context& context) = 0;
 };
 
 // static value register, which can be modified
@@ -27,13 +27,12 @@ public:
 class StaticRegister : public Register
 {
 public:
-    Register& operator=(ConstArrayView<String> values) override
+    void set(Context&, ConstArrayView<String> values) override
     {
         m_content = Vector<String, MemoryDomain::Registers>(values.begin(), values.end());
-        return *this;
     }
 
-    ConstArrayView<String> values(const Context&) override
+    ConstArrayView<String> get(const Context&) override
     {
         if (m_content.empty())
             return ConstArrayView<String>(String::ms_empty);
@@ -46,43 +45,51 @@ protected:
 
 // Dynamic value register, use it's RegisterRetriever
 // to get it's value when needed.
-template<typename Func>
+template<typename Getter, typename Setter>
 class DynamicRegister : public StaticRegister
 {
 public:
-    DynamicRegister(Func function)
-        : m_function(std::move(function)) {}
+    DynamicRegister(Getter getter, Setter setter)
+        : m_getter{std::move(getter)}, m_setter{std::move(setter)} {}
 
-    Register& operator=(ConstArrayView<String> values) override
+    void set(Context& context, ConstArrayView<String> values) override
     {
-        throw runtime_error("this register is not assignable");
+        m_setter(context, values);
     }
 
-    ConstArrayView<String> values(const Context& context) override
+    ConstArrayView<String> get(const Context& context) override
     {
-        m_content = m_function(context);
-        return StaticRegister::values(context);
+        m_content = m_getter(context);
+        return StaticRegister::get(context);
     }
 
 private:
-    Func m_function;
+    Getter m_getter;
+    Setter m_setter;
 };
 
 template<typename Func>
 std::unique_ptr<Register> make_dyn_reg(Func func)
 {
-    return make_unique<DynamicRegister<Func>>(std::move(func));
+    auto setter = [](Context&, ConstArrayView<String>)
+    {
+        throw runtime_error("this register is not assignable");
+    };
+    return make_unique<DynamicRegister<Func, decltype(setter)>>(std::move(func), setter);
+}
+
+template<typename Getter, typename Setter>
+std::unique_ptr<Register> make_dyn_reg(Getter getter, Setter setter)
+{
+    return make_unique<DynamicRegister<Getter, Setter>>(std::move(getter), std::move(setter));
 }
 
 class NullRegister : public Register
 {
 public:
-    Register& operator=(ConstArrayView<String> values) override
-    {
-        return *this;
-    }
+    void set(Context&, ConstArrayView<String>) override {}
 
-    ConstArrayView<String> values(const Context& context) override
+    ConstArrayView<String> get(const Context&) override
     {
         return ConstArrayView<String>(String::ms_empty);
     }