Как улучшить LSP поиск референсов в Neovim

vim.lsp.config("gopls", {
  cmd = { "gopls", "serve" },
  filetypes = { "go", "go.mod" },
  root_markers = { "go.work", "go.mod", ".git" },
})

vim.lsp.enable "gopls"

-- Прочие настройки
-- ...
require "lsp"

local utils = require "lsp.utils"

vim.lsp.config("gopls", {
  cmd = { "gopls", "serve" },
  filetypes = { "go", "go.mod" },
  root_markers = { "go.work", "go.mod", ".git" },
  on_init = utils.on_init,
  on_attach = utils.on_attach,
})

vim.lsp.enable "gopls"

local M = {}

local map = vim.keymap.set

M.on_init = function(client, _)
  if client.supports_method "textDocument/semanticTokens" then
    client.server_capabilities.semanticTokensProvider = nil
  end
end

M.on_attach = function(client, bufnr)
  local function map_opts(desc)
    return { buffer = bufnr, desc = "LSP: " .. desc }
  end

  local floating_window_opts = {
    border = "single",
    max_width = 80,
  }

  map("n", "gD", vim.lsp.buf.declaration, map_opts "go to declaration")
  map("n", "gd", vim.lsp.buf.definition, map_opts "go to definition")
  map("n", "gi", vim.lsp.buf.implementation, map_opts "go to implementation")
  map("n", "gj", vim.lsp.buf.type_definition, map_opts "go to type definition")

  map("n", "gk", function()
    require("lsp.handlers").references { includeDeclaration = false }
  end, map_opts "show references")

  map("n", "K", function()
    return vim.lsp.buf.hover(floating_window_opts)
  end, map_opts "hover")

  map("n", "<leader>ld", function()
    return vim.diagnostic.open_float(floating_window_opts)
  end, map_opts "diagnostic open float")

  map("n", "<leader>ls", function()
    return vim.lsp.buf.signature_help(floating_window_opts)
  end, map_opts "show signature help")

  map("i", "<c-s>", function()
    return vim.lsp.buf.signature_help(floating_window_opts)
  end, map_opts "show signature help")

  map("n", "<leader>lr", vim.lsp.buf.rename, map_opts "rename")

  map({ "n", "v" }, "<leader>la", vim.lsp.buf.code_action, map_opts "code action")
  map({ "n", "v" }, "<leader>ll", vim.lsp.codelens.run, map_opts "code lens")
end

return M

local M = {}

local util = require "vim.lsp.util"
local ms = require("vim.lsp.protocol").Methods

local filter_references = function(result)
  if result == nil or #result == 0 then
    return {}
  end

  local current_position = vim.api.nvim_win_get_cursor(0)
  local current_uri = vim.uri_from_bufnr(0)

  local filtered_result = {}

  for _, ref in ipairs(result) do
    if
      ref.user_data.uri ~= current_uri
      or ref.user_data.range.start.line ~= current_position[1] - 1
      or ref.user_data.range.start.character > current_position[2]
    then
      table.insert(filtered_result, ref)
    end
  end

  return filtered_result
end

local function on_done(client, items, bufnr)
  items = filter_references(items)

  if not next(items) then
    vim.notify "No references found"
    return
  end

  if #items == 1 then
    vim.lsp.util.show_document({
      uri = items[1].user_data.uri,
      range = items[1].user_data.range,
    }, client.offset_encoding, { reuse_win = true, focus = true })
    return
  end

  local list = {
    title = "References",
    items = items,
    context = {
      method = ms.textDocument_references,
      bufnr = bufnr,
    },
  }

  vim.fn.setloclist(0, {}, " ", list)
  vim.cmd.lopen()
end

M.references = function(context)
  local bufnr = vim.api.nvim_get_current_buf()
  local clients = vim.lsp.get_clients { method = ms.textDocument_references, bufnr = bufnr }
  if not next(clients) then
    return
  end

  local win = vim.api.nvim_get_current_win()

  local all_clients_items = {}

  local remaining = #clients
  for _, client in ipairs(clients) do
    local params = util.make_position_params(win, client.offset_encoding)

    ---@diagnostic disable-next-line: inject-field
    params.context = context or {
      includeDeclaration = true,
    }

    client:request(ms.textDocument_references, params, function(_, result)
      local client_items = util.locations_to_items(result or {}, client.offset_encoding)
      vim.list_extend(all_clients_items, client_items)
      remaining = remaining - 1
      if remaining == 0 then
        on_done(client, all_clients_items, bufnr)
      end
    end)
  end
end

return M