feat(move_item): preserve cursor position

PR: simrat39#386

Author: MunifTanjim

lua/rust-tools/move_item.lua

@@ -10,17 +10,53 @@ local function get_params(up)
   return params
 end
 
+local function extract_cursor_position(text_edits)
+  local cursor = { text_edits[1].range.start.line }
+
+  local prev_te
+  for _, te in ipairs(text_edits) do
+    if te.newText and te.insertTextFormat == 2 then
+      if not cursor[2] then
+        if prev_te then
+          cursor[1] = cursor[1]
+            + math.max(0, te.range.start.line - prev_te.range["end"].line - 1)
+            - (prev_te.range.start.line == te.range.start.line and 1 or 0)
+        end
+
+        local pos_start = string.find(te.newText, "%$0")
+        local lines = vim.split(string.sub(te.newText, 1, pos_start), "\n")
+        local total_lines = #lines
+
+        cursor[1] = cursor[1] + total_lines
+        if pos_start then
+          cursor[2] = (total_lines == 1 and te.range.start.character or 0)
+            + #lines[total_lines]
+            - 1
+        end
+      end
+
+      -- $0 -> Nothing
+      te.newText = string.gsub(te.newText, "%$%d", "")
+      -- ${0:_} -> _
+      te.newText = string.gsub(te.newText, "%${%d:(.-)}", "%1")
+    end
+    prev_te = te
+  end
+  return cursor
+end
+
 -- move it baby
 local function handler(_, result, ctx)
-  if result == nil then
+  if result == nil or #result == 0 then
     return
   end
-  rt.utils.snippet_text_edits_to_text_edits(result)
+  local cursor = extract_cursor_position(result)
   vim.lsp.util.apply_text_edits(
     result,
     ctx.bufnr,
     vim.lsp.get_client_by_id(ctx.client_id).offset_encoding
   )
+  vim.api.nvim_win_set_cursor(0, cursor)
 end
 
 -- Sends the request to rust-analyzer to move the item and handle the response