navigator.lua/lua/navigator/codelens.lua

-- codelenses
-- https://github.com/josa42/nvim-lsp-codelenses/blob/master/lua/jg/lsp/codelenses.lua
-- https://github.com/neovim/neovim/blob/master/runtime/lua/vim/lsp/codelens.lua
local codelens = require('vim.lsp.codelens')

local log = require('navigator.util').log
local mk_handler = require('navigator.util').mk_handler
local nvim_0_6_1 = require('navigator.util').nvim_0_6_1
local trace = require('navigator.util').trace

local lsphelper = require('navigator.lspwrapper')
local api = vim.api
local gui = require('navigator.gui')
local M = {}

local config = require('navigator').config_values()
local sign_name = 'NavigatorCodeLensLightBulb'
if vim.tbl_isempty(vim.fn.sign_getdefined(sign_name)) then
  vim.fn.sign_define(sign_name, { text = config.icons.code_lens_action_icon, texthl = 'LspDiagnosticsSignHint' })
end

local sign_group = 'nvcodelensaction'

local get_current_winid = require('navigator.util').get_current_winid

local is_enabled = true
local code_lens_action = {}

local function _update_sign(line)
  trace('update sign at line ', line)
  local winid = get_current_winid()
  if code_lens_action[winid] == nil then
    code_lens_action[winid] = {}
  end
  if code_lens_action[winid].lightbulb_line ~= 0 then
    vim.fn.sign_unplace(sign_group, { id = code_lens_action[winid].lightbulb_line, buffer = '%' })
  end

  if line then
    -- log("updatasign", line, sign_group, sign_name)
    vim.fn.sign_place(
      line,
      sign_group,
      sign_name,
      '%',
      { lnum = line + 1, priority = config.lsp.code_lens_action.sign_priority }
    )
    code_lens_action[winid].lightbulb_line = line
  end
end

local codelens_hdlr = mk_handler(function(err, result, ctx, cfg)
  trace(ctx, result)
  M.codelens_ctx = ctx
  if err or result == nil then
    if err then
      log('lsp code lens', vim.inspect(err), ctx, cfg)
    end
    return
  end
  trace('codelenes result', result)
  for _, v in pairs(result) do
    _update_sign(v.range.start.line)
  end
end)

function M.setup()
  vim.cmd('highlight! link LspCodeLens LspDiagnosticsHint')
  vim.cmd('highlight! link LspCodeLensText LspDiagnosticsInformation')
  vim.cmd('highlight! link LspCodeLensTextSign LspDiagnosticsSignInformation')
  vim.cmd('highlight! link LspCodeLensTextSeparator Boolean')

  vim.cmd('augroup navigator.codelenses')
  vim.cmd('  autocmd!')
  vim.cmd("autocmd BufEnter,CursorHold,InsertLeave <buffer> lua require('navigator.codelens').refresh()")
  vim.cmd('augroup end')
  local on_codelens = vim.lsp.handlers['textDocument/codeLens']
  vim.lsp.handlers['textDocument/codeLens'] = mk_handler(function(err, result, ctx, cfg)
    -- trace(err, result, ctx.client_id, ctx.bufnr, cfg or {})
    cfg = cfg or {}
    ctx = ctx or { bufnr = vim.api.nvim_get_current_buf() }
    if nvim_0_6_1() then
      on_codelens(err, result, ctx, cfg)
      codelens_hdlr(err, result, ctx, cfg)
    else
      on_codelens(err, ctx.method, result, ctx.client_id, ctx.bufnr)
      codelens_hdlr(err, nil, result, ctx.client_id or 0, ctx.bufnr or 0)
    end
  end)
end

M.lsp_clients = {}

function M.refresh()
  if #vim.lsp.buf_get_clients() < 1 then
    log('Must have a client running to use lsp code action')
    return
  end
  if not lsphelper.check_capabilities('code_lens') then
    return
  end
  vim.lsp.codelens.refresh()
end

local virtual_types_ns = api.nvim_create_namespace('ng_virtual_types')

function M.disable()
  local bufnr = vim.api.nvim_get_current_buf()
  vim.api.nvim_buf_clear_namespace(bufnr, virtual_types_ns, 0, -1)
  is_enabled = false
end

function M.run_action()
  local original_select = vim.ui.select
  vim.ui.select = require('guihua.gui').select

  log('codeaction')

  codelens.run()
  vim.defer_fn(function()
    vim.ui.select = original_select
  end, 1000)
end

M.inline = function()
  local lsp = vim.lsp
  if is_enabled == false then
    return
  end
  if vim.fn.getcmdwintype() == ':' then
    return
  end
  if #vim.lsp.buf_get_clients() == 0 then
    return
  end

  local bufnr = api.nvim_get_current_buf()
  local parameter = lsp.util.make_position_params()
  local response = lsp.buf_request_sync(bufnr, 'textDocument/codeLens', parameter)

  -- Clear previous highlighting
  api.nvim_buf_clear_namespace(bufnr, virtual_types_ns, 0, -1)

  if response then
    log(response)
    for _, v in ipairs(response) do
      if v == nil or v.result == nil then
        return
      end -- no response
      for _, vv in pairs(v.result) do
        local start_line = -1
        for _, vvv in pairs(vv.range) do
          start_line = tonumber(vvv.line)
        end

        local cmd = vv.command
        local msg = _NgConfigValues.icons.code_action_icon .. ' '
        if cmd then
          local txt = cmd.title or ''
          txt = txt .. ' ' .. (cmd.command or '') .. ' '
          msg = msg .. txt .. ' '
        end

        log(msg)
        api.nvim_buf_set_extmark(bufnr, virtual_types_ns, start_line, -1, {
          virt_text = { { msg, 'LspCodeLensText' } },
          virt_text_pos = 'overlay',
          hl_mode = 'combine',
        })
      end
    end
    -- else
    --   api.nvim_command("echohl WarningMsg | echo 'VirtualTypes: No response' | echohl None")
  end
end

return M