mirror of
				https://github.com/neovim/neovim.git
				synced 2025-11-04 09:44:31 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			1315 lines
		
	
	
		
			43 KiB
		
	
	
	
		
			Lua
		
	
	
	
	
	
			
		
		
	
	
			1315 lines
		
	
	
		
			43 KiB
		
	
	
	
		
			Lua
		
	
	
	
	
	
--- @brief
 | 
						|
--- The `vim.lsp.buf_…` functions perform operations for LSP clients attached to the current buffer.
 | 
						|
 | 
						|
local api = vim.api
 | 
						|
local lsp = vim.lsp
 | 
						|
local validate = vim.validate
 | 
						|
local util = require('vim.lsp.util')
 | 
						|
local npcall = vim.F.npcall
 | 
						|
local ms = require('vim.lsp.protocol').Methods
 | 
						|
 | 
						|
local M = {}
 | 
						|
 | 
						|
--- @param params? table
 | 
						|
--- @return fun(client: vim.lsp.Client): lsp.TextDocumentPositionParams
 | 
						|
local function client_positional_params(params)
 | 
						|
  local win = api.nvim_get_current_win()
 | 
						|
  return function(client)
 | 
						|
    local ret = util.make_position_params(win, client.offset_encoding)
 | 
						|
    if params then
 | 
						|
      ret = vim.tbl_extend('force', ret, params)
 | 
						|
    end
 | 
						|
    return ret
 | 
						|
  end
 | 
						|
end
 | 
						|
 | 
						|
local hover_ns = api.nvim_create_namespace('nvim.lsp.hover_range')
 | 
						|
 | 
						|
--- @class vim.lsp.buf.hover.Opts : vim.lsp.util.open_floating_preview.Opts
 | 
						|
--- @field silent? boolean
 | 
						|
 | 
						|
--- Displays hover information about the symbol under the cursor in a floating
 | 
						|
--- window. The window will be dismissed on cursor move.
 | 
						|
--- Calling the function twice will jump into the floating window
 | 
						|
--- (thus by default, "KK" will open the hover window and focus it).
 | 
						|
--- In the floating window, all commands and mappings are available as usual,
 | 
						|
--- except that "q" dismisses the window.
 | 
						|
--- You can scroll the contents the same as you would any other buffer.
 | 
						|
---
 | 
						|
--- Note: to disable hover highlights, add the following to your config:
 | 
						|
---
 | 
						|
--- ```lua
 | 
						|
--- vim.api.nvim_create_autocmd('ColorScheme', {
 | 
						|
---   callback = function()
 | 
						|
---     vim.api.nvim_set_hl(0, 'LspReferenceTarget', {})
 | 
						|
---   end,
 | 
						|
--- })
 | 
						|
--- ```
 | 
						|
--- @param config? vim.lsp.buf.hover.Opts
 | 
						|
