rework selection insert/replace using a for_each method

expose that method so that various commands can take advantage of
it for performance or simplicity purposes.

4 files changed, 97 insertions, 90 deletions

diff --git a/src/buffer_utils.hh b/src/buffer_utils.hh
index a3f18fe1..e0b40425 100644
--- a/src/buffer_utils.hh
+++ b/src/buffer_utils.hh
@@ -20,11 +20,6 @@ inline BufferCoord erase(Buffer& buffer, const Selection& range)
     return buffer.erase(range.min(), buffer.char_next(range.max()));
 }
 
-inline BufferRange replace(Buffer& buffer, const Selection& range, StringView content)
-{
-    return buffer.replace(range.min(), buffer.char_next(range.max()), content);
-}
-
 void replace(Buffer& buffer, ArrayView<BufferRange> ranges, ConstArrayView<String> strings);
 
 inline CharCount char_length(const Buffer& buffer, const Selection& range)
diff --git a/src/normal.cc b/src/normal.cc
index cba92d0e..d7222a13 100644
--- a/src/normal.cc
+++ b/src/normal.cc
@@ -413,11 +413,10 @@ void replace_with_char(Context& context, NormalParams)
             return;
         ScopedEdition edition(context);
         Buffer& buffer = context.buffer();
-        SelectionList& selections = context.selections();
-        selections.insert([&](size_t index, BufferCoord) {
-            CharCount count = char_length(buffer, selections[index]);
-            return String{*cp, count};
-        }, InsertMode::Replace);
+        context.selections().for_each([&](size_t index, Selection& sel) {
+            CharCount count = char_length(buffer, sel);
+            replace(buffer, sel, String{*cp, count});
+        });
     }, "replace with char", "enter char to replace with\n");
 }
 
@@ -434,17 +433,15 @@ void for_each_codepoint(Context& context, NormalParams)
 
     ScopedEdition edition(context);
     Buffer& buffer = context.buffer();
-    SelectionList& selections = context.selections();
 
-    selections.insert([&](size_t index, BufferCoord) {
-        auto& sel = selections[index];
+    context.selections().for_each([&](size_t index, Selection& sel) {
         String str;
         for (auto begin = Utf8It{buffer.iterator_at(sel.min()), buffer},
                   end = Utf8It{buffer.iterator_at(sel.max()), buffer}+1;
              begin != end; ++begin)
             utf8::dump(std::back_inserter(str), func(*begin));
-        return str;
-    }, InsertMode::Replace);
+        replace(buffer, sel, str);
+    });
 }
 
 void command(const Context& context, EnvVarMap env_vars, char reg = 0)
@@ -642,15 +639,17 @@ void insert_output(Context& context, NormalParams params)
 
             ScopedEdition edition(context);
             auto& selections = context.selections();
+            auto& buffer = context.buffer();
             const size_t old_main = selections.main_index();
 
-            selections.insert([&](size_t index, BufferCoord) {
+            selections.for_each([&](size_t index, Selection& sel) {
                 selections.set_main_index(index);
                 auto [out, status] = ShellManager::instance().eval(
-                    cmdline, context, content(context.buffer(), selections[index]),
+                    cmdline, context, content(context.buffer(), sel),
                     ShellManager::Flags::WaitForStdout);
-                return out;
-            }, mode);
+
+                insert(buffer, sel, out, mode);
+            });
 
             selections.set_main_index(old_main);
         });
@@ -688,15 +687,19 @@ void change(Context& context, NormalParams params)
     enter_insert_mode<InsertMode::Replace>(context, params);
 }
 
-InsertMode adapt_for_linewise(InsertMode mode)
+InsertMode adapt_for_linewise(InsertMode mode, bool linewise)
 {
+    if (not linewise)
+        return mode;
+
     switch (mode)
     {
         case InsertMode::Append: return InsertMode::InsertAtNextLineBegin;
         case InsertMode::Insert: return InsertMode::InsertAtLineBegin;
-        case InsertMode::Replace: return InsertMode::Replace;
-        default: return InsertMode::Insert;
+        default: break;
     }
+    kak_assert(false);
+    return InsertMode::Insert;
 }
 
 template<InsertMode mode>
@@ -707,10 +710,12 @@ void paste(Context& context, NormalParams params)
     const bool linewise = any_of(strings, [](StringView str) {
         return not str.empty() and str.back() == '\n';
     });
-    const auto effective_mode = linewise ? adapt_for_linewise(mode) : mode;
 
     ScopedEdition edition(context);
-    context.selections().insert(strings, effective_mode);
+    if (mode == InsertMode::Replace)
+        context.selections().replace(strings);
+    else
+        context.selections().insert(strings, adapt_for_linewise(mode, linewise));
 }
 
 template<InsertMode mode>
