finder: add references_opts option

Author: wiserfz

lua/lspsaga/finder/init.lua

@@ -489,16 +489,25 @@ function fd:new(args)
     self.inexist = config.finder.sp_inexist
   end
   self.layout = layout or config.finder.layout
+
+  local curbuf = api.nvim_get_current_buf()
+  local params = lsp.util.make_position_params(0, util.get_offset_encoding({ bufnr = curbuf }))
+  params.context = {
+    includeDeclaration = true,
+  }
+  if type(config.finder.references_opts) == 'boolean' then
+    params.context = {
+      includeDeclaration = config.finder.references_opts,
+    }
+  end
+
   if #meth == 0 then
     meth = vim.split(config.finder.default, '+', { plain = true })
   end
   local methods = box.get_methods(meth)
-
   methods = vim.tbl_filter(function(method)
     return #util.get_client_by_method(method) > 0
   end, methods)
-  local curbuf = api.nvim_get_current_buf()
-  self.ft = vim.bo[curbuf].filetype
   if #methods == 0 then
     vim.notify(
       ('[lspsaga] no servers of buffer %s makes these methods available %s'):format(
@@ -510,23 +519,24 @@ function fd:new(args)
     return
   end
 
+  local methods_with_params = vim.tbl_map(function(method)
+    return util.gen_param_by_config(method, params, config.finder.references_opts)
+  end, methods)
+
+  self.ft = vim.bo[curbuf].filetype
   self.list = slist.new()
-  local params = lsp.util.make_position_params(0, util.get_offset_encoding({ bufnr = curbuf }))
-  params.context = {
-    includeDeclaration = true,
-  }
 
   local spin_close = box.spinner()
   local count = 0
   coroutine.resume(coroutine.create(function()
     local retval = {}
     local co = coroutine.running()
-    for _, method in ipairs(methods) do
-      lsp.buf_request_all(curbuf, method, params, function(results)
+    for _, item in ipairs(methods_with_params) do
+      lsp.buf_request_all(curbuf, item[1], item[2], function(results)
         count = count + 1
-        results = box.filter(method, results)
+        results = box.filter(item[1], results)
         if results and not util.res_isempty(results) then
-          retval[method] = results
+          retval[item[1]] = results
         end
         if count == #methods then
           coroutine.resume(co)

lua/lspsaga/init.lua

@@ -86,6 +86,7 @@ local default_config = {
     ly_botright = false,
     number = vim.o.number,
     relativenumber = vim.o.relativenumber,
+    references_opts = true,
     keys = {
       shuttle = '[w',
       toggle_or_open = 'o',

lua/lspsaga/util.lua

@@ -73,6 +73,35 @@ function M.get_client_by_method(method)
   return supports
 end
 
+--- generate LSP RPC method `textDocument/references` parameters by configuration.
+---@param method string LSP RPC method name
+---@param params table parameters for LSP RPC method
+---@param config table|boolean configuration for config.finder.references_opts
+---@return table
+function M.gen_param_by_config(method, params, config)
+  if type(config) == 'table' and method == 'textDocument/references' then
+    local bufnr = vim.api.nvim_get_current_buf()
+    local clients = lsp.get_clients({ bufnr = bufnr })
+
+    for _, client in ipairs(clients or {}) do
+      -- Replace characters dash('-') to underscore('_') for LSP client name,
+      -- which is useful for some LSP client like `rust-analyzer`.
+      local client_name = string.gsub(client.name, '-', '_')
+      local v = vim.tbl_get(config, client_name)
+      if v ~= nil then
+        params.context = {
+          includeDeclaration = v,
+        }
+        break
+      end
+    end
+
+    return { method, params }
+  end
+
+  return { method, params }
+end
+
 function M.feedkeys(key)
   local k = api.nvim_replace_termcodes(key, true, false, true)
   api.nvim_feedkeys(k, 'nx', false)