function M.hover(config)
 | 
						|
  config = config or {}
 | 
						|
  config.focus_id = ms.textDocument_hover
 | 
						|
 | 
						|
  lsp.buf_request_all(0, ms.textDocument_hover, client_positional_params(), function(results, ctx)
 | 
						|
    local bufnr = assert(ctx.bufnr)
 | 
						|
    if api.nvim_get_current_buf() ~= bufnr then
 | 
						|
      -- Ignore result since buffer changed. This happens for slow language servers.
 | 
						|
      return
 | 
						|
    end
 | 
						|
 | 
						|
    -- Filter errors from results
 | 
						|
    local results1 = {} --- @type table<integer,lsp.Hover>
 | 
						|
 | 
						|
    for client_id, resp in pairs(results) do
 | 
						|
      local err, result = resp.err, resp.result
 | 
						|
      if err then
 | 
						|
        lsp.log.error(err.code, err.message)
 | 
						|
      elseif result then
 | 
						|
        results1[client_id] = result
 | 
						|
      end
 | 
						|
    end
 | 
						|
 | 
						|
    if vim.tbl_isempty(results1) then
 | 
						|
      if config.silent ~= true then
 | 
						|
        vim.notify('No information available')
 | 
						|
      end
 | 
						|
      return
 | 
						|
    end
 | 
						|
 | 
						|
    local contents = {} --- @type string[]
 | 
						|
 | 
						|
    local nresults = #vim.tbl_keys(results1)
 | 
						|
 | 
						|
    local format = 'markdown'
 | 
						|
 | 
						|
    for client_id, result in pairs(results1) do
 | 
						|
      local client = assert(lsp.get_client_by_id(client_id))
 | 
						|
      if nresults > 1 then
 | 
						|
        -- Show client name if there are multiple clients
 | 
						|
        contents[#contents + 1] = string.format('# %s', client.name)
 | 
						|
      end
 | 
						|
      if type(result.contents) == 'table' and result.contents.kind == 'plaintext' then
 | 
						|
        if #results1 == 1 then
 | 
						|
          format = 'plaintext'
 | 
						|
          contents = vim.split(result.contents.value or '', '\n', { trimempty = true })
 | 
						|
        else
 | 
						|
          -- Surround plaintext with ``` to get correct formatting
 | 
						|
          contents[#contents + 1] = '```'
 | 
						|
          vim.list_extend(
 | 
						|
            contents,
 | 
						|
            vim.split(result.contents.value or '', '\n', { trimempty = true })
 | 
						|
          )
 | 
						|
          contents[#contents + 1] = '```'
 | 
						|
        end
 | 
						|
      else
 | 
						|
        vim.list_extend(contents, util.convert_input_to_markdown_lines(result.contents))
 | 
						|
      end
 | 
						|
      local range = result.range
 | 
						|
      if range then
 | 
						|
        local start = range.start
 | 
						|
        local end_ = range['end']
 | 
						|
        local start_idx = util._get_line_byte_from_position(bufnr, start, client.offset_encoding)
 | 
						|
        local end_idx = util._get_line_byte_from_position(bufnr, end_, client.offset_encoding)
 | 
						|
 | 
						|
        vim.hl.range(
 | 
						|
          bufnr,
 | 
						|
          hover_ns,
 | 
						|
          'LspReferenceTarget',
 | 
						|
          { start.line, start_idx },
 | 
						|
          { end_.line, end_idx },
 | 
						|
          { priority = vim.hl.priorities.user }
 | 
						|
        )
 | 
						|
      end
 | 
						|
      contents[#contents + 1] = '---'
 | 
						|
    end
 | 
						|
 | 
						|
    -- Remove last linebreak ('---')
 | 
						|
    contents[#contents] = nil
 | 
						|
 | 
						|
    if vim.tbl_isempty(contents) then
 | 
						|
      if config.silent ~= true then
 | 
						|
        vim.notify('No information available')
 | 
						|
      end
 | 
						|
      return
 | 
						|
    end
 | 
						|
 | 
						|
    local _, winid = lsp.util.open_floating_preview(contents, format, config)
 | 
						|
 | 
						|
    api.nvim_create_autocmd('WinClosed', {
 | 
						|
      pattern = tostring(winid),
 | 
						|
      once = true,
 | 
						|
      callback = function()
 | 
						|
        api.nvim_buf_clear_namespace(bufnr, hover_ns, 0, -1)
 | 
						|
        return true
 | 
						|
      end,
 | 
						|
    })
 | 
						|
  end)
 | 
						|
end
 | 
						|
 | 
						|
local function request_with_opts(name, params, opts)
 | 
						|
  local req_handler --- @type function?
 | 
						|
  if opts then
 | 
						|
    req_handler = function(err, result, ctx, config)
 | 
						|
      local client = assert(lsp.get_client_by_id(ctx.client_id))
 | 
						|
      local handler = client.handlers[name] or lsp.handlers[name]
 | 
						|
      handler(err, result, ctx, vim.tbl_extend('force', config or {}, opts))
 | 
						|
    end
 | 
						|
  end
 | 
						|
  lsp.buf_request(0, name, params, req_handler)
 | 
						|
end
 | 
						|
 | 
						|
---@param method vim.lsp.protocol.Method.ClientToServer.Request
 | 
						|
---@param opts? vim.lsp.LocationOpts
 | 
						|
local function get_locations(method, opts)
 | 
						|
  opts = opts or {}
 | 
						|
  local bufnr = api.nvim_get_current_buf()
 | 
						|
  local clients = lsp.get_clients({ method = method, bufnr = bufnr })
 | 
						|
  if not next(clients) then
 | 
						|
    vim.notify(lsp._unsupported_method(method), vim.log.levels.WARN)
 | 
						|
    return
 | 
						|
  end
 | 
						|
  local win = api.nvim_get_current_win()
 | 
						|
  local from = vim.fn.getpos('.')
 | 
						|
  from[1] = bufnr
 | 
						|
  local tagname = vim.fn.expand('<cword>')
 | 
						|
  local remaining = #clients
 | 
						|
 | 
						|
  ---@type vim.quickfix.entry[]
 | 
						|
  local all_items = {}
 | 
						|
 | 
						|
  ---@param result nil|lsp.Location|lsp.Location[]
 | 
						|
  ---@param client vim.lsp.Client
 | 
						|
  local function on_response(_, result, client)
 | 
						|
    local locations = {}
 | 
						|
    if result then
 | 
						|
      locations = vim.islist(result) and result or { result }
 | 
						|
    end
 | 
						|
    local items = util.locations_to_items(locations, client.offset_encoding)
 | 
						|
    vim.list_extend(all_items, items)
 | 
						|
    remaining = remaining - 1
 | 
						|
    if remaining == 0 then
 | 
						|
      if vim.tbl_isempty(all_items) then
 | 
						|
        vim.notify('No locations found', vim.log.levels.INFO)
 | 
						|
        return
 | 
						|
      end
 | 
						|
 | 
						|
      local title = 'LSP locations'
 | 
						|
      if opts.on_list then
 | 
						|
        assert(vim.is_callable(opts.on_list), 'on_list is not a function')
 | 
						|
        opts.on_list({
 | 
						|
          title = title,
 | 
						|
          items = all_items,
 | 
						|
          context = { bufnr = bufnr, method = method },
 | 
						|
        })
 | 
						|
        return
 | 
						|
      end
 | 
						|
 | 
						|
      if #all_items == 1 then
 | 
						|
        local item = all_items[1]
 | 
						|
        local b = item.bufnr or vim.fn.bufadd(item.filename)
 | 
						|
 | 
						|
        -- Save position in jumplist
 | 
						|
        vim.cmd("normal! m'")
 | 
						|
        -- Push a new item into tagstack
 | 
						|
        local tagstack = { { tagname = tagname, from = from } }
 | 
						|
        vim.fn.settagstack(vim.fn.win_getid(win), { items = tagstack }, 't')
 | 
						|
 | 
						|
        vim.bo[b].buflisted = true
 | 
						|
        local w = win
 | 
						|
        if opts.reuse_win then
 | 
						|
          w = vim.fn.win_findbuf(b)[1] or w
 | 
						|
          if w ~= win then
 | 
						|
            api.nvim_set_current_win(w)
 | 
						|
          end
 | 
						|
        end
 | 
						|
        api.nvim_win_set_buf(w, b)
 | 
						|
        api.nvim_win_set_cursor(w, { item.lnum, item.col - 1 })
 | 
						|
        vim._with({ win = w }, function()
 | 
						|
          -- Open folds under the cursor
 | 
						|
          vim.cmd('normal! zv')
 | 
						|
        end)
 | 
						|
        return
 | 
						|
      end
 | 
						|
      if opts.loclist then
 | 
						|
        vim.fn.setloclist(0, {}, ' ', { title = title, items = all_items })
 | 
						|
        vim.cmd.lopen()
 | 
						|
      else
 | 
						|
        vim.fn.setqflist({}, ' ', { title = title, items = all_items })
 | 
						|
        vim.cmd('botright copen')
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
  for _, client in ipairs(clients) do
 | 
						|
    local params = util.make_position_params(win, client.offset_encoding)
 | 
						|
    client:request(method, params, function(_, result)
 | 
						|
      on_response(_, result, client)
 | 
						|
    end)
 | 
						|
  end
 | 
						|
end
 | 
						|
 | 
						|
--- @class vim.lsp.ListOpts
 | 
						|
---
 | 
						|
--- list-handler replacing the default handler.
 | 
						|
--- Called for any non-empty result.
 | 
						|
--- This table can be used with |setqflist()| or |setloclist()|. E.g.:
 | 
						|
--- ```lua
 | 
						|
--- local function on_list(options)
 | 
						|
---   vim.fn.setqflist({}, ' ', options)
 | 
						|
---   vim.cmd.cfirst()
 | 
						|
--- end
 | 
						|
---
 | 
						|
--- vim.lsp.buf.definition({ on_list = on_list })
 | 
						|
--- vim.lsp.buf.references(nil, { on_list = on_list })
 | 
						|
--- ```
 | 
						|
--- @field on_list? fun(t: vim.lsp.LocationOpts.OnList)
 | 
						|
---
 | 
						|
--- Whether to use the |location-list| or the |quickfix| list in the default handler.
 | 
						|
--- ```lua
 | 
						|
--- vim.lsp.buf.definition({ loclist = true })
 | 
						|
--- vim.lsp.buf.references(nil, { loclist = false })
 | 
						|
--- ```
 | 
						|
--- @field loclist? boolean
 | 
						|
 | 
						|
--- @class vim.lsp.LocationOpts.OnList
 | 
						|
--- @field items table[] Structured like |setqflist-what|
 | 
						|
--- @field title? string Title for the list.
 | 
						|
--- @field context? { bufnr: integer, method: string } Subset of `ctx` from |lsp-handler|.
 | 
						|
 | 
						|
--- @class vim.lsp.LocationOpts: vim.lsp.ListOpts
 | 
						|
---
 | 
						|
--- Jump to existing window if buffer is already open.
 | 
						|
--- @field reuse_win? boolean
 | 
						|
 | 
						|
--- Jumps to the declaration of the symbol under the cursor.
 | 
						|
--- @note Many servers do not implement this method. Generally, see |vim.lsp.buf.definition()| instead.
 | 
						|
--- @param opts? vim.lsp.LocationOpts
 | 
						|
function M.declaration(opts)
 | 
						|
  get_locations(ms.textDocument_declaration, opts)
 | 
						|
end
 | 
						|
 | 
						|
--- Jumps to the definition of the symbol under the cursor.
 | 
						|
--- @param opts? vim.lsp.LocationOpts
 | 
						|
function M.definition(opts)
 | 
						|
  get_locations(ms.textDocument_definition, opts)
 | 
						|
end
 | 
						|
 | 
						|
--- Jumps to the definition of the type of the symbol under the cursor.
 | 
						|
--- @param opts? vim.lsp.LocationOpts
 | 
						|
function M.type_definition(opts)
 | 
						|
  get_locations(ms.textDocument_typeDefinition, opts)
 | 
						|
end
 | 
						|
 | 
						|
--- Lists all the implementations for the symbol under the cursor in the
 | 
						|
--- quickfix window.
 | 
						|
--- @param opts? vim.lsp.LocationOpts
 | 
						|
function M.implementation(opts)
 | 
						|
  get_locations(ms.textDocument_implementation, opts)
 | 
						|
end
 | 
						|
 | 
						|
--- @param results table<integer,{err: lsp.ResponseError?, result: lsp.SignatureHelp?}>
 | 
						|
local function process_signature_help_results(results)
 | 
						|
  local signatures = {} --- @type [vim.lsp.Client,lsp.SignatureInformation][]
 | 
						|
  local active_signature = 1
 | 
						|
 | 
						|
  -- Pre-process results
 | 
						|
  for client_id, r in pairs(results) do
 | 
						|
    local err = r.err
 | 
						|
    local client = assert(lsp.get_client_by_id(client_id))
 | 
						|
    if err then
 | 
						|
      vim.notify(
 | 
						|
        client.name .. ': ' .. tostring(err.code) .. ': ' .. err.message,
 | 
						|
        vim.log.levels.ERROR
 | 
						|
      )
 | 
						|
      api.nvim_command('redraw')
 | 
						|
    else
 | 
						|
      local result = r.result --- @type lsp.SignatureHelp
 | 
						|
      if result and result.signatures and result.signatures[1] then
 | 
						|
        for i, sig in ipairs(result.signatures) do
 | 
						|
          sig.activeParameter = sig.activeParameter or result.activeParameter
 | 
						|
          local idx = #signatures + 1
 | 
						|
          if (result.activeSignature or 0) + 1 == i then
 | 
						|
            active_signature = idx
 | 
						|
          end
 | 
						|
          signatures[idx] = { client, sig }
 | 
						|
        end
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  return signatures, active_signature
 | 
						|
end
 | 
						|
 | 
						|
local sig_help_ns = api.nvim_create_namespace('nvim.lsp.signature_help')
 | 
						|
 | 
						|
--- @class vim.lsp.buf.signature_help.Opts : vim.lsp.util.open_floating_preview.Opts
 | 
						|
--- @field silent? boolean
 | 
						|
 | 
						|
--- Displays signature information about the symbol under the cursor in a
 | 
						|
--- floating window.
 | 
						|
--- @param config? vim.lsp.buf.signature_help.Opts
 | 
						|
function M.signature_help(config)
 | 
						|
  local method = ms.textDocument_signatureHelp
 | 
						|
 | 
						|
  config = config and vim.deepcopy(config) or {}
 | 
						|
  config.focus_id = method
 | 
						|
 | 
						|
  lsp.buf_request_all(0, method, client_positional_params(), function(results, ctx)
 | 
						|
    if api.nvim_get_current_buf() ~= ctx.bufnr then
 | 
						|
      -- Ignore result since buffer changed. This happens for slow language servers.
 | 
						|
      return
 | 
						|
    end
 | 
						|
 | 
						|
    local signatures, active_signature = process_signature_help_results(results)
 | 
						|
 | 
						|
    if not next(signatures) then
 | 
						|
      if config.silent ~= true then
 | 
						|
        print('No signature help available')
 | 
						|
      end
 | 
						|
      return
 | 
						|
    end
 | 
						|
 | 
						|
    local ft = vim.bo[ctx.bufnr].filetype
 | 
						|
    local total = #signatures
 | 
						|
    local can_cycle = total > 1 and config.focusable ~= false
 | 
						|
    local idx = active_signature - 1
 | 
						|
 | 
						|
    --- @param update_win? integer
 | 
						|
    local function show_signature(update_win)
 | 
						|
      idx = (idx % total) + 1
 | 
						|
      local client, result = signatures[idx][1], signatures[idx][2]
 | 
						|
      --- @type string[]?
 | 
						|
      local triggers =
 | 
						|
        vim.tbl_get(client.server_capabilities, 'signatureHelpProvider', 'triggerCharacters')
 | 
						|
      local lines, hl =
 | 
						|
        util.convert_signature_help_to_markdown_lines({ signatures = { result } }, ft, triggers)
 | 
						|
      if not lines then
 | 
						|
        return
 | 
						|
      end
 | 
						|
 | 
						|
      local sfx = total > 1
 | 
						|
          and string.format(' (%d/%d)%s', idx, total, can_cycle and ' (<C-s> to cycle)' or '')
 | 
						|
        or ''
 | 
						|
      local title = string.format('Signature Help: %s%s', client.name, sfx)
 | 
						|
      if config.border then
 | 
						|
        config.title = title
 | 
						|
      else
 | 
						|
        table.insert(lines, 1, '# ' .. title)
 | 
						|
        if hl then
 | 
						|
          hl[1] = hl[1] + 1
 | 
						|
          hl[3] = hl[3] + 1
 | 
						|
        end
 | 
						|
      end
 | 
						|
 | 
						|
      config._update_win = update_win
 | 
						|
 | 
						|
      local buf, win = util.open_floating_preview(lines, 'markdown', config)
 | 
						|
 | 
						|
      if hl then
 | 
						|
        vim.api.nvim_buf_clear_namespace(buf, sig_help_ns, 0, -1)
 | 
						|
        vim.hl.range(
 | 
						|
          buf,
 | 
						|
          sig_help_ns,
 | 
						|
          'LspSignatureActiveParameter',
 | 
						|
          { hl[1], hl[2] },
 | 
						|
          { hl[3], hl[4] }
 | 
						|
        )
 | 
						|
      end
 | 
						|
      return buf, win
 | 
						|
    end
 | 
						|
 | 
						|
    local fbuf, fwin = show_signature()
 | 
						|
 | 
						|
    if can_cycle then
 | 
						|
      vim.keymap.set('n', '<C-s>', function()
 | 
						|
        show_signature(fwin)
 | 
						|
      end, {
 | 
						|
        buffer = fbuf,
 | 
						|
        desc = 'Cycle next signature',
 | 
						|
      })
 | 
						|
    end
 | 
						|
  end)
 | 
						|
end
 | 
						|
 | 
						|
--- @deprecated
 | 
						|
--- Retrieves the completion items at the current cursor position. Can only be
 | 
						|
--- called in Insert mode.
 | 
						|
---
 | 
						|
---@param context table (context support not yet implemented) Additional information
 | 
						|
--- about the context in which a completion was triggered (how it was triggered,
 | 
						|
--- and by which trigger character, if applicable)
 | 
						|
---
 | 
						|
---@see vim.lsp.protocol.CompletionTriggerKind
 | 
						|
function M.completion(context)
 | 
						|
  vim.depends('vim.lsp.buf.completion', 'vim.lsp.completion.trigger', '0.12')
 | 
						|
  return lsp.buf_request(
 | 
						|
    0,
 | 
						|
    ms.textDocument_completion,
 | 
						|
    client_positional_params({
 | 
						|
      context = context,
 | 
						|
    })
 | 
						|
  )
 | 
						|
end
 | 
						|
 | 
						|
---@param bufnr integer
 | 
						|
---@param mode "v"|"V"
 | 
						|
---@return table {start={row,col}, end={row,col}} using (1, 0) indexing
 | 
						|
local function range_from_selection(bufnr, mode)
 | 
						|
  -- TODO: Use `vim.fn.getregionpos()` instead.
 | 
						|
 | 
						|
  -- [bufnum, lnum, col, off]; both row and column 1-indexed
 | 
						|
  local start = vim.fn.getpos('v')
 | 
						|
  local end_ = vim.fn.getpos('.')
 | 
						|
  local start_row = start[2]
 | 
						|
  local start_col = start[3]
 | 
						|
  local end_row = end_[2]
 | 
						|
  local end_col = end_[3]
 | 
						|
 | 
						|
  -- A user can start visual selection at the end and move backwards
 | 
						|
  -- Normalize the range to start < end
 | 
						|
  if start_row == end_row and end_col < start_col then
 | 
						|
    end_col, start_col = start_col, end_col --- @type integer, integer
 | 
						|
  elseif end_row < start_row then
 | 
						|
    start_row, end_row = end_row, start_row --- @type integer, integer
 | 
						|
    start_col, end_col = end_col, start_col --- @type integer, integer
 | 
						|
  end
 | 
						|
  if mode == 'V' then
 | 
						|
    start_col = 1
 | 
						|
    local lines = api.nvim_buf_get_lines(bufnr, end_row - 1, end_row, true)
 | 
						|
    end_col = #lines[1]
 | 
						|
  end
 | 
						|
  return {
 | 
						|
    ['start'] = { start_row, start_col - 1 },
 | 
						|
    ['end'] = { end_row, end_col - 1 },
 | 
						|
  }
 | 
						|
end
 | 
						|
 | 
						|
--- @class vim.lsp.buf.format.Opts
 | 
						|
--- @inlinedoc
 | 
						|
---
 | 
						|
--- Can be used to specify FormattingOptions. Some unspecified options will be
 | 
						|
--- automatically derived from the current Nvim options.
 | 
						|
--- See https://microsoft.github.io/language-server-protocol/specification/#formattingOptions
 | 
						|
--- @field formatting_options? table
 | 
						|
---
 | 
						|
--- Time in milliseconds to block for formatting requests. No effect if async=true.
 | 
						|
--- (default: `1000`)
 | 
						|
--- @field timeout_ms? integer
 | 
						|
---
 | 
						|
--- Restrict formatting to the clients attached to the given buffer.
 | 
						|
--- (default: current buffer)
 | 
						|
--- @field bufnr? integer
 | 
						|
---
 | 
						|
--- Predicate used to filter clients. Receives a client as argument and must
 | 
						|
--- return a boolean. Clients matching the predicate are included. Example:
 | 
						|
--- ```lua
 | 
						|
--- -- Never request typescript-language-server for formatting
 | 
						|
--- vim.lsp.buf.format {
 | 
						|
---   filter = function(client) return client.name ~= "ts_ls" end
 | 
						|
--- }
 | 
						|
--- ```
 | 
						|
--- @field filter? fun(client: vim.lsp.Client): boolean?
 | 
						|
---
 | 
						|
--- If true the method won't block.
 | 
						|
--- Editing the buffer while formatting asynchronous can lead to unexpected
 | 
						|
--- changes.
 | 
						|
--- (Default: false)
 | 
						|
--- @field async? boolean
 | 
						|
---
 | 
						|
--- Restrict formatting to the client with ID (client.id) matching this field.
 | 
						|
--- @field id? integer
 | 
						|
---
 | 
						|
--- Restrict formatting to the client with name (client.name) matching this field.
 | 
						|
--- @field name? string
 | 
						|
---
 | 
						|
--- Range to format.
 | 
						|
--- Table must contain `start` and `end` keys with {row,col} tuples using
 | 
						|
--- (1,0) indexing.
 | 
						|
--- Can also be a list of tables that contain `start` and `end` keys as described above,
 | 
						|
--- in which case `textDocument/rangesFormatting` support is required.
 | 
						|
--- (Default: current selection in visual mode, `nil` in other modes,
 | 
						|
--- formatting the full buffer)
 | 
						|
--- @field range? {start:[integer,integer],end:[integer, integer]}|{start:[integer,integer],end:[integer,integer]}[]
 | 
						|
 | 
						|
--- Formats a buffer using the attached (and optionally filtered) language
 | 
						|
--- server clients.
 | 
						|
---
 | 
						|
--- @param opts? vim.lsp.buf.format.Opts
 | 
						|
function M.format(opts)
 | 
						|
  opts = opts or {}
 | 
						|
  local bufnr = vim._resolve_bufnr(opts.bufnr)
 | 
						|
  local mode = api.nvim_get_mode().mode
 | 
						|
  local range = opts.range
 | 
						|
  -- Try to use visual selection if no range is given
 | 
						|
  if not range and mode == 'v' or mode == 'V' then
 | 
						|
    range = range_from_selection(bufnr, mode)
 | 
						|
  end
 | 
						|
 | 
						|
  local passed_multiple_ranges = (range and #range ~= 0 and type(range[1]) == 'table')
 | 
						|
  local method ---@type string
 | 
						|
  if passed_multiple_ranges then
 | 
						|
    method = ms.textDocument_rangesFormatting
 | 
						|
  elseif range then
 | 
						|
    method = ms.textDocument_rangeFormatting
 | 
						|
  else
 | 
						|
    method = ms.textDocument_formatting
 | 
						|
  end
 | 
						|
 | 
						|
  local clients = lsp.get_clients({
 | 
						|
    id = opts.id,
 | 
						|
    bufnr = bufnr,
 | 
						|
    name = opts.name,
 | 
						|
    method = method,
 | 
						|
  })
 | 
						|
  if opts.filter then
 | 
						|
    clients = vim.tbl_filter(opts.filter, clients)
 | 
						|
  end
 | 
						|
 | 
						|
  if #clients == 0 then
 | 
						|
    vim.notify('[LSP] Format request failed, no matching language servers.')
 | 
						|
  end
 | 
						|
 | 
						|
  --- @param client vim.lsp.Client
 | 
						|
  --- @param params lsp.DocumentFormattingParams
 | 
						|
  --- @return lsp.DocumentFormattingParams|lsp.DocumentRangeFormattingParams|lsp.DocumentRangesFormattingParams
 | 
						|
  local function set_range(client, params)
 | 
						|
    ---  @param r {start:[integer,integer],end:[integer, integer]}
 | 
						|
    local function to_lsp_range(r)
 | 
						|
      return util.make_given_range_params(r.start, r['end'], bufnr, client.offset_encoding).range
 | 
						|
    end
 | 
						|
 | 
						|
    local ret = params --[[@as lsp.DocumentFormattingParams|lsp.DocumentRangeFormattingParams|lsp.DocumentRangesFormattingParams]]
 | 
						|
    if passed_multiple_ranges then
 | 
						|
      ret = params --[[@as lsp.DocumentRangesFormattingParams]]
 | 
						|
      --- @cast range {start:[integer,integer],end:[integer, integer]}
 | 
						|
      ret.ranges = vim.tbl_map(to_lsp_range, range)
 | 
						|
    elseif range then
 | 
						|
      ret = params --[[@as lsp.DocumentRangeFormattingParams]]
 | 
						|
      ret.range = to_lsp_range(range)
 | 
						|
    end
 | 
						|
    return ret
 | 
						|
  end
 | 
						|
 | 
						|
  if opts.async then
 | 
						|
    --- @param idx? integer
 | 
						|
    --- @param client? vim.lsp.Client
 | 
						|
    local function do_format(idx, client)
 | 
						|
      if not idx or not client then
 | 
						|
        return
 | 
						|
      end
 | 
						|
      local params = set_range(client, util.make_formatting_params(opts.formatting_options))
 | 
						|
      client:request(method, params, function(...)
 | 
						|
        local handler = client.handlers[method] or lsp.handlers[method]
 | 
						|
        handler(...)
 | 
						|
        do_format(next(clients, idx))
 | 
						|
      end, bufnr)
 | 
						|
    end
 | 
						|
    do_format(next(clients))
 | 
						|
  else
 | 
						|
    local timeout_ms = opts.timeout_ms or 1000
 | 
						|
    for _, client in pairs(clients) do
 | 
						|
      local params = set_range(client, util.make_formatting_params(opts.formatting_options))
 | 
						|
      local result, err = client:request_sync(method, params, timeout_ms, bufnr)
 | 
						|
      if result and result.result then
 | 
						|
        util.apply_text_edits(result.result, bufnr, client.offset_encoding)
 | 
						|
      elseif err then
 | 
						|
        vim.notify(string.format('[LSP][%s] %s', client.name, err), vim.log.levels.WARN)
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
end
 | 
						|
 | 
						|
--- @class vim.lsp.buf.rename.Opts
 | 
						|
--- @inlinedoc
 | 
						|
---
 | 
						|
--- Predicate used to filter clients. Receives a client as argument and
 | 
						|
--- must return a boolean. Clients matching the predicate are included.
 | 
						|
--- @field filter? fun(client: vim.lsp.Client): boolean?
 | 
						|
---
 | 
						|
--- Restrict clients used for rename to ones where client.name matches
 | 
						|
--- this field.
 | 
						|
--- @field name? string
 | 
						|
---
 | 
						|
--- (default: current buffer)
 | 
						|
--- @field bufnr? integer
 | 
						|
 | 
						|
--- Renames all references to the symbol under the cursor.
 | 
						|
---
 | 
						|
---@param new_name string|nil If not provided, the user will be prompted for a new
 | 
						|
---                name using |vim.ui.input()|.
 | 
						|
---@param opts? vim.lsp.buf.rename.Opts Additional options:
 | 
						|
function M.rename(new_name, opts)
 | 
						|
  opts = opts or {}
 | 
						|
  local bufnr = vim._resolve_bufnr(opts.bufnr)
 | 
						|
  local clients = lsp.get_clients({
 | 
						|
    bufnr = bufnr,
 | 
						|
    name = opts.name,
 | 
						|
    -- Clients must at least support rename, prepareRename is optional
 | 
						|
    method = ms.textDocument_rename,
 | 
						|
  })
 | 
						|
  if opts.filter then
 | 
						|
    clients = vim.tbl_filter(opts.filter, clients)
 | 
						|
  end
 | 
						|
 | 
						|
  if #clients == 0 then
 | 
						|
    vim.notify('[LSP] Rename, no matching language servers with rename capability.')
 | 
						|
  end
 | 
						|
 | 
						|
  local win = api.nvim_get_current_win()
 | 
						|
 | 
						|
  -- Compute early to account for cursor movements after going async
 | 
						|
  local cword = vim.fn.expand('<cword>')
 | 
						|
 | 
						|
  --- @param range lsp.Range
 | 
						|
  --- @param position_encoding string
 | 
						|
  local function get_text_at_range(range, position_encoding)
 | 
						|
    return api.nvim_buf_get_text(
 | 
						|
      bufnr,
 | 
						|
      range.start.line,
 | 
						|
      util._get_line_byte_from_position(bufnr, range.start, position_encoding),
 | 
						|
      range['end'].line,
 | 
						|
      util._get_line_byte_from_position(bufnr, range['end'], position_encoding),
 | 
						|
      {}
 | 
						|
    )[1]
 | 
						|
  end
 | 
						|
 | 
						|
  --- @param idx? integer
 | 
						|
  --- @param client? vim.lsp.Client
 | 
						|
  local function try_use_client(idx, client)
 | 
						|
    if not idx or not client then
 | 
						|
      return
 | 
						|
    end
 | 
						|
 | 
						|
    --- @param name string
 | 
						|
    local function rename(name)
 | 
						|
      local params = util.make_position_params(win, client.offset_encoding) --[[@as lsp.RenameParams]]
 | 
						|
      params.newName = name
 | 
						|
      local handler = client.handlers[ms.textDocument_rename]
 | 
						|
        or lsp.handlers[ms.textDocument_rename]
 | 
						|
      client:request(ms.textDocument_rename, params, function(...)
 | 
						|
        handler(...)
 | 
						|
        try_use_client(next(clients, idx))
 | 
						|
      end, bufnr)
 | 
						|
    end
 | 
						|
 | 
						|
    if client:supports_method(ms.textDocument_prepareRename) then
 | 
						|
      local params = util.make_position_params(win, client.offset_encoding)
 | 
						|
      client:request(ms.textDocument_prepareRename, params, function(err, result)
 | 
						|
        if err or result == nil then
 | 
						|
          if next(clients, idx) then
 | 
						|
            try_use_client(next(clients, idx))
 | 
						|
          else
 | 
						|
            local msg = err and ('Error on prepareRename: ' .. (err.message or ''))
 | 
						|
              or 'Nothing to rename'
 | 
						|
            vim.notify(msg, vim.log.levels.INFO)
 | 
						|
          end
 | 
						|
          return
 | 
						|
        end
 | 
						|
 | 
						|
        if new_name then
 | 
						|
          rename(new_name)
 | 
						|
          return
 | 
						|
        end
 | 
						|
 | 
						|
        local prompt_opts = {
 | 
						|
          prompt = 'New Name: ',
 | 
						|
        }
 | 
						|
        -- result: Range | { range: Range, placeholder: string }
 | 
						|
        if result.placeholder then
 | 
						|
          prompt_opts.default = result.placeholder
 | 
						|
        elseif result.start then
 | 
						|
          prompt_opts.default = get_text_at_range(result, client.offset_encoding)
 | 
						|
        elseif result.range then
 | 
						|
          prompt_opts.default = get_text_at_range(result.range, client.offset_encoding)
 | 
						|
        else
 | 
						|
          prompt_opts.default = cword
 | 
						|
        end
 | 
						|
        vim.ui.input(prompt_opts, function(input)
 | 
						|
          if not input or #input == 0 then
 | 
						|
            return
 | 
						|
          end
 | 
						|
          rename(input)
 | 
						|
        end)
 | 
						|
      end, bufnr)
 | 
						|
    else
 | 
						|
      assert(
 | 
						|
        client:supports_method(ms.textDocument_rename),
 | 
						|
        'Client must support textDocument/rename'
 | 
						|
      )
 | 
						|
      if new_name then
 | 
						|
        rename(new_name)
 | 
						|
        return
 | 
						|
      end
 | 
						|
 | 
						|
      local prompt_opts = {
 | 
						|
        prompt = 'New Name: ',
 | 
						|
        default = cword,
 | 
						|
      }
 | 
						|
      vim.ui.input(prompt_opts, function(input)
 | 
						|
        if not input or #input == 0 then
 | 
						|
          return
 | 
						|
        end
 | 
						|
        rename(input)
 | 
						|
      end)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  try_use_client(next(clients))
 | 
						|
end
 | 
						|
 | 
						|
--- Lists all the references to the symbol under the cursor in the quickfix window.
 | 
						|
---
 | 
						|
---@param context lsp.ReferenceContext? Context for the request
 | 
						|
---@see https://microsoft.github.io/language-server-protocol/specifications/specification-current/#textDocument_references
 | 
						|
---@param opts? vim.lsp.ListOpts
 | 
						|
function M.references(context, opts)
 | 
						|
  validate('context', context, 'table', true)
 | 
						|
  local bufnr = api.nvim_get_current_buf()
 | 
						|
  local clients = lsp.get_clients({ method = ms.textDocument_references, bufnr = bufnr })
 | 
						|
  if not next(clients) then
 | 
						|
    return
 | 
						|
  end
 | 
						|
  local win = api.nvim_get_current_win()
 | 
						|
  opts = opts or {}
 | 
						|
 | 
						|
  local all_items = {}
 | 
						|
  local title = 'References'
 | 
						|
 | 
						|
  local function on_done()
 | 
						|
    if not next(all_items) then
 | 
						|
      vim.notify('No references found')
 | 
						|
    else
 | 
						|
      local list = {
 | 
						|
        title = title,
 | 
						|
        items = all_items,
 | 
						|
        context = {
 | 
						|
          method = ms.textDocument_references,
 | 
						|
          bufnr = bufnr,
 | 
						|
        },
 | 
						|
      }
 | 
						|
      if opts.loclist then
 | 
						|
        vim.fn.setloclist(0, {}, ' ', list)
 | 
						|
        vim.cmd.lopen()
 | 
						|
      elseif opts.on_list then
 | 
						|
        assert(vim.is_callable(opts.on_list), 'on_list is not a function')
 | 
						|
        opts.on_list(list)
 | 
						|
      else
 | 
						|
        vim.fn.setqflist({}, ' ', list)
 | 
						|
        vim.cmd('botright copen')
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  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 items = util.locations_to_items(result or {}, client.offset_encoding)
 | 
						|
      vim.list_extend(all_items, items)
 | 
						|
      remaining = remaining - 1
 | 
						|
      if remaining == 0 then
 | 
						|
        on_done()
 | 
						|
      end
 | 
						|
    end)
 | 
						|
  end
 | 
						|
end
 | 
						|
 | 
						|
--- Lists all symbols in the current buffer in the |location-list|.
 | 
						|
--- @param opts? vim.lsp.ListOpts
 | 
						|
function M.document_symbol(opts)
 | 
						|
  opts = vim.tbl_deep_extend('keep', opts or {}, { loclist = true })
 | 
						|
  local params = { textDocument = util.make_text_document_params() }
 | 
						|
  request_with_opts(ms.textDocument_documentSymbol, params, opts)
 | 
						|
end
 | 
						|
 | 
						|
--- @param client_id integer
 | 
						|
--- @param method vim.lsp.protocol.Method.ClientToServer.Request
 | 
						|
--- @param params table
 | 
						|
--- @param handler? lsp.Handler
 | 
						|
--- @param bufnr? integer
 | 
						|
local function request_with_id(client_id, method, params, handler, bufnr)
 | 
						|
  local client = lsp.get_client_by_id(client_id)
 | 
						|
  if not client then
 | 
						|
    vim.notify(
 | 
						|
      string.format('Client with id=%d disappeared during hierarchy request', client_id),
 | 
						|
      vim.log.levels.WARN
 | 
						|
    )
 | 
						|
    return
 | 
						|
  end
 | 
						|
  client:request(method, params, handler, bufnr)
 | 
						|
end
 | 
						|
 | 
						|
--- @param item lsp.TypeHierarchyItem|lsp.CallHierarchyItem
 | 
						|
local function format_hierarchy_item(item)
 | 
						|
  if not item.detail or #item.detail == 0 then
 | 
						|
    return item.name
 | 
						|
  end
 | 
						|
  return string.format('%s %s', item.name, item.detail)
 | 
						|
end
 | 
						|
 | 
						|
local hierarchy_methods = {
 | 
						|
  [ms.typeHierarchy_subtypes] = 'type',
 | 
						|
  [ms.typeHierarchy_supertypes] = 'type',
 | 
						|
  [ms.callHierarchy_incomingCalls] = 'call',
 | 
						|
  [ms.callHierarchy_outgoingCalls] = 'call',
 | 
						|
}
 | 
						|
 | 
						|
--- @param method vim.lsp.protocol.Method.ClientToServer.Request
 | 
						|
local function hierarchy(method)
 | 
						|
  local kind = hierarchy_methods[method]
 | 
						|
  if not kind then
 | 
						|
    error('unsupported method ' .. method)
 | 
						|
  end
 | 
						|
 | 
						|
  local prepare_method = kind == 'type' and ms.textDocument_prepareTypeHierarchy
 | 
						|
    or ms.textDocument_prepareCallHierarchy
 | 
						|
 | 
						|
  local bufnr = api.nvim_get_current_buf()
 | 
						|
  local clients = lsp.get_clients({ bufnr = bufnr, method = prepare_method })
 | 
						|
  if not next(clients) then
 | 
						|
    vim.notify(lsp._unsupported_method(method), vim.log.levels.WARN)
 | 
						|
    return
 | 
						|
  end
 | 
						|
 | 
						|
  local win = api.nvim_get_current_win()
 | 
						|
 | 
						|
  --- @param results [integer, lsp.TypeHierarchyItem|lsp.CallHierarchyItem][]
 | 
						|
  local function on_response(results)
 | 
						|
    if #results == 0 then
 | 
						|
      vim.notify('No item resolved', vim.log.levels.WARN)
 | 
						|
    elseif #results == 1 then
 | 
						|
      local client_id, item = results[1][1], results[1][2]
 | 
						|
      request_with_id(client_id, method, { item = item }, nil, bufnr)
 | 
						|
    else
 | 
						|
      vim.ui.select(results, {
 | 
						|
        prompt = string.format('Select a %s hierarchy item:', kind),
 | 
						|
        kind = kind .. 'hierarchy',
 | 
						|
        format_item = function(x)
 | 
						|
          return format_hierarchy_item(x[2])
 | 
						|
        end,
 | 
						|
      }, function(x)
 | 
						|
        if x then
 | 
						|
          local client_id, item = x[1], x[2]
 | 
						|
          request_with_id(client_id, method, { item = item }, nil, bufnr)
 | 
						|
        end
 | 
						|
      end)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  local results = {} --- @type [integer, lsp.TypeHierarchyItem|lsp.CallHierarchyItem][]
 | 
						|
 | 
						|
  local remaining = #clients
 | 
						|
 | 
						|
  for _, client in ipairs(clients) do
 | 
						|
    local params = util.make_position_params(win, client.offset_encoding)
 | 
						|
    --- @param result lsp.CallHierarchyItem[]|lsp.TypeHierarchyItem[]?
 | 
						|
    client:request(prepare_method, params, function(err, result, ctx)
 | 
						|
      if err then
 | 
						|
        vim.notify(err.message, vim.log.levels.WARN)
 | 
						|
      elseif result then
 | 
						|
        for _, item in ipairs(result) do
 | 
						|
          results[#results + 1] = { ctx.client_id, item }
 | 
						|
        end
 | 
						|
      end
 | 
						|
 | 
						|
      remaining = remaining - 1
 | 
						|
      if remaining == 0 then
 | 
						|
        on_response(results)
 | 
						|
      end
 | 
						|
    end, bufnr)
 | 
						|
  end
 | 
						|
end
 | 
						|
 | 
						|
--- Lists all the call sites of the symbol under the cursor in the
 | 
						|
--- |quickfix| window. If the symbol can resolve to multiple
 | 
						|
--- items, the user can pick one in the |inputlist()|.
 | 
						|
function M.incoming_calls()
 | 
						|
  hierarchy(ms.callHierarchy_incomingCalls)
 | 
						|
end
 | 
						|
 | 
						|
--- Lists all the items that are called by the symbol under the
 | 
						|
--- cursor in the |quickfix| window. If the symbol can resolve to
 | 
						|
--- multiple items, the user can pick one in the |inputlist()|.
 | 
						|
function M.outgoing_calls()
 | 
						|
  hierarchy(ms.callHierarchy_outgoingCalls)
 | 
						|
end
 | 
						|
 | 
						|
--- Lists all the subtypes or supertypes of the symbol under the
 | 
						|
--- cursor in the |quickfix| window. If the symbol can resolve to
 | 
						|
--- multiple items, the user can pick one using |vim.ui.select()|.
 | 
						|
---@param kind "subtypes"|"supertypes"
 | 
						|
function M.typehierarchy(kind)
 | 
						|
  local method = kind == 'subtypes' and ms.typeHierarchy_subtypes or ms.typeHierarchy_supertypes
 | 
						|
  hierarchy(method)
 | 
						|
end
 | 
						|
 | 
						|
--- List workspace folders.
 | 
						|
---
 | 
						|
function M.list_workspace_folders()
 | 
						|
  local workspace_folders = {}
 | 
						|
  for _, client in pairs(lsp.get_clients({ bufnr = 0 })) do
 | 
						|
    for _, folder in pairs(client.workspace_folders or {}) do
 | 
						|
      table.insert(workspace_folders, folder.name)
 | 
						|
    end
 | 
						|
  end
 | 
						|
  return workspace_folders
 | 
						|
end
 | 
						|
 | 
						|
--- Add the folder at path to the workspace folders. If {path} is
 | 
						|
--- not provided, the user will be prompted for a path using |input()|.
 | 
						|
--- @param workspace_folder? string
 | 
						|
function M.add_workspace_folder(workspace_folder)
 | 
						|
  workspace_folder = workspace_folder
 | 
						|
    or npcall(vim.fn.input, 'Workspace Folder: ', vim.fn.expand('%:p:h'), 'dir')
 | 
						|
  api.nvim_command('redraw')
 | 
						|
  if not (workspace_folder and #workspace_folder > 0) then
 | 
						|
    return
 | 
						|
  end
 | 
						|
  if vim.fn.isdirectory(workspace_folder) == 0 then
 | 
						|
    print(workspace_folder, ' is not a valid directory')
 | 
						|
    return
 | 
						|
  end
 | 
						|
  local bufnr = api.nvim_get_current_buf()
 | 
						|
  for _, client in pairs(lsp.get_clients({ bufnr = bufnr })) do
 | 
						|
    client:_add_workspace_folder(workspace_folder)
 | 
						|
  end
 | 
						|
end
 | 
						|
 | 
						|
--- Remove the folder at path from the workspace folders. If
 | 
						|
--- {path} is not provided, the user will be prompted for
 | 
						|
--- a path using |input()|.
 | 
						|
--- @param workspace_folder? string
 | 
						|
function M.remove_workspace_folder(workspace_folder)
 | 
						|
  workspace_folder = workspace_folder
 | 
						|
    or npcall(vim.fn.input, 'Workspace Folder: ', vim.fn.expand('%:p:h'))
 | 
						|
  api.nvim_command('redraw')
 | 
						|
  if not workspace_folder or #workspace_folder == 0 then
 | 
						|
    return
 | 
						|
  end
 | 
						|
  local bufnr = api.nvim_get_current_buf()
 | 
						|
  for _, client in pairs(lsp.get_clients({ bufnr = bufnr })) do
 | 
						|
    client:_remove_workspace_folder(workspace_folder)
 | 
						|
  end
 | 
						|
  print(workspace_folder, 'is not currently part of the workspace')
 | 
						|
end
 | 
						|
 | 
						|
--- Lists all symbols in the current workspace in the quickfix window.
 | 
						|
---
 | 
						|
--- The list is filtered against {query}; if the argument is omitted from the
 | 
						|
--- call, the user is prompted to enter a string on the command line. An empty
 | 
						|
--- string means no filtering is done.
 | 
						|
---
 | 
						|
--- @param query string? optional
 | 
						|
--- @param opts? vim.lsp.ListOpts
 | 
						|
function M.workspace_symbol(query, opts)
 | 
						|
  query = query or npcall(vim.fn.input, 'Query: ')
 | 
						|
  if query == nil then
 | 
						|
    return
 | 
						|
  end
 | 
						|
  local params = { query = query }
 | 
						|
  request_with_opts(ms.workspace_symbol, params, opts)
 | 
						|
end
 | 
						|
 | 
						|
--- Send request to the server to resolve document highlights for the current
 | 
						|
--- text document position. This request can be triggered by a  key mapping or
 | 
						|
--- by events such as `CursorHold`, e.g.:
 | 
						|
---
 | 
						|
--- ```vim
 | 
						|
--- autocmd CursorHold  <buffer> lua vim.lsp.buf.document_highlight()
 | 
						|
--- autocmd CursorHoldI <buffer> lua vim.lsp.buf.document_highlight()
 | 
						|
--- autocmd CursorMoved <buffer> lua vim.lsp.buf.clear_references()
 | 
						|
--- ```
 | 
						|
---
 | 
						|
--- Note: Usage of |vim.lsp.buf.document_highlight()| requires the following highlight groups
 | 
						|
---       to be defined or you won't be able to see the actual highlights.
 | 
						|
---         |hl-LspReferenceText|
 | 
						|
---         |hl-LspReferenceRead|
 | 
						|
---         |hl-LspReferenceWrite|
 | 
						|
function M.document_highlight()
 | 
						|
  lsp.buf_request(0, ms.textDocument_documentHighlight, client_positional_params())
 | 
						|
end
 | 
						|
 | 
						|
--- Removes document highlights from current buffer.
 | 
						|
function M.clear_references()
 | 
						|
  util.buf_clear_references()
 | 
						|
end
 | 
						|
 | 
						|
---@nodoc
 | 
						|
---@class vim.lsp.CodeActionResultEntry
 | 
						|
---@field error? lsp.ResponseError
 | 
						|
---@field result? (lsp.Command|lsp.CodeAction)[]
 | 
						|
---@field ctx lsp.HandlerContext
 | 
						|
 | 
						|
--- @class vim.lsp.buf.code_action.Opts
 | 
						|
--- @inlinedoc
 | 
						|
---
 | 
						|
--- Corresponds to `CodeActionContext` of the LSP specification:
 | 
						|
---   - {diagnostics}? (`table`) LSP `Diagnostic[]`. Inferred from the current
 | 
						|
---     position if not provided.
 | 
						|
---   - {only}? (`table`) List of LSP `CodeActionKind`s used to filter the code actions.
 | 
						|
---     Most language servers support values like `refactor`
 | 
						|
---     or `quickfix`.
 | 
						|
---   - {triggerKind}? (`integer`) The reason why code actions were requested.
 | 
						|
--- @field context? lsp.CodeActionContext
 | 
						|
---
 | 
						|
--- Predicate taking an `CodeAction` and returning a boolean.
 | 
						|
--- @field filter? fun(x: lsp.CodeAction|lsp.Command):boolean
 | 
						|
---
 | 
						|
--- When set to `true`, and there is just one remaining action
 | 
						|
--- (after filtering), the action is applied without user query.
 | 
						|
--- @field apply? boolean
 | 
						|
---
 | 
						|
--- Range for which code actions should be requested.
 | 
						|
--- If in visual mode this defaults to the active selection.
 | 
						|
--- Table must contain `start` and `end` keys with {row,col} tuples
 | 
						|
--- using mark-like indexing. See |api-indexing|
 | 
						|
--- @field range? {start: integer[], end: integer[]}
 | 
						|
 | 
						|
--- This is not public because the main extension point is
 | 
						|
--- vim.ui.select which can be overridden independently.
 | 
						|
---
 | 
						|
--- Can't call/use vim.lsp.handlers['textDocument/codeAction'] because it expects
 | 
						|
--- `(err, CodeAction[] | Command[], ctx)`, but we want to aggregate the results
 | 
						|
--- from multiple clients to have 1 single UI prompt for the user, yet we still
 | 
						|
--- need to be able to link a `CodeAction|Command` to the right client for
 | 
						|
--- `codeAction/resolve`
 | 
						|
---@param results table<integer, vim.lsp.CodeActionResultEntry>
 | 
						|
---@param opts? vim.lsp.buf.code_action.Opts
 | 
						|
local function on_code_action_results(results, opts)
 | 
						|
  ---@param a lsp.Command|lsp.CodeAction
 | 
						|
  local function action_filter(a)
 | 
						|
    -- filter by specified action kind
 | 
						|
    if opts and opts.context and opts.context.only then
 | 
						|
      if not a.kind then
 | 
						|
        return false
 | 
						|
      end
 | 
						|
      local found = false
 | 
						|
      for _, o in ipairs(opts.context.only) do
 | 
						|
        -- action kinds are hierarchical with . as a separator: when requesting only 'type-annotate'
 | 
						|
        -- this filter allows both 'type-annotate' and 'type-annotate.foo', for example
 | 
						|
        if a.kind == o or vim.startswith(a.kind, o .. '.') then
 | 
						|
          found = true
 | 
						|
          break
 | 
						|
        end
 | 
						|
      end
 | 
						|
      if not found then
 | 
						|
        return false
 | 
						|
      end
 | 
						|
    end
 | 
						|
    -- filter by user function
 | 
						|
    if opts and opts.filter and not opts.filter(a) then
 | 
						|
      return false
 | 
						|
    end
 | 
						|
    -- no filter removed this action
 | 
						|
    return true
 | 
						|
  end
 | 
						|
 | 
						|
  ---@type {action: lsp.Command|lsp.CodeAction, ctx: lsp.HandlerContext}[]
 | 
						|
  local actions = {}
 | 
						|
  for _, result in pairs(results) do
 | 
						|
    for _, action in pairs(result.result or {}) do
 | 
						|
      if action_filter(action) then
 | 
						|
        table.insert(actions, { action = action, ctx = result.ctx })
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
  if #actions == 0 then
 | 
						|
    vim.notify('No code actions available', vim.log.levels.INFO)
 | 
						|
    return
 | 
						|
  end
 | 
						|
 | 
						|
  ---@param action lsp.Command|lsp.CodeAction
 | 
						|
  ---@param client vim.lsp.Client
 | 
						|
  ---@param ctx lsp.HandlerContext
 | 
						|
  local function apply_action(action, client, ctx)
 | 
						|
    if action.edit then
 | 
						|
      util.apply_workspace_edit(action.edit, client.offset_encoding)
 | 
						|
    end
 | 
						|
    local a_cmd = action.command
 | 
						|
    if a_cmd then
 | 
						|
      local command = type(a_cmd) == 'table' and a_cmd or action
 | 
						|
      --- @cast command lsp.Command
 | 
						|
      client:exec_cmd(command, ctx)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  ---@param choice {action: lsp.Command|lsp.CodeAction, ctx: lsp.HandlerContext}
 | 
						|
  local function on_user_choice(choice)
 | 
						|
    if not choice then
 | 
						|
      return
 | 
						|
    end
 | 
						|
 | 
						|
    -- textDocument/codeAction can return either Command[] or CodeAction[]
 | 
						|
    --
 | 
						|
    -- CodeAction
 | 
						|
    --  ...
 | 
						|
    --  edit?: WorkspaceEdit    -- <- must be applied before command
 | 
						|
    --  command?: Command
 | 
						|
    --
 | 
						|
    -- Command:
 | 
						|
    --  title: string
 | 
						|
    --  command: string
 | 
						|
    --  arguments?: any[]
 | 
						|
 | 
						|
    local client = assert(lsp.get_client_by_id(choice.ctx.client_id))
 | 
						|
    local action = choice.action
 | 
						|
    local bufnr = assert(choice.ctx.bufnr, 'Must have buffer number')
 | 
						|
 | 
						|
    -- Only code actions are resolved, so if we have a command, just apply it.
 | 
						|
    if type(action.title) == 'string' and type(action.command) == 'string' then
 | 
						|
      apply_action(action, client, choice.ctx)
 | 
						|
      return
 | 
						|
    end
 | 
						|
 | 
						|
    if not (action.edit and action.command) and client:supports_method(ms.codeAction_resolve) then
 | 
						|
      client:request(ms.codeAction_resolve, action, function(err, resolved_action)
 | 
						|
        if err then
 | 
						|
          -- If resolve fails, try to apply the edit/command from the original code action.
 | 
						|
          if action.edit or action.command then
 | 
						|
            apply_action(action, client, choice.ctx)
 | 
						|
          else
 | 
						|
            vim.notify(err.code .. ': ' .. err.message, vim.log.levels.ERROR)
 | 
						|
          end
 | 
						|
        else
 | 
						|
          apply_action(resolved_action, client, choice.ctx)
 | 
						|
        end
 | 
						|
      end, bufnr)
 | 
						|
    else
 | 
						|
      apply_action(action, client, choice.ctx)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  -- If options.apply is given, and there are just one remaining code action,
 | 
						|
  -- apply it directly without querying the user.
 | 
						|
  if opts and opts.apply and #actions == 1 then
 | 
						|
    on_user_choice(actions[1])
 | 
						|
    return
 | 
						|
  end
 | 
						|
 | 
						|
  ---@param item {action: lsp.Command|lsp.CodeAction, ctx: lsp.HandlerContext}
 | 
						|
  local function format_item(item)
 | 
						|
    local clients = lsp.get_clients({ bufnr = item.ctx.bufnr })
 | 
						|
    local title = item.action.title:gsub('\r\n', '\\r\\n'):gsub('\n', '\\n')
 | 
						|
 | 
						|
    if #clients == 1 then
 | 
						|
      return title
 | 
						|
    end
 | 
						|
 | 
						|
    local source = lsp.get_client_by_id(item.ctx.client_id).name
 | 
						|
    return ('%s [%s]'):format(title, source)
 | 
						|
  end
 | 
						|
 | 
						|
  local select_opts = {
 | 
						|
    prompt = 'Code actions:',
 | 
						|
    kind = 'codeaction',
 | 
						|
    format_item = format_item,
 | 
						|
  }
 | 
						|
  vim.ui.select(actions, select_opts, on_user_choice)
 | 
						|
end
 | 
						|
 | 
						|
--- Selects a code action (LSP: "textDocument/codeAction" request) available at cursor position.
 | 
						|
---
 | 
						|
---@param opts? vim.lsp.buf.code_action.Opts
 | 
						|
---@see https://microsoft.github.io/language-server-protocol/specifications/specification-current/#textDocument_codeAction
 | 
						|
---@see vim.lsp.protocol.CodeActionTriggerKind
 | 
						|
function M.code_action(opts)
 | 
						|
  validate('options', opts, 'table', true)
 | 
						|
  opts = opts or {}
 | 
						|
  -- Detect old API call code_action(context) which should now be
 | 
						|
  -- code_action({ context = context} )
 | 
						|
  --- @diagnostic disable-next-line:undefined-field
 | 
						|
  if opts.diagnostics or opts.only then
 | 
						|
    opts = { options = opts }
 | 
						|
  end
 | 
						|
  local context = opts.context and vim.deepcopy(opts.context) or {}
 | 
						|
  if not context.triggerKind then
 | 
						|
    context.triggerKind = lsp.protocol.CodeActionTriggerKind.Invoked
 | 
						|
  end
 | 
						|
  local mode = api.nvim_get_mode().mode
 | 
						|
  local bufnr = api.nvim_get_current_buf()
 | 
						|
  local win = api.nvim_get_current_win()
 | 
						|
  local clients = lsp.get_clients({ bufnr = bufnr, method = ms.textDocument_codeAction })
 | 
						|
  local remaining = #clients
 | 
						|
  if remaining == 0 then
 | 
						|
    if next(lsp.get_clients({ bufnr = bufnr })) then
 | 
						|
      vim.notify(lsp._unsupported_method(ms.textDocument_codeAction), vim.log.levels.WARN)
 | 
						|
    end
 | 
						|
    return
 | 
						|
  end
 | 
						|
 | 
						|
  ---@type table<integer, vim.lsp.CodeActionResultEntry>
 | 
						|
  local results = {}
 | 
						|
 | 
						|
  ---@param err? lsp.ResponseError
 | 
						|
  ---@param result? (lsp.Command|lsp.CodeAction)[]
 | 
						|
  ---@param ctx lsp.HandlerContext
 | 
						|
  local function on_result(err, result, ctx)
 | 
						|
    results[ctx.client_id] = { error = err, result = result, ctx = ctx }
 | 
						|
    remaining = remaining - 1
 | 
						|
    if remaining == 0 then
 | 
						|
      on_code_action_results(results, opts)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  for _, client in ipairs(clients) do
 | 
						|
    ---@type lsp.CodeActionParams
 | 
						|
    local params
 | 
						|
 | 
						|
    if opts.range then
 | 
						|
      assert(type(opts.range) == 'table', 'code_action range must be a table')
 | 
						|
      local start = assert(opts.range.start, 'range must have a `start` property')
 | 
						|
      local end_ = assert(opts.range['end'], 'range must have a `end` property')
 | 
						|
      params = util.make_given_range_params(start, end_, bufnr, client.offset_encoding)
 | 
						|
    elseif mode == 'v' or mode == 'V' then
 | 
						|
      local range = range_from_selection(bufnr, mode)
 | 
						|
      params =
 | 
						|
        util.make_given_range_params(range.start, range['end'], bufnr, client.offset_encoding)
 | 
						|
    else
 | 
						|
      params = util.make_range_params(win, client.offset_encoding)
 | 
						|
    end
 | 
						|
 | 
						|
    --- @cast params lsp.CodeActionParams
 | 
						|
 | 
						|
    if context.diagnostics then
 | 
						|
      params.context = context
 | 
						|
    else
 | 
						|
      local ns_push = lsp.diagnostic.get_namespace(client.id, false)
 | 
						|
      local ns_pull = lsp.diagnostic.get_namespace(client.id, true)
 | 
						|
      local diagnostics = {}
 | 
						|
      local lnum = api.nvim_win_get_cursor(0)[1] - 1
 | 
						|
      vim.list_extend(diagnostics, vim.diagnostic.get(bufnr, { namespace = ns_pull, lnum = lnum }))
 | 
						|
      vim.list_extend(diagnostics, vim.diagnostic.get(bufnr, { namespace = ns_push, lnum = lnum }))
 | 
						|
      params.context = vim.tbl_extend('force', context, {
 | 
						|
        ---@diagnostic disable-next-line: no-unknown
 | 
						|
        diagnostics = vim.tbl_map(function(d)
 | 
						|
          return d.user_data.lsp
 | 
						|
        end, diagnostics),
 | 
						|
      })
 | 
						|
    end
 | 
						|
 | 
						|
    client:request(ms.textDocument_codeAction, params, on_result, bufnr)
 | 
						|
  end
 | 
						|
end
 | 
						|
 | 
						|
--- @deprecated
 | 
						|
--- Executes an LSP server command.
 | 
						|
--- @param command_params lsp.ExecuteCommandParams
 | 
						|
--- @see https://microsoft.github.io/language-server-protocol/specifications/specification-current/#workspace_executeCommand
 | 
						|
function M.execute_command(command_params)
 | 
						|
  validate('command', command_params.command, 'string')
 | 
						|
  validate('arguments', command_params.arguments, 'table', true)
 | 
						|
  vim.deprecate('execute_command', 'client:exec_cmd', '0.12')
 | 
						|
  command_params = {
 | 
						|
    command = command_params.command,
 | 
						|
    arguments = command_params.arguments,
 | 
						|
    workDoneToken = command_params.workDoneToken,
 | 
						|
  }
 | 
						|
  lsp.buf_request(0, ms.workspace_executeCommand, command_params)
 | 
						|
end
 | 
						|
 | 
						|
return M
 |