@@ -718,7 +723,7 @@ void paste_all(Context& context, NormalParams params)
 {
     const char reg = params.reg ? params.reg : '"';
     auto strings = RegisterManager::instance()[reg].get(context);
-    InsertMode effective_mode = mode;
+    bool linewise = false;
     String all;
     Vector<ByteCount> offsets;
     for (auto& str : strings)
@@ -727,37 +732,35 @@ void paste_all(Context& context, NormalParams params)
             continue;
 
         if (str.back() == '\n')
-            effective_mode = adapt_for_linewise(mode);
+            linewise = true;
         all += str;
         offsets.push_back(all.length());
     }
 
-    Vector<BufferCoord> insert_pos;
-    auto& selections = context.selections();
-    {
-        ScopedEdition edition(context);
-        selections.insert([&](size_t, BufferCoord pos) {
-            insert_pos.push_back(pos);
-            return String::no_copy(all);
-        }, effective_mode);
-    }
+    InsertMode effective_mode = adapt_for_linewise(mode, linewise);
 
-    const Buffer& buffer = context.buffer();
+    Buffer& buffer = context.buffer();
     Vector<Selection> result;
-    for (auto& ins_pos : insert_pos)
+    auto& selections = context.selections();
     {
-        ByteCount pos_offset = 0;
-        BufferCoord pos = ins_pos;
-        for (auto offset : offsets)
-        {
-            BufferCoord end = buffer.advance(pos, offset - pos_offset - 1);
-            result.emplace_back(pos, end);
-            pos = buffer.next(end);
-            pos_offset = offset;
-        }
+        ScopedEdition edition(context);
+        selections.for_each([&](size_t, const Selection& sel) {
+            auto range = (mode == InsertMode::Replace) ?
+                buffer.replace(sel.min(), buffer.char_next(sel.max()), all)
+              : buffer.insert(get_insert_pos(buffer, sel, effective_mode), all);
+
+            ByteCount pos_offset = 0;
+            BufferCoord pos = range.begin;
+            for (auto offset : offsets)
+            {
+                BufferCoord end = buffer.advance(pos, offset - pos_offset - 1);
+                result.emplace_back(pos, end);
+                pos = buffer.next(end);
+                pos_offset = offset;
+            }
+        });
     }
-    if (not result.empty())
-        selections = std::move(result);
+    selections = std::move(result);
 }
 
 constexpr RegexCompileFlags direction_flags(RegexMode mode)
@@ -1079,7 +1082,7 @@ void join_lines_select_spaces(Context& context, NormalParams)
         return;
     context.selections_write_only() = std::move(selections);
     ScopedEdition edition(context);
-    context.selections().insert(" "_str, InsertMode::Replace);
+    context.selections().replace({" "_str});
 }
 
 void join_lines(Context& context, NormalParams params)
@@ -1478,7 +1481,7 @@ void rotate_selections_content(Context& context, NormalParams params)
             main = main < new_beg ? end - (new_beg - main) : it + (main - new_beg);
         it = end;
     }
-    selections.insert(strings, InsertMode::Replace);
+    selections.replace(strings);
     selections.set_main_index(main - strings.begin());
 }
 
@@ -1696,7 +1699,7 @@ void tabs_to_spaces(Context& context, NormalParams params)
         }
     }
     if (not tabs.empty())
-        SelectionList{ buffer, std::move(tabs) }.insert(spaces, InsertMode::Replace);
+        SelectionList{ buffer, std::move(tabs) }.replace(spaces);
 }
 
 void spaces_to_tabs(Context& context, NormalParams params)
@@ -1732,7 +1735,7 @@ void spaces_to_tabs(Context& context, NormalParams params)
         }
     }
     if (not spaces.empty())
-        SelectionList{ buffer, std::move(spaces) }.insert("\t"_str, InsertMode::Replace);
+        SelectionList{ buffer, std::move(spaces) }.replace("\t"_str);
 }
 
 void trim_selections(Context& context, NormalParams)
diff --git a/src/selection.cc b/src/selection.cc
index a552526d..27a40156 100644
--- a/src/selection.cc
+++ b/src/selection.cc
@@ -367,7 +367,7 @@ BufferCoord get_insert_pos(const Buffer& buffer, const Selection& sel,
     case InsertMode::AppendAtLineEnd:
         return {sel.max().line, buffer[sel.max().line].length() - 1};
     case InsertMode::InsertAtNextLineBegin:
-        return sel.max().line+1;
+        return std::min(buffer.line_count(), sel.max().line+1);
     default:
         kak_assert(false);
         return {};
@@ -390,22 +390,15 @@ void SelectionList::insert(ConstArrayView<String> strings, InsertMode mode)
     if (strings.empty())
         return;
 
-    insert([&](size_t index, BufferCoord) {
-        return String::no_copy(strings[std::min(strings.size()-1, index)]);
-    }, mode);
+    for_each([&](size_t index, Selection& sel) {
+        Kakoune::insert(*m_buffer, sel, strings[std::min(strings.size()-1, index)], mode);
+    });
 }
 
