From 5b2c0332ad96171c8f6e93b30f6e702339a5ec84 Mon Sep 17 00:00:00 2001
From: Chen Shuaimin <chen_shuaimin@outlook.com>
Date: Sun, 23 Jun 2024 20:44:29 +0800
Subject: [PATCH] fix: preserve extmarks by only edit changed range instead of
 the whole line

---
 lua/dial/command.lua | 19 +++++++++++++------
 lua/dial/handle.lua  | 16 ++++------------
 2 files changed, 17 insertions(+), 18 deletions(-)

diff --git a/lua/dial/command.lua b/lua/dial/command.lua
index 03e3b55..17f1a2f 100644
--- a/lua/dial/command.lua
+++ b/lua/dial/command.lua
@@ -119,8 +119,15 @@ function M.operator_normal(direction, additive)
 
     local result = handler:operate(line, col, direction, additive)
 
-    if result.line ~= nil then
-        vim.fn.setline(".", result.line)
+    if result.text ~= nil then
+        vim.api.nvim_buf_set_text(
+            0,
+            line_num - 1,
+            result.range.from - 1,
+            line_num - 1,
+            result.range.to,
+            { result.text }
+        )
     end
     if result.cursor ~= nil then
         vim.fn.cursor { line_num, result.cursor }
@@ -141,8 +148,8 @@ function M.operator_visual(direction, stairlike)
     local function operate_line(lnum, range)
         local line = vim.fn.getline(lnum)
         local result = handler:operate_visual(line, range, direction, tier)
-        if result.line ~= nil then
-            vim.fn.setline(lnum, result.line)
+        if result.text ~= nil then
+            vim.api.nvim_buf_set_text(0, lnum - 1, result.range.from - 1, lnum - 1, result.range.to, { result.text })
             if stairlike then
                 tier = tier + 1
             end
@@ -217,8 +224,8 @@ function M.command(direction, line_range, groups)
     local function operate_line(lnum, range)
         local line = vim.fn.getline(lnum)
         local result = handler:operate_visual(line, range, direction, 1)
-        if result.line ~= nil then
-            vim.fn.setline(lnum, result.line)
+        if result.text ~= nil then
+            vim.api.nvim_buf_set_text(0, lnum - 1, result.range.from - 1, lnum - 1, result.range.to, { result.text })
         end
     end
 
diff --git a/lua/dial/handle.lua b/lua/dial/handle.lua
index e154e3a..96b4ebb 100644
--- a/lua/dial/handle.lua
+++ b/lua/dial/handle.lua
@@ -197,7 +197,7 @@ end
 ---@param cursor integer
 ---@param direction direction
 ---@param additive? boolean
----@return {line?: string, cursor?: integer}
+---@return {range?: textrange, text?: string, cursor?: integer}
 function Handler:operate(line, cursor, direction, additive)
     if self.range == nil or self.active_augend == nil then
         return {}
@@ -206,12 +206,8 @@ function Handler:operate(line, cursor, direction, additive)
     local text = line:sub(self.range.from, self.range.to)
     local addend = self:get_addend(direction)
     local add_result = self.active_augend:add(text, addend * (self.cumsum + 1), cursor)
-    local new_line = nil
     local new_cursor = nil
 
-    if add_result.text ~= nil then
-        new_line = line:sub(1, self.range.from - 1) .. add_result.text .. line:sub(self.range.to + 1)
-    end
     if add_result.cursor ~= nil then
         new_cursor = self.range.from - 1 + add_result.cursor
     end
@@ -220,14 +216,14 @@ function Handler:operate(line, cursor, direction, additive)
         self.cumsum = self.cumsum + 1
     end
 
-    return { line = new_line, cursor = new_cursor }
+    return { range = self.range, text = add_result.text, cursor = new_cursor }
 end
 
 ---The process that runs when operator is called (in VISUAL mode).
 ---@param selected_range {from: integer, to?: integer}
 ---@param direction direction
 ---@param tier integer
----@return {result?: string}
+---@return {range?: textrange, text?: string}
 function Handler:operate_visual(line, selected_range, direction, tier)
     if self.active_augend == nil then
         return {}
@@ -243,11 +239,7 @@ function Handler:operate_visual(line, selected_range, direction, tier)
     local to = selected_range.from + range.to - 1
     local text = line:sub(from, to)
     local add_result = self.active_augend:add(text, addend * tier)
-    local newline = nil
-    if add_result.text ~= nil then
-        newline = line:sub(1, from - 1) .. add_result.text .. line:sub(to + 1)
-    end
-    return { line = newline }
+    return { range = { from = from, to = to }, text = add_result.text }
 end
 
 ---Set self.range to the target range of the currently active augend (without side effects).