feat: use luasnip snippet text edit handler if available

Author: mrcjkb

CHANGELOG.md

@@ -8,6 +8,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Added
+
+- LSP: Use LuaSnip's [`SnippetTextEdit`](https://github.com/rust-lang/rust-analyzer/blob/master/docs/dev/lsp-extensions.md#snippet-textedit)
+  handler if available.
+  This enables snippet text edits in code actions.
+
 ### Fixed
 
 - Error when decoding invalid JSON or blank string from cargo metadata.

lua/rustaceanvim/config/internal.lua

@@ -84,6 +84,14 @@ local function get_test_executor()
   return get_crate_test_executor()
 end
 
+---@return fun(edits: rust.lsp.SnippetTextEdit[], bufnr: integer, offset_encoding: string, apply_text_edits: function) | nil
+local function get_snippet_text_edit_handler()
+  local ok, luasnip = pcall(require, 'luasnip.extras.lsp')
+  if ok and luasnip.apply_text_edits then
+    return luasnip.apply_text_edits
+  end
+end
+
 ---@class RustaceanConfig
 local RustaceanDefaultConfig = {
   ---@class RustaceanToolsConfig
@@ -100,6 +108,9 @@ local RustaceanDefaultConfig = {
     ---@type RustaceanExecutor
     crate_test_executor = get_crate_test_executor(),
 
+    ---@type fun(edits: rust.lsp.SnippetTextEdit[], bufnr: integer, offset_encoding: string, apply_text_edits: function) | nil
+    snippet_text_edit_handler = get_snippet_text_edit_handler(),
+
     ---@type string | nil
     cargo_override = nil,
 

lua/rustaceanvim/lsp.lua

@@ -11,11 +11,21 @@ local os = require('rustaceanvim.os')
 local function override_apply_text_edits()
   local old_func = vim.lsp.util.apply_text_edits
   ---@diagnostic disable-next-line
-  vim.lsp.util.apply_text_edits = function(edits, bufnr, offset_encoding)
-    local overrides = require('rustaceanvim.overrides')
-    overrides.snippet_text_edits_to_text_edits(edits)
-    old_func(edits, bufnr, offset_encoding)
-  end
+  vim.lsp.util.apply_text_edits = config.tools.snippet_text_edit_handler
+      ---@param edits rust.lsp.SnippetTextEdit[]
+      ---@param bufnr number
+      ---@param offset_encoding string
+      and function(edits, bufnr, offset_encoding)
+        config.tools.snippet_text_edit_handler(edits, bufnr, offset_encoding, old_func)
+      end
+    ---@param edits rust.lsp.SnippetTextEdit[]
+    ---@param bufnr number
+    ---@param offset_encoding string
+    or function(edits, bufnr, offset_encoding)
+      local overrides = require('rustaceanvim.overrides')
+      overrides.snippet_text_edits_to_text_edits(edits)
+      old_func(edits, bufnr, offset_encoding)
+    end
 end
 
 ---@param client lsp.Client

lua/rustaceanvim/meta.lua

@@ -0,0 +1,4 @@
+---@class rust.lsp.SnippetTextEdit
+---@field insertTextFormat number
+---@field newText string
+---@field range table see |vim.lsp.util.make_range_params()|

lua/rustaceanvim/overrides.lua

@@ -23,7 +23,7 @@ local function parse_snippet(input)
   return ok and tostring(parsed) or parse_snippet_fallback(input)
 end
 
----@param spe? table
+---@param spe? rust.lsp.SnippetTextEdit[]
 function M.snippet_text_edits_to_text_edits(spe)
   if type(spe) ~= 'table' then
     return