-void SelectionList::insert(ContentFunc get_content, InsertMode mode)
+void SelectionList::for_each(ApplyFunc func)
 {
     update();
 
-    Vector<BufferCoord> insert_pos;
-    if (mode != InsertMode::Replace)
-    {
-        for (auto& sel : m_selections)
-            insert_pos.push_back(get_insert_pos(*m_buffer, sel, mode));
-    }
-
     ForwardChangesTracker changes_tracker;
     for (size_t index = 0; index < m_selections.size(); ++index)
     {
@@ -413,42 +406,49 @@ void SelectionList::insert(ContentFunc get_content, InsertMode mode)
 
         sel.anchor() = changes_tracker.get_new_coord_tolerant(sel.anchor());
         sel.cursor() = changes_tracker.get_new_coord_tolerant(sel.cursor());
-        kak_assert(m_buffer->is_valid(sel.anchor()) and
-                   m_buffer->is_valid(sel.cursor()));
+        kak_assert(m_buffer->is_valid(sel.anchor()) and m_buffer->is_valid(sel.cursor()));
 
-        const auto pos = (mode == InsertMode::Replace) ?
-            sel.min() : changes_tracker.get_new_coord(insert_pos[index]);
-
-        String str = get_content(index, pos);
-
-        if (mode == InsertMode::Replace)
-        {
-            auto range = replace(*m_buffer, sel, str);
-            // we want min and max from *before* we do any change
-            auto& min = sel.min();
-            auto& max = sel.max();
-            min = range.begin;
-            max = range.end > range.begin ? m_buffer->char_prev(range.end) : range.begin;
-        }
-        else
-        {
-            auto range = m_buffer->insert(pos, str);
-            sel.anchor() = m_buffer->clamp(update_insert(sel.anchor(), range.begin, range.end));
-            sel.cursor() = m_buffer->clamp(update_insert(sel.cursor(), range.begin, range.end));
-        }
+        func(index, sel);
 
         changes_tracker.update(*m_buffer, m_timestamp);
     }
 
     // We might just have been deleting text if strings were empty,
     // in which case we could have some selections pushed out of the buffer
-    if (mode == InsertMode::Replace)
-        fix_overflowing_selections(m_selections, *m_buffer);
+    fix_overflowing_selections(m_selections, *m_buffer);
 
     check_invariant();
     m_buffer->check_invariant();
 }
 
+
+void replace(Buffer& buffer, Selection& sel, StringView content)
+{
+    // we want min and max from *before* we do any change
+    auto& min = sel.min();
+    auto& max = sel.max();
+    BufferRange range = buffer.replace(min, buffer.char_next(max), content);
+    min = range.begin;
+    max = range.end > range.begin ? buffer.char_prev(range.end) : range.begin;
+}
+
+void insert(Buffer& buffer, Selection& sel, StringView content, InsertMode mode)
+{
+    auto range = buffer.insert(get_insert_pos(buffer, sel, mode), content);
+    sel.anchor() = buffer.clamp(update_insert(sel.anchor(), range.begin, range.end));
+    sel.cursor() = buffer.clamp(update_insert(sel.cursor(), range.begin, range.end));
+}
+
+void SelectionList::replace(ConstArrayView<String> strings)
+{
+    if (strings.empty())
+        return;
+
+    for_each([&](size_t index, Selection& sel) {
+        Kakoune::replace(*m_buffer, sel, strings[std::min(strings.size()-1, index)]);
+    });
+}
+
 void SelectionList::erase()
 {
     update();
diff --git a/src/selection.hh b/src/selection.hh
index 3c6d879d..5d6e7427 100644
--- a/src/selection.hh
+++ b/src/selection.hh
@@ -87,6 +87,12 @@ enum class InsertMode : unsigned
     OpenLineAbove
 };
 
+BufferCoord get_insert_pos(const Buffer& buffer, const Selection& sel, InsertMode mode);
+
+void replace(Buffer& buffer, Selection& sel, StringView content);
+void insert(Buffer& buffer, Selection& sel, StringView content, InsertMode mode);
+
+
 struct SelectionList
 {
     static constexpr MemoryDomain Domain = MemoryDomain::Selections;
@@ -145,9 +151,12 @@ struct SelectionList
     size_t timestamp() const { return m_timestamp; }
     void force_timestamp(size_t timestamp) { m_timestamp = timestamp; }
 
+    using ApplyFunc = FunctionRef<void (size_t index, Selection& sel)>;
+    void for_each(ApplyFunc apply);
+
     void insert(ConstArrayView<String> strings, InsertMode mode);
-    using ContentFunc = FunctionRef<String (size_t index, BufferCoord insert_pos)>;
-    void insert(ContentFunc get_content, InsertMode mode);
+    void replace(ConstArrayView<String> strings);
+
     void erase();
 
 private: